def test_machine_can_rearrange(self):
machine = Machine(slots=4, slot_depth=10)
first_delivery = {
ProductName("coca-cola"):
Product(name=ProductName("coca-cola"),
quantity=11,
price=Decimal('2.1')),
ProductName("mars"):
Product(name=ProductName("mars"),
quantity=15,
price=Decimal('1.9')),
}
machine.load_products(first_delivery)
cola_slot_code, _ = machine.get_available_products()[ProductName(
'coca-cola')]
_, _ = machine.choose_product(cola_slot_code,
Coins({Decimal('2.1'): 1}))
for _ in range(6):
mars_slot_code, _ = machine.get_available_products()[ProductName(
'mars')]
_, _ = machine.choose_product(mars_slot_code,
Coins({Decimal('1.9'): 1}))
second_delivery = {
ProductName("orbit"):
Product(name=ProductName("orbit"),
quantity=15,
price=Decimal('2.3')),
}
machine.load_products(second_delivery)
self.assertIn(ProductName("orbit"),
machine.get_available_products().keys())
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_loading_products_twice(self):
machine = Machine(slots=6, 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')),
}
machine.load_products(products_2)
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.assertIn(ProductName("lays"), available_products)
def test_can_give_smart_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('1.4')),
}
machine.load_products(products)
money = Coins({Decimal('0.2'): 3, Decimal('0.5'): 1})
machine.load_coins(money)
slot_code, _ = machine.get_available_products()[ProductName("orbit")]
product, change = machine.choose_product(slot_code,
Coins({Decimal(2): 1}))
self.assertIsNotNone(product)
self.assertIsNotNone(change)
self.assertEqual(product.name, ProductName("orbit"))
self.assertEqual(change, Coins({Decimal('0.2'): 3}))
def test_one_product_can_take_many_slots(self):
machine = Machine(slots=2, slot_depth=10)
products = {
ProductName("coca-cola"): Product(name=ProductName("coca-cola"), quantity=20, price=Decimal('2.1')),
}
machine.load_products(products)
available_products = machine.get_available_products().keys()
self.assertIn(ProductName("coca-cola"), available_products)
self.assertNotIn(ProductName("mars"), available_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_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_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_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"))
