def test_normal_case(self): jeff = User.objects.create_user('jeff', '*****@*****.**') vinky = User.objects.create_user('vinky', '*****@*****.**') vera = User.objects.create_user('vera', '*****@*****.**') Favorite.objects.create_favorite(jeff, vinky) Favorite.objects.create_favorite(vera, vinky) result = favorite_count(vinky) self.assertEqual('2', result)
def test_invalid_obj(self): value = 'str obj' try: favorite_count(value) except Exception, e: self.assertEqual(e.__class__, ValueError)