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)