class Shop(object):
def __init__(self, name):
self.name = name
self.stock = Stock()
def add_stock(self, item, quantity):
self.stock.update_stock(item, quantity)
class TestStock(unittest.TestCase):
def setUp(self):
self.stock = Stock()
def test_stock_adds_new_stock_items(self):
item = StockItem('Fresh Diary Milk', 1000)
self.stock.update_stock(item, 20)
self.assertEqual(
self.stock.stock_items, {
'Fresh Diary Milk': {
'name': 'Fresh Diary Milk',
'quantity': 20,
'unit_price': 1000
}
})
def test_stock_adds_multiple_stock_items(self):
items = [
StockItem('Fresh Diary Milk', 1000),
StockItem('Bic Pen', 2000)
]
for item in items:
self.stock.update_stock(item, 20)
self.assertEqual(
self.stock.stock_items, {
'Fresh Diary Milk': {
'name': 'Fresh Diary Milk',
'quantity': 20,
'unit_price': 1000
},
'Bic Pen': {
'name': 'Bic Pen',
'quantity': 20,
'unit_price': 2000
}
})
def test_stock_updates_existing_stock_item(self):
item = StockItem('Fresh Diary Milk', 1000)
self.stock.update_stock(item, 20)
self.stock.update_stock(item, 40)
self.assertEqual(
self.stock.stock_items, {
'Fresh Diary Milk': {
'name': 'Fresh Diary Milk',
'quantity': 60,
'unit_price': 1000
}
})
def test_stock_get_sub_total_return_the_cost_of_a_line_item(self):
item = StockItem('Fresh Diary Milk', 1000)
self.stock.update_stock(item, 20)
line_item = LineItem('Fresh Diary Milk', 10)
sub_total = self.stock.get_sub_total(line_item)
self.assertEqual(sub_total, 10000)