def test_increase_till(self):
drinks = Drinks("Wine", 3.00, 1)
self.pub.increase_till(drinks)
self.assertEqual(103.00, self.pub.till)
def test_buy_drink(self):
drink = Drinks("Wine", 3.00, 1)
self.customer.buy_drink(drink)
self.assertEqual(47.00, self.customer.wallet)
def test_drink_units(self):
drink = Drinks("Wine", 3.00, 1)
self.customer.add_drink_units(drink)
self.assertEqual(1, self.customer.drunkeness)
def test_sell_drink_to_customer_till(self):
drink_1 = Drinks("Stella", 3.50)
customer = Customer("Frodo Baggins", 50, 30.00)
self.pub.add_drink_to_menu(drink_1)
self.pub.sell_drink_to_customer(customer, "Stella")
self.assertEqual(103.50, self.pub.till)
def setUp(self):
self.drinks = Drinks('Beer', 5.00, 2.8)
def test_sell_drink_to_customer_wallet(self):
drink_1 = Drinks("Stella", 3.50)
customer = Customer("Frodo Baggins", 50, 30.00)
self.pub.add_drink_to_menu(drink_1)
self.pub.sell_drink_to_customer(customer, "Stella")
self.assertEqual(26.50, customer.wallet)
def test_get_drink_price(self):
drink_1 = Drinks("Stella", 3.50)
self.pub.add_drink_to_menu(drink_1)
price = self.pub.get_drink_price("Stella")
self.assertEqual(3.50, price)
def test_pub_has_drinks_list(self):
drink_1 = Drinks("Stella", 3.50)
self.pub.add_drink_to_menu(drink_1)
self.assertEqual(1, self.pub.menu_length())
def setUp(self):
self.drinks = Drinks("Lager", 2.50, 2)
def setUp(self):
self.customer = Customer('Colin', 8.50, 27)
self.drink = Drinks('Wine', 6.50, 4)
self.food = Food('Kebab', 6.00, 4)