class TestOrder(unittest.TestCase):
def setUp(self):
self.inventory = Inventory()
self.inventory.add("item #1", 50)
def test_order_filled(self):
order = Order("item #1", 50)
order.fill(self.inventory)
self.assertTrue(order.filled)
self.assertEquals(0, self.inventory.get("item #1"))
def test_order_not_filled(self):
order = Order("item #1", 51)
order.fill(self.inventory)
self.assertFalse(order.filled)
self.assertEquals(50, self.inventory.get("item #1"))
def test_add_item_into_inventory():
inventory = Inventory()
inventory.add("item #1", 1)
assert inventory.get("item #1") == 1
def test_remove_invalid_quantity_inventory():
inventory = Inventory()
inventory.add("item #1", 3).remove("item #1", 5)
assert inventory.get("item #1") == 3
def test_remove_then_add_item_in_inventory():
inventory = Inventory()
inventory.add("item #1", 5)
inventory.remove("item #1", 3).add("item #1", 4)
assert inventory.get("item #1") == 6
def test_remove_item_from_inventory():
inventory = Inventory()
inventory.add("item #1", 5).remove("item #1", 3)
assert inventory.get("item #1") == 2
