def test_knockback__true(self):
customer_1 = Customer("Al McWhiggin", 40, 21)
drink_2 = Drink("Death_Drink", 3, 11)
self.pub.sell_drink(customer_1, drink_2)
self.pub.sell_drink(customer_1, drink_2)
self.assertEqual(37, customer_1.wallet)
self.assertEqual(203, self.pub.till)
self.assertEqual(11, customer_1.drunkenness)
def test_sell_food(self):
customer_1 = Customer("Al McWhiggin", 40, 21)
drink_2 = Drink("Death_Drink", 3, 11)
food_1 = Food("Chicken", 5, 3)
self.pub.sell_drink(customer_1, drink_2)
self.pub.sell_food(customer_1, food_1)
self.assertEqual(32, customer_1.wallet)
self.assertEqual(208, self.pub.till)
self.assertEqual(8, customer_1.drunkenness)
class TestCustomer(unittest.TestCase):
def setUp(self):
self.customer = Customer("Al McWhiggin", 40, 21)
def test_customer_has_name(self):
self.assertEqual("Al McWhiggin", self.customer.name)
def test_customer_has_wallet(self):
self.assertEqual(40, self.customer.wallet)
def test_customer_has_age(self):
self.assertEqual(21, self.customer.age)
def test_customer_drunkenness_level(self):
self.assertEqual(0, self.customer.drunkenness)
def test_increase_drunkenness(self):
drink_1 = Drink("Guinness", 3, 2)
self.customer.increase_drunkenness(drink_1)
self.assertEqual(2, self.customer.drunkenness)
def test_decrease_drunkenness(self):
drink_1 = Drink("Guinness", 3, 2)
food_1 = Food("Chicken", 5, 3)
self.customer.increase_drunkenness(drink_1)
self.customer.increase_drunkenness(drink_1)
self.customer.decrease_drunkenness(food_1)
self.assertEqual(1, self.customer.drunkenness)
def setUp(self):
self.customer = Customer("Al McWhiggin", 40, 21)
def test_make_customer_drunk(self):
customer_1 = Customer("Al McWhiggin", 40, 21)
drink_1 = Drink("Guinness", 3, 2)
self.pub.sell_drink(customer_1, drink_1)
self.assertEqual(2, customer_1.drunkenness)
def test_pub_can_sell_drink__under_18(self):
customer_2 = Customer("Andy", 500, 14)
drink_1 = Drink("Guinness", 3, 2)
self.pub.sell_drink(customer_2, drink_1)
self.assertEqual(500, customer_2.wallet)
self.assertEqual(200, self.pub.till)
def test_pub_can_sell_drink__over_18(self):
customer_1 = Customer("Al McWhiggin", 40, 21)
drink_1 = Drink("Guinness", 3, 2)
self.pub.sell_drink(customer_1, drink_1)
self.assertEqual(37, customer_1.wallet)
self.assertEqual(203, self.pub.till)