class TestPub(unittest.TestCase): def setUp(self): self.pub = Pub("Duke's Corner", 100.00) self.drink_1 = Drink("Tennents", 3.40, 3.5) self.drink_2 = Drink("Wine", 7.50, 12) self.customer_1 = Customer("Harrison", 22, 10.00) self.pub.add_drink_to_pub(self.drink_1) self.pub.add_drink_to_pub(self.drink_2) def test_pub_has_name(self): self.assertEqual("Duke's Corner", self.pub.name) def test_pub_has_money(self): self.assertEqual(100.00, self.pub.till) def test_pub_has_drinks_list(self): self.assertEqual(type([]),type(self.pub.drinks)) def test_pub_has_drinks(self): self.assertEqual(2, len(self.pub.drinks)) def test_add_money_to_till(self): self.pub.add_money_to_till(self.drink_2.price) self.assertEqual(107.50, self.pub.till) def test_remove_drink_from_pub(self): self.pub.remove_drink_from_pub(self.drink_1) self.assertEqual(1, len(self.pub.drinks)) def test_sell_drink(self): self.pub.sell_drink(self.drink_1, self.drink_1.price, self.customer_1.age) self.assertEqual(1, len(self.pub.drinks)) self.assertEqual(103.40, self.pub.till) def test_age_check(self): result = self.pub.age_check(self.customer_1.age) self.assertEqual("OK to serve", result)
class TestPub(unittest.TestCase): def setUp(self): self.pub = Pub("Ox", 100.00) self.drink_1 = Drink("Tennents", 3.40) self.drink_2 = Drink("Wine", 7.50) def test_pub_has_name(self): self.assertEqual("Ox", self.pub.name) def test_pub_has_money(self): self.assertEqual(100.00, self.pub.till) def test_pub_has_empty_drinks_list(self): self.assertEqual(0, len(self.pub.drinks)) def test_pub_has_drinks(self): self.pub.add_drink_to_pub(self.drink_1) self.pub.add_drink_to_pub(self.drink_2) self.assertEqual(2, len(self.pub.drinks)) def test_add_money_to_till(self): self.pub.add_money_to_till(self.drink_2.price) self.assertEqual(107.50, self.pub.till) def test_remove_drink_from_pub(self): self.pub.add_drink_to_pub(self.drink_1) self.pub.add_drink_to_pub(self.drink_2) self.pub.remove_drink_from_pub(self.drink_1) self.assertEqual(1, len(self.pub.drinks)) def test_sell_drink(self): self.pub.add_drink_to_pub(self.drink_1) self.pub.add_drink_to_pub(self.drink_2) self.pub.sell_drink(self.drink_1, self.drink_1.price) self.assertEqual(1, len(self.pub.drinks)) self.assertEqual(103.40, self.pub.till)