def test_update_inventory(self):
""" Update an existing inventory """
inventory = Inventory(product_id=1,
quantity=100,
restock_level=50,
condition="new",
available=True)
inventory.save()
self.assertNotEqual(inventory.id, None)
# Change it and save it
inventory.product_id = 2
inventory.quantity = 200
inventory.restock_level = 100
inventory.condition = "used"
inventory.available = False
inventory.save()
# Fetch it back and make sure the id hasn't change
# but the data did change
inventory = Inventory.all()
self.assertEqual(len(inventory), 1)
self.assertNotEqual(inventory[0].id, None)
self.assertEqual(inventory[0].product_id, 2)
self.assertEqual(inventory[0].quantity, 200)
self.assertEqual(inventory[0].restock_level, 100)
self.assertEqual(inventory[0].condition, 'used')
self.assertEqual(inventory[0].available, False)
def test_disable_an_inventory(self):
""" Disable an existing product """
inventory = Inventory(product_id=1, quantity=100,
restock_level=50, condition="new",\
available=True)
inventory.save()
self.assertEqual(inventory.available, True)
# Change the status and save it
inventory.available = False
inventory.save()
# Fetch it back and make sure the data did change
inventory = Inventory.all()
self.assertEqual(len(inventory), 1)
self.assertEqual(inventory[0].available, False)