def test_create_guest_list(self):
luke = Guest('Luke', LUKE_STATS)
guest_list = GuestList()
self.assertFalse(guest_list.has_guest('Luke'))
guest_list.add_guest(luke)
self.assertTrue(guest_list.has_guest('Luke'))
self.assertEqual(guest_list.get_guest_status('Luke'), 'available')
self.assertEqual(guest_list.get_free_guests().names()[0], 'Luke')
def test_move_guests_around(self):
luke = Guest('Luke', LUKE_STATS)
kota = Guest('Kota', KOTA_STATS)
guest_list = GuestList()
guest_list.add_guest(luke)
guest_list.add_guest(kota)
self.assertEqual(guest_list.get_guest_status('Luke'), 'available')
self.assertEqual(guest_list.get_guest_status('Kota'), 'available')
self.assertTrue('Luke' in guest_list.get_free_guests().names())
guest_list.send_guest_to_rd('Luke')
self.assertFalse('Luke' in guest_list.get_free_guests().names())
self.assertTrue('Luke' in guest_list.get_rd_team().names())
guest_list.send_guest_to_exploration('Kota')
self.assertFalse('Kota' in guest_list.get_free_guests().names())
self.assertTrue('Kota' in guest_list.get_exploration_team().names())
guest_list.free_guest('Luke')
self.assertTrue('Luke' in guest_list.get_free_guests().names())
self.assertFalse('Luke' in guest_list.get_rd_team().names())