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): self.pub = Pub("The Prancing Pony", 100) self.customer_1 = Customer("Ben", 10, 18, 0) self.customer_2 = Customer("Adam", 1, 12, 0) self.drink = Drink("Tennents", 2, 1) def test_pub_has_name(self): self.assertEqual("The Prancing Pony", self.pub.pub_name) #@unittest.skip("Delete this line to run the test") def test_pub_has_money(self): self.assertEqual(100, self.pub.till) #@unittest.skip("Delete this line to run the test") def test_pub_has_drinks(self): self.assertEqual(0, self.pub.stock_count()) #@unittest.skip("Delete this line to run the test") def test_customer_has_money(self): self.assertEqual(True, self.pub.check_customer_wallet(self.customer_1, self.drink)) #@unittest.skip("Delete this line to run the test" def test_customer_has_money(self): self.assertEqual(False, self.pub.check_customer_wallet(self.customer_2, self.drink)) def test_customer_money_reduced(self): self.assertEqual(8, self.pub.take_customer_money(self.customer_1, self.drink)) def test_add_money_to_till(self): self.assertEqual(102, self.pub.add_to_till(self.customer_1, self.drink)) def test_customer_id(self): self.assertEqual(True, self.pub.age_id(self.customer_1)) def test_customer_id(self): self.assertEqual(False, self.pub.age_id(self.customer_2)) def test_customer_drunk(self): self.assertEqual(1, self.pub.get_customer_drunk(self.customer_1, self.drink))