class TestGuest(unittest.TestCase):
def setUp(self):
self.thunderstruck = Song("Thunderstruck", "AC/DC", 292, 1990, "Rock")
self.guest = Guest("Dave", 50, self.thunderstruck)
# testing Guest attributes
def test_guest_has_name(self):
self.assertEqual("Dave", self.guest.name)
def test_guest_has_favourite_song(self):
self.assertEqual(self.thunderstruck, self.guest.favourite_song)
def test_guest_has_wallet(self):
self.assertEqual(50, self.guest.wallet)
# test methods
def test_guest_pays_for_item(self):
self.guest.pay(5)
self.assertEqual(45, self.guest.wallet)
def test_favourite_song_is_on_playlist(self):
playlist = [self.thunderstruck]
self.assertEqual("YAAAASSSS!!!!, they've got Thunderstruck!",
self.guest.woop_obnoxiously(playlist))
def test_guest_can_afford_item(self):
self.assertTrue(self.guest.can_afford_item(10))
def test_guest_cannot_afford_item(self):
self.assertFalse(self.guest.can_afford_item(50.01))
class TestGuest(unittest.TestCase):
def setUp(self):
self.guest_loud_bob = Guest("Loud Bob", 23, 200.00, "Song 2", 100.00)
self.guest_shy_bob = Guest("Shy Bob", 18, 1000.00, "Three Blind Mice",
0.00)
def test_guest_exists(self):
self.assertEqual("Loud Bob", self.guest_loud_bob.name)
self.assertEqual(23, self.guest_loud_bob.age)
self.assertEqual(200.00, self.guest_loud_bob.wallet)
def test_guest_pays_for_room(self):
self.guest_loud_bob.pay(15.00)
self.assertEqual(185.00, self.guest_loud_bob.wallet)
def test_customer_has_money_to_pay__True(self):
entry_fee = 15.00
self.assertEqual(True, self.guest_loud_bob.check_can_pay(entry_fee))
def test_customer_has_money_to_pay__False(self):
entry_fee = 201.00
self.assertEqual(False, self.guest_loud_bob.check_can_pay(entry_fee))
def test_guest_has_favourite_song(self):
self.assertEqual("Song 2", self.guest_loud_bob.fav_song)
def test_guest_can_pay_drink(self):
test_drink = Drink("Blue Lagoon", 15.00, 4.00)
self.guest_loud_bob.pay(test_drink.price)
self.assertEqual(185.00, self.guest_loud_bob.wallet)
def test_guest_can_increase_drunkeness(self):
test_drink = Drink("Blue Lagoon", 15.00, 4.00)
self.guest_loud_bob.increase_drunkeness(test_drink.alcohol_units)
self.assertEqual(104.00, self.guest_loud_bob.drunkeness)