def save_idea_by_json(json_data, user): idea = Idea() idea.title = json_data['title'] idea.description = json_data['description'] idea.category = json_data['category'] idea.tags = json_data['tags'] idea.author = user return save_idea(idea)
def add_idea(user): idea = Idea() idea.title = '{} Awesome Idea'.format(user.username) idea.description = 'Description of an awesome Idea of {} with the title {}'.format(user.username, idea.title) idea.category = 'Engineering' idea.tags = '{},tag-{},#{},{}'.format(user.name.lower(), user.name.lower(), user.name.lower(), idea.title) idea.user_id = user.id db.session.add(idea) db.session.commit()
def test_get_random_unvoted_idea_for_user(self): self.addModel(self.testUser) self.addModel(self.testIdea) idea = Idea() idea.title = 'Unique Title' idea.category = 'Unique Category' idea.tags = 'Unique Tag' self.addModel(idea) self.assertEqual(self.testIdea, get_random_unvoted_idea_for_user(self.testUser.id))
def test_get_ideas_by_search(self): self.addTestModels() idea = Idea() idea.title = 'Unique Title' idea.category = 'Unique Category' idea.tags = 'Unique Tag' idea.user_id = self.testUser.id self.addModel(idea) search = Search() self.assertCountEqual([self.testIdea, idea], get_ideas_by_search(search)) search.title = 'Unique' self.assertEqual([idea], get_ideas_by_search(search)) search.category = 'Unique Category' self.assertEqual([idea], get_ideas_by_search(search)) search.tags = 'Unique' self.assertEqual([idea], get_ideas_by_search(search))