class TestCreateRoom(unittest.TestCase): def setUp(self): self.amity = Amity() #test_if_can_add_more than one room #test_if_duplicate_entries_not_added def test_whether_office_already_exists(self): create1 = self.amity.create_room("O", "asgard") self.assertEqual("Room already exists", create1, msg="Duplicate room is being created") def test_if_office_created(self): # print (self.amity.offices[0].current_occupants) self.assertEqual( self.amity.create_room("O", "cave"), "We have successfully created a new office called: CAVE", msg="Office CANNOT be created successfully") def test_whether_livingspace_already_exists(self): create1 = self.amity.create_room("L", "statehouse") self.assertEqual("Room already exists", create1, msg="Duplicate room is being created") def test_if_livingspace_created(self): self.assertEqual(self.amity.create_room("L", "cave"), "New living quarters ( CAVE ) successfully created!", msg="Office CANNOT be created successfully")
class TestAddPerson(unittest.TestCase): def setUp(self): self.amity = Amity() def test_whether_fellow_already_exists(self): create = self.amity.add_person("WONDERWOMAN", "FELLOW", "Y") self.assertEqual("Person already exists", create, msg="Duplicate person being created") #test_if_person_identifier_exists ## what if 2 people have the same name? def test_if_fellow_added(self): room = self.amity.create_room("o", "oculus") ls = self.amity.create_room("l", "london") count = len(self.amity.fellows) create = self.amity.add_person("WOLVERINE", "FELLOW", "Y") self.assertEqual(len(self.amity.fellows), count + 1, msg="Fellow CANNOT be added successfully") def test_whether_staff_already_exists(self): create = self.amity.add_person("STRANGE", "STAFF", "N") self.assertEqual("Person already exists", create, msg="Duplicate person being created") def test_if_staff_added(self): room = self.amity.create_room("o", "oculus") count = len(self.amity.staff) self.amity.add_person("CYBORG", "STAFF", "N") import pdb pdb.set_trace() self.assertEqual(len(self.amity.staff), count + 1, msg="Staff CANNOT be added successfully") def test_if_randomly_allocated(self): person = self.amity.add_person("CYBORG", "STAFF", "N") room = self.amity.create_room("o", "Mombasa") self.assertTrue(person in room.current_occupants)