class TestPub(unittest.TestCase): def setUp(self): self.pub = Pub('The Prancing Pony', 100.00) self.drink = Drink("Beer", 5.00, 6) self.customer = Customer("TJ", 50.00, 21) def test_pub_has_name(self): pub_name = self.pub.name self.assertEqual('The Prancing Pony', pub_name) def test_pub_has_till(self): pub_till = self.pub.till self.assertEqual(100.00, pub_till) def test_drink_count(self): count = self.pub.drinks self.assertEqual(0, len(count)) def test_add_to_till(self): self.pub.add_to_till(self.drink) self.assertEqual(105.00, self.pub.till) def test_check_age(self): check = self.pub.check_age(self.customer) self.assertEqual('What can I get you?', check) def test_check_drunkenness_okay(self): check = self.pub.check_drunkenness(self.customer) self.assertEqual("Yeehaw", check) def test_check_drunkenness_too_drunk(self): self.customer.buy_drink(self.drink) check = self.pub.check_drunkenness(self.customer) self.assertEqual("You've had enough.", check)
class TestPub(unittest.TestCase): def setUp(self): drink_1 = Drink("beer", 5.00, 2, 120) drink_2 = Drink("vodka", 3.00, 3, 30) drink_3 = Drink("cocktail", 7.00, 8, 25) list_of_drinks = [drink_1, drink_2, drink_3] self.pub = Pub("The Prancing Pony", 100.00, list_of_drinks) def test_pub_name(self): self.assertEqual("The Prancing Pony", self.pub.name) def test_pub_till(self): self.assertEqual(100.00, self.pub.till) def test_drink_availability(self): drink_1 = Drink("beer", 5.00, 2, 1) self.assertEqual(True, self.pub.drink_available(drink_1)) def test_pub_drinks_number(self): self.assertEqual(3, len(self.pub.drinks)) def test_age_challange(self): customer_2 = Customer("Ed", 6.00, 16) self.assertEqual(False, self.pub.age_challange(customer_2)) def test_drunkenness(self): customer_1 = Customer("Mark", 100.00, 33) drink_1 = Drink("beer", 5.00, 6, 1) customer_1.make_drunk(drink_1) customer_1.make_drunk(drink_1) self.assertEqual(True, self.pub.check_drunkenness(customer_1)) def test_add_money_to_till(self): drink_1 = Drink("beer", 5.00, 2, 1) self.pub.add_money_to_till(self.pub, drink_1) self.assertEqual(105.00, self.pub.till) def test_stock_count(self): self.assertEqual(865, self.pub.stock_count()) def test_remove_item_from_stock(self): drink_1 = Drink("beer", 5.00, 2, 1) self.pub.remove_item_from_stock(drink_1) self.assertEqual(119, self.pub.check_quantity(drink_1))
class TestPub(unittest.TestCase): def setUp(self): self.pub = Pub("The Prancing Pony", 100.00) self.drink1 = Drink("Gin and Tonic", 2.50, 3, 10) self.drink2 = Drink("Rum and Coke", 2.50, 2, 15) self.drink3 = Drink("Brandy", 4.00, 5, 8) def test_pub_has_name(self): self.assertEqual("The Prancing Pony", self.pub.name) def test_pub_till_value(self): self.assertEqual(100.00, self.pub.till) def test_pub_drinks_collection(self): self.pub.drinks_stock = ["Gin and Tonic", "Rum and Coke", "Brandy"] self.assertEqual(3, len(self.pub.drinks_stock)) def test_serve_drink(self): drink = Drink("Brandy", 4.00, 5, 8) self.pub.serve_drink(drink) self.assertEqual(104.00, self.pub.till) def test_check_age__true(self): customer = Customer("Frodo Baggins", 100.00, 50, 0) self.pub.check_age(customer) self.assertEqual(True, self.pub.check_age(customer)) def test_check_age__false(self): customer = Customer("Peregrin Took", 80.00, 28, 0) self.pub.check_age(customer) self.assertEqual(False, self.pub.check_age(customer)) def test_check_drunkenness__true(self): customer = Customer("Frodo Baggins", 100.00, 50, 0) self.pub.check_drunkenness(customer) self.assertEqual(True, self.pub.check_drunkenness(customer)) def test_check_drunkenness__false(self): customer = Customer("Peregrin Took", 80.00, 28, 21) self.pub.check_drunkenness(customer) self.assertEqual(False, self.pub.check_drunkenness(customer)) def test_check_stock_level(self): drink1 = Drink("Gin and Tonic", 2.50, 3, 10) self.assertEqual(10, drink1.stock_level) def test_check_total_stock_level(self): self.drinks_stock = [self.drink1, self.drink2, self.drink3] self.pub.check_total_stock_level(self.drinks_stock) self.assertEqual(33, self.pub.check_total_stock_level(self.drinks_stock)) def test_check_total_stock_price(self): self.drinks_stock = [self.drink1, self.drink2, self.drink3] self.pub.check_total_stock_price(self.drinks_stock) self.assertEqual(94.50, self.pub.check_total_stock_price(self.drinks_stock)) def test_serve_food(self): food = Food("Second Breakfast", 1.50, 2) self.pub.serve_food(food) self.assertEqual(101.50, self.pub.till) def test_at_the_bar__true(self): drink = self.drink1 customer = Customer("Peregrin Took", 80.00, 31, 19) self.pub.at_the_bar(drink, customer) self.assertEqual(102.50, self.pub.till) def test_at_the_bar__false(self): drink = self.drink1 customer = Customer("Peregrin Took", 80.00, 28, 21) self.pub.at_the_bar(drink, customer) self.assertEqual(100.00, self.pub.till)