def test_get_ideas_by_category(self):
'''
Testing the method get_ideas_by_category
'''
# Existing category
category_a_ideas = IdeaService.get_ideas_by_category(self.category.id)
self.assertEqual(len(category_a_ideas), 2)
# Make sure it contains only the desired fields
idea = category_a_ideas[0]
self.assertIn('id', idea)
self.assertIn('title', idea)
self.assertIn('user__first_name', idea)
self.assertIn('created_date', idea)
self.assertIn('excerpt', idea)
self.assertIn('votes', idea)
# Category without idea
category_a_ideas = IdeaService.get_ideas_by_category(2)
self.assertEqual(len(category_a_ideas), 0)
# Category does not exist
category_a_ideas = IdeaService.get_ideas_by_category(999)
self.assertEqual(len(category_a_ideas), 0)
# Using none
category_a_ideas = IdeaService.get_ideas_by_category(None)
self.assertEqual(len(category_a_ideas), 0)
def test_get_ideas_by_category(self):
'''
Testing the method get_ideas_by_category
'''
# Existing category
category_a_ideas = IdeaService.get_ideas_by_category(
self.category.id
)
self.assertEqual(len(category_a_ideas), 2)
# Make sure it contains only the desired fields
idea = category_a_ideas[0]
self.assertIn('id', idea)
self.assertIn('title', idea)
self.assertIn('user__first_name', idea)
self.assertIn('created_date', idea)
self.assertIn('excerpt', idea)
self.assertIn('votes', idea)
# Category without idea
category_a_ideas = IdeaService.get_ideas_by_category(2)
self.assertEqual(len(category_a_ideas), 0)
# Category does not exist
category_a_ideas = IdeaService.get_ideas_by_category(999)
self.assertEqual(len(category_a_ideas), 0)
# Using none
category_a_ideas = IdeaService.get_ideas_by_category(None)
self.assertEqual(len(category_a_ideas), 0)
def get_queryset(self):
'''
Return a dictionnary of fields, it will not be objects.
'''
return IdeaService.get_ideas_by_category(
self.kwargs['category_id']
)
def get_queryset(self):
'''
Return a dictionnary of fields, it will not be objects.
'''
return IdeaService.get_ideas_by_category(self.kwargs['category_id'])
