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_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)
def test_invalid_obj(self):
value = 'str obj'
try:
favorite_count(value)
except Exception, e:
self.assertEqual(e.__class__, ValueError)
