class TestRoom(unittest.TestCase): def setUp(self): self.song_1 = Song("Juicy", "The Notorious B.I.G", "Hip Hop") self.song_2 = Song("Club Tropicana", "Wham", "Classics") self.song_3 = Song("Thunderstruck", "ACDC", "Rock") self.guest_1 = Guest("Rory", 50) self.guest_2 = Guest("Steve", 40) self.guest_3 = Guest("Ally", 60) self.guest_4 = Guest("Stuart", 30) self.guest_5 = Guest("Liz", 50) self.guest_6 = Guest("Rosie", 40) self.room_1 = Room("Hip Hop Room", self.song_1.title, self.song_1.artist, 6, 5, 100) self.room_2 = Room("Classics Room", self.song_2.title, self.song_2.artist, 6, 3, 150) self.room_3 = Room("Rock Room", self.song_3.title, self.song_3.artist, 6, 4, 200) def test_room_has_name(self): self.assertEqual("Hip Hop Room", self.room_1.name) self.assertEqual("Classics Room", self.room_2.name) self.assertEqual("Rock Room", self.room_3.name) def test_room_has_fee(self): self.assertEqual(5, self.room_1.fee) self.assertEqual(3, self.room_2.fee) self.assertEqual(4, self.room_3.fee) def test_room_has_capacity(self): self.assertEqual(6, self.room_1.capacity) self.assertEqual(6, self.room_2.capacity) self.assertEqual(6, self.room_3.capacity) def test_room_has_till(self): self.assertEqual(100, self.room_1.till) self.assertEqual(150, self.room_2.till) self.assertEqual(200, self.room_3.till) def test_room_has_song_played_title(self): self.assertEqual("Juicy", self.room_1.song_played_title) self.assertEqual("Club Tropicana", self.room_2.song_played_title) self.assertEqual("Thunderstruck", self.room_3.song_played_title) def test_room_has_song_played_artist(self): self.assertEqual("The Notorious B.I.G", self.room_1.song_played_artist) self.assertEqual("Wham", self.room_2.song_played_artist) self.assertEqual("ACDC", self.room_3.song_played_artist) def test_room_has_no_guests(self): self.assertEqual(0, self.room_1.guest_count()) def test_room_can_check_in_guests(self): self.room_1.check_in_guest(self.guest_1) self.assertEqual("Rory", self.room_1.guest_list[0].name) self.assertEqual(1, self.room_1.guest_count()) def test_room_can_check_out_guests(self): self.room_1.check_in_guest(self.guest_1) self.room_1.check_out_guest(self.guest_1) self.assertEqual(0, self.room_1.guest_count()) def test_play_list_has_no_songs(self): self.assertEqual(0, self.room_1.song_count()) def test_add_song_to_room(self): self.room_1.add_song_to_play_list(self.song_1) self.assertEqual("Juicy", self.room_1.play_list[0].title) self.assertEqual(1, self.room_1.song_count())