def test_get_all_ideas(self): self.addTestModels() idea = Idea() idea.title = 'Testy Title' idea.description = 'Testy Description' idea.user_id = self.testUser.id self.addModel(idea) self.assertCountEqual([idea, self.testIdea], get_all_ideas())
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_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))