def test_loading_products(self):
machine = Machine(slots=3, slot_depth=10)
products = {
ProductName("coca-cola"):
Product(name=ProductName("coca-cola"),
quantity=7,
price=Decimal('2.1')),
ProductName("mars"):
Product(name=ProductName("mars"), quantity=5,
price=Decimal('1.9')),
ProductName("orbit"):
Product(name=ProductName("orbit"),
quantity=6,
price=Decimal('2.3')),
}
machine.load_products(products)
available_products = machine.get_available_products().keys()
self.assertIn(ProductName("coca-cola"), available_products)
self.assertIn(ProductName("mars"), available_products)
self.assertIn(ProductName("orbit"), available_products)
self.assertNotIn(ProductName("lays"), available_products)
def test_machine_keeps_track_of_available_products(self):
machine = Machine(slots=1, slot_depth=7)
products = {
ProductName("coca-cola"):
Product(name=ProductName("coca-cola"),
quantity=7,
price=Decimal('2.1'))
}
machine.load_products(products)
slot_code, _ = machine.get_available_products()[ProductName(
'coca-cola')]
for _ in range(7):
product, change = machine.choose_product(
slot_code, Coins({Decimal('2.1'): 1}))
self.assertIsNotNone(product)
self.assertIsNone(change)
new_cola, change = machine.choose_product(slot_code,
Coins({Decimal('2.1'): 1}))
self.assertIsNone(new_cola)
self.assertIsNotNone(change)
self.assertEqual(change, Coins({Decimal('2.1'): 1}))
def test_cant_buy_unavailable_product(self):
machine = Machine(slots=3, slot_depth=10)
products = {
ProductName("coca-cola"):
Product(name=ProductName("coca-cola"),
quantity=1,
price=Decimal('2.1')),
ProductName("mars"):
Product(name=ProductName("mars"), quantity=5,
price=Decimal('1.9')),
ProductName("orbit"):
Product(name=ProductName("orbit"),
quantity=6,
price=Decimal('2.3')),
}
machine.load_products(products)
money = Coins({Decimal(5): 1})
nonexistent_slot_code = SlotCode("doesn't exist")
self.assertNotIn(
nonexistent_slot_code,
[code for code, _ in machine.get_available_products().items()])
product, change = machine.choose_product(nonexistent_slot_code, money)
self.assertIsNone(product)
self.assertEqual(money, change)
self.assertEqual(machine.get_balance(), Decimal(0))
def test_can_buy_available_product_with_exact_change(self):
machine = Machine(slots=3, slot_depth=10)
products = {
ProductName("coca-cola"): Product(name=ProductName("coca-cola"), quantity=7, price=Decimal('2.1')),
ProductName("mars"): Product(name=ProductName("mars"), quantity=5, price=Decimal('1.9')),
ProductName("orbit"): Product(name=ProductName("orbit"), quantity=6, price=Decimal('2.3')),
}
money = Coins({Decimal(2): 1, Decimal('0.1'): 1})
machine.load_products(products)
slot_code, _ = machine.get_available_products()[ProductName("coca-cola")]
product, change = machine.choose_product(slot_code, money)
self.assertIsNotNone(product)
self.assertIsNone(change)
self.assertEqual(product.name, ProductName("coca-cola"))
def test_too_many_product_kinds(self):
machine = Machine(slots=2, slot_depth=10)
products = {
ProductName("coca-cola"): Product(name=ProductName("coca-cola"), quantity=7, price=Decimal('2.1')),
ProductName("mars"): Product(name=ProductName("mars"), quantity=5, price=Decimal('1.9')),
ProductName("orbit"): Product(name=ProductName("orbit"), quantity=6, price=Decimal('2.3')),
}
with self.assertRaises(MachineOverloadedException):
machine.load_products(products)
def test_buy_a_product_if_cant_give_a_change(self):
machine = Machine(slots=3, slot_depth=10)
products = {
ProductName("coca-cola"): Product(name=ProductName("coca-cola"), quantity=1, price=Decimal('2.1')),
ProductName("mars"): Product(name=ProductName("mars"), quantity=5, price=Decimal('1.9')),
ProductName("orbit"): Product(name=ProductName("orbit"), quantity=6, price=Decimal('2.3')),
}
machine.load_products(products)
money = Coins({Decimal(5): 1})
slot_code, _ = machine.get_available_products()[ProductName("coca-cola")]
product, change = machine.choose_product(slot_code, money)
self.assertIsNone(product)
self.assertEqual(money, change)
self.assertEqual(machine.get_balance(), Decimal(0))
def test_cant_get_product_for_too_little_money(self):
machine = Machine(slots=3, slot_depth=10)
products = {
ProductName("coca-cola"): Product(name=ProductName("coca-cola"), quantity=7, price=Decimal('2.1')),
ProductName("mars"): Product(name=ProductName("mars"), quantity=5, price=Decimal('1.9')),
ProductName("orbit"): Product(name=ProductName("orbit"), quantity=6, price=Decimal('2.3')),
}
too_little_money_to_buy_coke = Coins({Decimal('0.2'): 2})
machine.load_products(products)
slot_code, _ = machine.get_available_products()[ProductName("coca-cola")]
product, change = machine.choose_product(slot_code, too_little_money_to_buy_coke)
self.assertIsNone(product)
self.assertIsNotNone(change)
self.assertEqual(change, too_little_money_to_buy_coke)
def test_can_buy_available_product_and_different_change(self):
machine = Machine(slots=3, slot_depth=10)
products = {
ProductName("coca-cola"):
Product(name=ProductName("coca-cola"),
quantity=7,
price=Decimal(2.1)),
ProductName("mars"):
Product(name=ProductName("mars"), quantity=5, price=Decimal(1.9)),
ProductName("orbit"):
Product(name=ProductName("orbit"), quantity=6, price=Decimal(2.3)),
}
money = Coins({Decimal(2): 1, Decimal(0.5): 1})
machine.load_products(products)
machine.load_coins(Coins({Decimal(0.2): 2}))
slot_code, _ = machine.get_available_products()[ProductName(
"coca-cola")]
product, change = machine.choose_product(slot_code, money)
self.assertIsNotNone(product)
self.assertIsNotNone(change)
self.assertEqual(product.name, ProductName("coca-cola"))
self.assertEqual(change, Coins({Decimal(0.2): 2}))
def test_second_delivery_to_big(self):
machine = Machine(slots=5, slot_depth=10)
products = {
ProductName("coca-cola"):
Product(name=ProductName("coca-cola"),
quantity=7,
price=Decimal('2.1')),
ProductName("mars"):
Product(name=ProductName("mars"), quantity=5,
price=Decimal('1.9')),
ProductName("orbit"):
Product(name=ProductName("orbit"),
quantity=12,
price=Decimal('2.3')),
}
machine.load_products(products)
products_2 = {
ProductName("mars"):
Product(name=ProductName("mars"), quantity=5,
price=Decimal('1.9')),
ProductName("lays"):
Product(name=ProductName("lays"), quantity=8,
price=Decimal('2.3')),
ProductName("coca-cola"):
Product(name=ProductName("coca-cola"),
quantity=7,
price=Decimal('2.1')),
}
with self.assertRaises(MachineOverloadedException):
machine.load_products(products_2)