def test_has_reservation(self):
"""
Test has_reservation method
"""
self.assertEqual(self.food.has_reservation(), False)
create_reservation(food=self.food, user=self.user, quantity=3)
self.assertEqual(self.food.has_reservation(), True)
def test_is_reserved_by_me(self):
"""
Test is_reserved_by_me method
"""
create_reservation(food=self.food, user=self.user, quantity=3)
self.assertEqual(self.food.is_reserved_by_me(self.user), True)
self.assertEqual(self.food.is_reserved_by_me(self.another_user), False)
def test_quantity_available(self):
"""
Test quantity_available method
"""
self.assertEqual(self.food.quantity_available(), range(1, 5))
create_reservation(food=self.food, user=self.another_user, quantity=2)
self.assertEqual(self.food.quantity_available(), range(1, 3))
create_reservation(food=self.food, user=self.another_user, quantity=1)
self.assertEqual(self.food.quantity_available(), range(1, 2))
def test_food_reserved(self):
"""
If you have reserved food
"""
self.client.login(username='******', password='******')
food = create_food(self.fridge, self.user)
create_reservation(food, self.user, 4)
response = self.client.get(
reverse('fridge:food-list', args=(self.fridge.id, )))
self.assertEqual(response.status_code, 200)
self.assertQuerysetEqual(response.context['food_available'], [])
self.assertQuerysetEqual(response.context['food_reserve'],
['<Food: food_test>'])
def test_available_and_reserved_food(self):
"""
Test get_available_food and get_reserved_food method
"""
food = create_food(self.fridge, self.user)
self.assertEqual(self.fridge.get_available_food().count(), 1)
self.assertEqual(len(self.fridge.get_reserved_food()), 0)
reservation = create_reservation(food, self.user2, 2)
self.assertEqual(self.fridge.get_available_food().count(), 1)
self.assertEqual(len(self.fridge.get_reserved_food()), 1)
reservation.delete()
create_reservation(food, self.user2, 4)
self.assertEqual(self.fridge.get_available_food().count(), 0)
self.assertEqual(len(self.fridge.get_reserved_food()), 1)
def setUp(self):
self.user = create_user('test', '*****@*****.**', 'test')
self.fridge = create_fridge(self.user, is_active=True)
self.food = create_food(fridge=self.fridge, user=self.user)
self.reservation = create_reservation(self.food, self.user, 3)
self.client.login(username='******', password='******')