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))
