def test_get_categories(self): ''' Testing the method get_categories ''' # Retrieving one category categories = IdeaService.get_categories(1) self.assertEqual(len(categories), 1) # Retrieving all the categories categories = IdeaService.get_categories(10) self.assertEqual(len(categories), 3) # Using None categories = IdeaService.get_categories(None) self.assertEqual(len(categories), 3)
def get_queryset(self): return IdeaService.get_categories(8)