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())
class TestRoom(unittest.TestCase): def setUp(self): self.room_1 = Room("CBGB", 25.00) self.song_1 = Song("Juice", "Lizzo") self.song_2 = Song("Dr Dre", "Forgot About Dre") self.song_3 = Song("David Byrne", "Finite=Alright") self.song_4 = Song("Portishead", "Glory Box") self.guest_1 = Guest("Mick", "Jagger", 20, 12000.00, "Gimme Shelter") self.guest_2 = Guest("Justin", "Bieber", 16, 0.82, "Sorry") self.guest_3 = Guest("Joni", "Mitchell", 21, 100.00, "Juice") self.guest_4 = Guest("Susan", "Boyle", 47, 36.42, "F**k tha Police") self.guest_5 = Guest("Courtney", "Love", 31, 666.66, "Getting Away With It") def test_room_has_name(self): self.assertEqual("CBGB", self.room_1.get_room_name()) def test_room_name_change(self): self.room_1.change_room_name("Bongo Club") self.assertEqual("Bongo Club", self.room_1.get_room_name()) def test_room_has_song_list(self): self.assertEqual([], self.room_1.get_song_list()) def test_song_present_in_list(self): self.song_present_in_list() self.assertEqual(True, self.room_1.song_present_in_list()) def test_initial_song_list_count_is_zero(self): self.room_1.song_count() self.assertEqual(0, self.room_1.song_count()) def test_add_song_to_list(self): self.room_1.add_song_to_list(self.song_3) self.assertEqual(1, self.room_1.song_count()) def test_add_song_to_list__multiple_songs(self): self.room_1.add_song_to_list(self.song_3) self.room_1.add_song_to_list(self.song_2) self.room_1.add_song_to_list(self.song_1) self.assertEqual(3, self.room_1.song_count()) def test_remove_song_from_list(self): self.room_1.song_list = [ self.song_1, self.song_3, self.song_2, self.song_4 ] self.room_1.remove_song_from_list(self.song_4) self.assertEqual(3, self.room_1.song_count()) def test_remove_song_from_list__multiple(self): self.room_1.song_list = [ self.song_1, self.song_3, self.song_2, self.song_4 ] self.room_1.remove_song_from_list(self.song_1) self.room_1.remove_song_from_list(self.song_2) self.room_1.remove_song_from_list(self.song_4) self.assertEqual(1, self.room_1.song_count()) def test_clear_song_list(self): self.room_1.clear_song_list() self.assertEqual(0, self.room_1.song_count()) def test_room_has_guest_list(self): self.assertEqual([], self.room_1.get_guest_list()) def test_check_in_guest(self): self.room_1.check_in_guest(self.guest_1) self.assertEqual(1, self.room_1.guest_count()) def test_check_in_guest__multiple(self): self.room_1.check_in_guest(self.guest_1) self.room_1.check_in_guest(self.guest_3) self.room_1.check_in_guest(self.guest_4) self.assertEqual(3, self.room_1.guest_count()) def test_check_in_room_full(self): self.room_1.guest_list = [ self.guest_1, self.guest_3, self.guest_2, self.guest_4 ] self.room_1.check_in_guest(self.guest_5) self.assertEqual("GO AWAY COURTNEY, YOU KILLED KURT!!!", self.room_1.check_in_guest(self.guest_5)) self.assertEqual(4, self.room_1.guest_count()) def test_check_out_guest(self): self.room_1.guest_list = [ self.guest_1, self.guest_3, self.guest_2, self.guest_4 ] self.room_1.check_out_guest(self.guest_2) self.assertEqual(3, self.room_1.guest_count()) def test_check_out_guest__multiple(self): self.room_1.guest_list = [ self.guest_1, self.guest_3, self.guest_2, self.guest_4 ] self.room_1.check_out_guest(self.guest_2) self.room_1.check_out_guest(self.guest_4) self.assertEqual(2, self.room_1.guest_count()) def test_clear_room(self): self.room_1.clear_room() self.assertEqual(0, self.room_1.guest_count()) def test_add_cash_to_till(self): self.room_1.add_cash_to_till(100) self.assertEqual(200, self.room_1.till_count()) def test_charge_entry_fee(self): self.room_1.charge_entry_fee(self.guest_3) self.assertEqual(75.00, self.guest_3.wallet_cash_count()) self.assertEqual(125.00, self.room_1.till_count())