class TestCustomer(unittest.TestCase):
def setUp(self):
self.customer = Customer("Gordon", 100, 25)
self.customer_2 = Customer("Lina", 100, 17)
self.pub = Pub("The Prancing Pony", 100)
self.drink = Drink("Tennents Lager", 5, 10)
def test_customer_has_name(self):
self.assertEqual("Gordon", self.customer.name)
def test_customer_has_wallet(self):
self.assertEqual(100, self.customer.wallet)
def test_customer_can_buy_drink(self):
self.customer.customer_buys_drink(self.drink, self.pub)
self.assertEqual(95, self.customer.wallet)
self.assertEqual(105, self.pub.till)
def test_customer_over_18(self):
self.assertEqual(True, self.pub.check_customer_age(self.customer.age))
def test_customer_under_18(self):
self.assertEqual(
False, self.pub.check_customer_age(self.customer_2.age))
class TestCustomer(unittest.TestCase):
def setUp(self):
self.customer = Customer("customer name", 17, 3, 4, 5, 100)
self.customer2 = Customer("customer name", 18, 3, 4, 5, 100)
self.customer3 = Customer("customer name", 40, 3, 4, 5, 100)
## name, age, hunger, thirst, drunk_level
def test_customer_has_name(self):
self.assertEqual("customer name", self.customer.name)
def test_customer_has_age(self):
self.assertEqual(17, self.customer.age)
def test_customer_has_hunger(self):
self.assertEqual(3, self.customer.hunger)
def test_customer_has_thirst(self):
self.assertEqual(4, self.customer.thirst)
def test_customer_has_drunk_level(self):
self.assertEqual(5, self.customer.drunk_level)
def test_customer_has_naughty_level(self):
self.assertEqual(100, self.customer.naughty_level)
def test_customer_can_drink_alchohol(self):
self.assertEqual(False, self.customer.is_over_18())
def test_customer__can_drink_alchohol(self):
self.assertEqual(True, self.customer2.is_over_18())
def test_customer___can_drink_alchohol(self):
self.assertEqual(True, self.customer3.is_over_18())
class TestCustomer(unittest.TestCase):
def setUp(self):
self.customer_1 = Customer("David", 50.00)
self.drink_beer = Drink("Beer", 5.00)
def test_customer_has_name(self):
self.assertEqual("David", self.customer_1.name)
def test_customer_wallet_amount(self):
self.assertEqual(50.00, self.customer_1.wallet)
def test_customer_age(self):
self.assertEqual(20, self.customer_1.age)
def test_number_of_drinks_customer_has_in_hand__isnone(self):
self.assertEqual(0, len(self.customer_1.drinks_in_hand))
def test_number_of_drinks_customer_has_in_hand__added_drink(self):
self.customer_1.add_drink_to_customer_hand(self.drink_beer)
self.assertEqual(1, len(self.customer_1.drinks_in_hand))
def test_customer_has_paid_for_drink(self):
self.customer_1.alter_customer_wallet_amount(self.drink_beer.price)
self.assertEqual(45, self.customer_1.wallet)
def test_check_customer_age(self):
self.assertEqual(True, self.customer_1.check_customer_age)
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)
class TestCustomer(unittest.TestCase):
def setUp(self):
self.customer = Customer("Brian", 300, 28)
self.drink = Drink("Martini", 5, 4)
def test_customer_has_name(self):
expected = "Brian"
actual = self.customer.name
self.assertEqual(expected, actual)
def test_customer_has_wallet(self):
expected = 300
actual = self.customer.wallet
self.assertEqual(expected, actual)
def test_customer_can_buy_drink(self):
self.customer.buy_drink(self.drink)
self.assertEqual(295, self.customer.wallet)
def test_customer_has_age(self):
self.assertEqual(28, self.customer.age)
def test_customer_drunkenness_goes_up_by_drink(self):
expected = 4
self.customer.buy_drink(self.drink)
actual = self.customer.drunkenness
self.assertEqual(expected, actual)
def setUp(self):
self.pub = Pub("Ox", 100.00)
self.customer_1 = Customer("Billy", 50.00, 18)
self.customer_2 = Customer("Jimmy", 50.00, 16)
self.drink_1 = Drink("Beer", 5.0, 5.5)
self.drink_2 = Drink("Wine", 7.5, 15.0)
self.drink_3 = Drink("Spirit", 3.5, 40.0)
self.drink_4 = Drink("Cocktail", 10.0, 25.0)
def setUp(self):
self.pub = Pub("The prancing pony", 100)
self.drink = Drink("Beer", 2, 6)
self.food = Food("Burger", 3, 2)
self.customer = Customer("name", 10, 21)
self.underage_customer = Customer("name", 10, 16)
self.pub.add_stocklist(self.drink)
self.pub.add_food(self.food)
def test_can_sell_drink_to_customer(self):
drink = Drink("Tequilla Sunrise", 4.00, 5)
customer = Customer("Bob", 25.55)
customer.age = 22
self.pub.add_to_stock(drink)
self.pub.sell_drink_to_customer(customer, drink)
self.assertEqual(21.55, customer.wallet)
self.assertEqual(104.5, self.pub.till)
def setUp(self):
self.food = Food("Lasagne", 2.50, 5)
self.pub = Pub("The Prancing Pony", 100.00,
["Budweiser", "White Wine", "Red Wine", "Coke"])
self.drink = Drink("Budweiser", 3.75, 4)
self.customer = Customer("Joe Bloggs", 34, 10.00, 0)
self.customer_2 = Customer("Bob Smith", 17, 5.00, 0)
self.customer_3 = Customer("Bob Dylan", 55, 30.00, 15)
def setUp(self):
self.pub = Pub("The Prancing Pony", 100.00)
self.customer_1 = Customer("Mark", 10.00, 33, 0)
self.customer_2 = Customer("Andrew", 20.00, 17, 10)
self.customer_3 = Customer("Steven", 0.00, 37, 20)
self.drink_1 = Drink("Whisky", 5, 5, 10)
self.drink_2 = Drink("Punk IPA", 4.50, 1, 10)
self.food_1 = Food("Pie", 2.75, 5, 10)
self.food_2 = Food("Crisps", 0.99, 1, 0)
def setUp(self):
self.drink_1 = Drink("Guinness", 10, 2)
self.drink_2 = Drink("Gin", 15, 4)
self.drink_3 = Drink("Whisky", 20, 6)
self.BrewDog = Pub("Brewdog_edinburgh", 100,
[self.drink_1, self.drink_2, self.drink_3])
self.customer = Customer("Billy", 200, 45)
self.customer_2 = Customer("Jane", 5, 30)
self.customer_3 = Customer("Gerorge", 50, 16)
def setUp(self):
self.drink_1=Drink("beer", 3.00, 1, 10)
self.drink_2=Drink("red wine", 2.00, 2, 8)
self.drinks=[self.drink_1, self.drink_2]
self.pub = Pub("The Prancing Pony", 100.00, self.drinks)
self.customer_1 = Customer("Callum", 10.00, 24, 6)
self.customer_2 = Customer("John", 5.00, 16, 0)
self.customer_3 = Customer("James", 15.00, 30, 3)
def setUp(self):
self.drink_1 = Drink("Mojito", 8.0, 7)
self.drink_2 = Drink("Pilsen", 6.5, 4)
self.customer_1 = Customer("Malcolm", 25.0, 28)
self.customer_2 = Customer("Erika", 30.0, 17)
drinks = [self.drink_2, self.drink_1]
self.pub = Pub("JP's", drinks, 500)
class TestCustomer(unittest.TestCase):
def setUp(self):
self.customer = Customer("Michael Sinclair", 1000.00)
def test_does_customer_have_cash(self):
self.assertEqual(1000.00, self.customer.wallet)
def test_can_remove_money(self):
self.customer.remove_money(10.00)
self.assertEqual(990.00, self.customer.wallet)
def setUp(self):
self.pub = Pub("Ox", 100.00)
self.drink = Drink("Highland Park 18yr", 6.95, 8)
self.drink2 = Drink("Rum", 5.50, 4)
self.drink3 = Drink("Beer", 3.20, 2)
self.food = Food("Chips and gravy", 3.50, 100)
self.customer1 = Customer("Malcolm", 33, 25.00)
self.customer2 = Customer("Harrison", 15, 30.00)
self.customer3 = Customer("Dave", 21, 50.00)
class TestPub(unittest.TestCase):
def setUp(self):
self.pub = Pub("Ox", 100.00)
self.customer_1 = Customer("Billy", 50.00, 18)
self.customer_2 = Customer("Jimmy", 50.00, 16)
self.drink_1 = Drink("Beer", 5.0, 5.5)
self.drink_2 = Drink("Wine", 7.5, 15.0)
self.drink_3 = Drink("Spirit", 3.5, 40.0)
self.drink_4 = Drink("Cocktail", 10.0, 25.0)
def test_pub_has_name(self):
self.assertEqual("Ox", self.pub.name)
def test_pub_has_cash(self):
self.assertEqual(100.00, self.pub.cash)
def test_customer_has_name(self):
self.assertEqual("Billy", self.customer_1.name)
def test_customer_has_wallet(self):
self.assertEqual(50.00, self.customer_1.wallet)
def test_drink_has_name(self):
self.assertEqual("Beer", self.drink_1.name)
def test_drink_has_price(self):
self.assertEqual(5.00, self.drink_1.price)
def test_customer_buy_drink(self):
self.customer_1.buy_drink(self.drink_1)
self.assertEqual(45, self.customer_1.wallet)
def test_add_drink_to_menu(self):
self.pub.add_drink_to_menu(self.drink_1)
self.pub.add_drink_to_menu(self.drink_2)
self.pub.add_drink_to_menu(self.drink_3)
self.pub.add_drink_to_menu(self.drink_4)
self.assertEqual(4, len(self.pub.menu))
def test_add_to_cash(self):
self.pub.sell_drink(self.drink_1)
self.assertEqual(105.00, self.pub.cash)
def test_customer_has_age(self):
self.assertEqual(18, self.customer_1.age)
self.assertEqual(16, self.customer_2.age)
def test_age_check_adult(self):
self.assertEqual(True, self.pub.age_check(self.customer_1))
def test_age_check_adult(self):
self.assertEqual(False, self.pub.age_check(self.customer_2))
class TestCustomer(unittest.TestCase):
def setUp(self):
self.customer = Customer("Frodo Baggins", 50, 30.00)
def test_customer_has_name(self):
self.assertEqual("Frodo Baggins", self.customer.name)
def test_customer_has_wallet(self):
self.assertEqual(30.00, self.customer.wallet)
def test_reduce_wallet(self):
self.customer.reduce_wallet(10.00)
self.assertEqual(20.00, self.customer.wallet)
class TestCustomer(unittest.TestCase):
def setUp(self):
self.customer = Customer("Bill", 50)
def test_can_decrease_wallet(self):
drink = Drink("Beer", 3)
self.customer.decrease_wallet(drink)
self.assertEqual(47, self.customer.wallet)
def test_add_drink_to_customer(self):
drink = Drink("Beer", 3)
self.customer.add_drink_to_customer(drink)
self.assertEqual(1, len(self.customer.drinks))
def setUp(self):
stock = {
"drinks": [Drink("cosmo", 4.00, 3),
Drink("beer", 2.50, 2)],
"food": [Food("burger", 2.00, 10),
Food("hotdog", 2.00, 5)]
}
self.pub = Pub("The Prancing Pony", 100.00, stock)
self.customer = Customer("Bob", 50.00, 30)
self.drink = self.pub.stock["drinks"][0]
self.young_customer = Customer("Betty", 20.00, 16)
self.food = self.pub.stock["food"][1]
