def test_customer_is_refused_service_underage(self):
pub = Pub("The Prancing Pony", 100.00)
drink = Drink("Vodka Martini", 6.50, 3)
self.customer = Customer("Emily Cullen", 20.00, 17)
pub.add_drink_to_list(drink)
self.customer.buy_drink(drink, pub)
self.assertEqual(100, pub.till)
def test_customer_drunkenness_doesnt_go_below_zero(self):
pub = Pub("The Prancing Pony", 100.00)
drink = Drink("Vodka Martini", 6.50, 3)
food = Food("Sunday Roast", 8.20, 8)
pub.add_drink_to_list(drink)
pub.add_food_to_list(food)
self.customer.buy_drink(drink, pub)
self.customer.buy_food(food, pub)
self.assertEqual(0, self.customer.drunkenness)
def test_customer_drunkenness_goes_down_with_food(self):
pub = Pub("The Prancing Pony", 100.00)
drink = Drink("Vodka Martini", 6.50, 3)
food = Food("Pizza", 7.20, 3)
pub.add_drink_to_list(drink)
pub.add_food_to_list(food)
self.customer.buy_drink(drink, pub)
self.customer.buy_drink(drink, pub)
self.customer.buy_food(food, pub)
self.assertEqual(3, self.customer.drunkenness)
def test_customer_refused_service_too_drunk(self):
pub = Pub("The Prancing Pony", 100.00)
drink = Drink("Vodka Martini", 6.50, 3)
pub.add_drink_to_list(drink)
self.customer.buy_drink(drink, pub)
self.customer.buy_drink(drink, pub)
self.customer.buy_drink(drink, pub)
self.customer.buy_drink(drink, pub)
self.customer.buy_drink(drink, pub)
self.customer.buy_drink(drink, pub)
self.assertEqual(117.50, self.customer.wallet)
class TestPub(unittest.TestCase):
def setUp(self):
self.pub = Pub("The Prancing Pony", 100.00)
def test_pub_has_name(self):
self.assertEqual("The Prancing Pony", self.pub.name)
def test_till_has_money(self):
self.assertEqual(100.00, self.pub.till)
def test_add_drink_to_list(self):
drink = Drink("Vodka Martini", 6.50, 3)
self.pub.add_drink_to_list(drink)
self.assertEqual(1, self.pub.drinks_list_count())
def test_add_food_to_list(self):
food = Food("Mac and cheese", 5.50, 2)
self.pub.add_food_to_list(food)
self.assertEqual(1, self.pub.food_list_count())
def test_stock_count_works(self):
drink = Drink("Vodka Martini", 6.50, 3)
self.pub.add_drink_to_list(drink)
self.pub.add_drink_to_list(drink)
self.pub.add_drink_to_list(drink)
self.assertEqual(3, self.pub.drinks[drink])
def test_stock_value(self):
drink = Drink("Vodka Martini", 6.50, 3)
self.pub.add_drink_to_list(drink)
self.pub.add_drink_to_list(drink)
self.assertEqual(13.00, self.pub.stock_value())
def test_customer_drunkenness_goes_up(self):
pub = Pub("The Prancing Pony", 100.00)
drink = Drink("Vodka Martini", 6.50, 3)
pub.add_drink_to_list(drink)
self.customer.buy_drink(drink, pub)
self.assertEqual(3, self.customer.drunkenness)
def test_customer_can_buy_drink_add_money_to_pub(self):
pub = Pub("The Prancing Pony", 100.00)
drink = Drink("Vodka Martini", 6.50, 3)
pub.add_drink_to_list(drink)
self.customer.buy_drink(drink, pub)
self.assertEqual(106.50, pub.till)
