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 TestLoadPeople(unittest.TestCase):
def setUp(self):
self.amity = Amity()
def test_file_created(self):
self.assertTrue(os.path.exists("file.txt")) #change filename
def test_if_file_is_not_already_open(self):
self.assertEqual(self.amity.load_people(),
"file is already opened",
msg="File not closed being reopened"
) #to correct for with parameter
def test_if_file_does_not_exists(self):
self.assertEqual(self.amity.load_people(), "file does not exist")
def test_whether_input_is_txt_file(self):
pass
def test_if_input_file_is_not_blank(self):
self.assertEqual(self.amity.load_people(),
"File may be empty or in incorrect format")
def test_if_input_file_in_correct_format(self):
self.assertEqual(self.amity.load_people(),
"File may be empty or in incorrect format")
class TestPrintRoom(unittest.TestCase):
def setUp(self):
self.room = Room("statehouse")
self.person = Person()
self.amity = Amity()
self.amity.reallocate_person(1, "statehouse")
def test_if_room_exists(self):
self.assertTrue(self.room.name in rooms, msg="Room does NOT exist")
def test_room_prints_all_members(self):
self.assertTrue(len(self.room.current_occupants) == 1,
msg="All members not printed 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)
def setUp(self):
self.amity = Amity()
def setUp(self):
amity = Amity()
amity.load_people(
) #refactor to accept an optional argument for filename
def setUp(self):
amity = Amity()
amity.save_state(
) #refactor to accept an optional argument for filename
def setUp(self):
self.room = Room("statehouse")
self.person = Person()
self.amity = Amity()
self.amity.reallocate_person(1, "statehouse")