class TestGuest(unittest.TestCase):
def setUp(self):
self.guest1 = Guest("John", 10.00, "Wannabe")
def test_guest_has_name(self):
self.assertEqual("John", self.guest1.name)
def test_guest_has_wallet(self):
self.assertEqual(10.00, self.guest1.wallet)
def test_guest_remove_cash(self):
self.guest1.remove_cash(5.00)
self.assertEqual(5.00, self.guest1.wallet)
def test_guest_has_favourite_song(self):
self.assertEqual("Wannabe", self.guest1.favourite_song)
def test_guest_can_cheer(self):
self.assertEqual("Woo!", self.guest1.cheer())
class TestGuest(unittest.TestCase):
def setUp(self):
self.song = Song("Castle Theme")
self.guest = Guest("Mario", self.song, 100.00)
self.room = Room("Nintendo")
def test_guest_has_name(self):
self.assertEqual("Mario", self.guest.name)
def test_guest_has_favourite_song(self):
self.assertEqual(self.song, self.guest.favourite_song)
def test_guest_has_wallet(self):
self.assertEqual(100.00, self.guest.wallet)
def test_pay_entry_fee_removes_money_from_wallet(self):
self.guest.pay_entry_fee(self.room)
self.assertEqual(90.00, self.guest.wallet)
def test_guest_can_cheer(self):
self.assertEqual("Woop!", self.guest.cheer())
class TestGuest(unittest.TestCase):
def setUp(self):
self.room_2 = Room("Studio 54", 45.00)
self.guest_1 = Guest("Mick", "Jagger", 20, 12000.00, "Dead Flowers")
self.guest_2 = Guest("Justin", "Bieber", 16, 5.00, "Sorry")
self.guest_3 = Guest("Joni", "Mitchell", 21, 100.00, "Juice")
def test_guest_has_whole_name(self):
self.assertEqual("Justin", self.guest_2.get_guest_first_name())
self.assertEqual("Bieber", self.guest_2.get_guest_surname())
def test_guest_has_age(self):
self.assertEqual(20, self.guest_1.get_guest_age())
def test_guest_has_funds(self):
self.assertEqual(100.00, self.guest_3.wallet_cash_count())
def test_remove_cash_from_wallet(self):
self.guest_1.remove_cash_from_wallet(self.room_2.entry_fee)
self.assertEqual(11955.00, self.guest_1.wallet_cash_count())
def test_cheer(self):
self.cheer()
self.assertEqual("CHOON!", self.guest_3.cheer())