def do_load_people(self, arg): '''Usage: load_people <filename>''' person = Person() status = person.load_people_data(arg['<filename>']) print(status)
class TestPerson(unittest.TestCase): def setUp(self): self.person = Person() sys.stdout = StringIO() def test_person_class_instance(self): self.assertIsInstance(self.person, Person) @patch.dict('app.room.Room.total_rooms', {'oculus': [], 'haskel': []}) @patch.dict('app.person.Person.total_people', {1: 'MigwiNdungu'}) @patch('app.room.Room.offices') @patch('app.room.Room.livingspaces') @patch('app.person.Person.fellows') @patch('app.person.Person.staff') def test_it_adds_a_person(self, mock_staff, mock_fellows, mock_livingspaces, mock_offices): mock_livingspaces.__iter__.return_value = ['haskel'] mock_offices.__iter__.return_value = ['oculus'] mock_fellows.__iter__.return_value = [] mock_staff.__iter__.return_value = [] already_present = self.person.add_person('Migwi', 'Ndungu', 'Fellow', 'Y') self.assertEqual( already_present, 'oops! Someone with the username MigwiNdungu already exists') no_accomodation = self.person.add_person('Ben', 'Kamau', 'Fellow', 'N') self.assertEqual(no_accomodation, 'BenKamau added to unallocated-people') staff_accomodation = self.person.add_person('Michelle', 'Korir', 'Staff', 'Y') self.assertEqual( staff_accomodation, 'NOTE: Staff members are not allocated livings paces') successfull_staff_add = self.person.add_person('Buck', 'Speed', 'Staff', 'N') self.assertEqual(successfull_staff_add, 'Staff member added successfully') @patch.dict( 'app.room.Room.total_rooms', { 'oculus': [1, 2, 5, 6, 9, 10], 'narnia': [3, 4, 7, 8, 11, 12], 'haskel': [1, 2, 3, 4] }) @patch.dict( 'app.person.Person.total_people', { 1: 'MigwiNdungu', 2: 'JosephMuli', 3: 'Josh', 4: 'Njira', 5: 'kevin', 6: 'mwangi', 7: 'john', 8: 'milkah', 9: 'noelah', 10: 'serah', 11: 'sila', 12: 'mary' }) @patch('app.room.Room.offices') @patch('app.room.Room.livingspaces') @patch('app.person.Person.fellows') @patch('app.person.Person.staff') def test_adding_to_fully_occupied_rooms(self, mock_staff, mock_fellows, mock_livingspaces, mock_offices): mock_livingspaces.__iter__.return_value = ['haskel'] mock_offices.__iter__.return_value = ['oculus'] mock_fellows.__iter__.return_value = [] mock_staff.__iter__.return_value = [] fully_occupied_office = self.person.add_person('Jimmy', 'Kamau', 'Staff', 'N') self.assertEqual( fully_occupied_office, 'all offices are currently fully occupied, adding to staff-unallocated-without-offices...' ) fully_occupied_ls = self.person.add_person('Alex', 'Magana', 'Fellow', 'Y') self.assertEqual( fully_occupied_ls, 'sorry all living spaces are currently fully occupied, adding to unallocated...' ) @patch.dict('app.room.Room.total_rooms', {'oculus': []}) @patch.dict('app.person.Person.total_people', {1: 'Migwi'}) @patch('app.room.Room.offices') @patch('app.room.Room.livingspaces') @patch('app.person.Person.fellows') @patch('app.person.Person.staff') def test_response_on_no_rooms(self, mock_staff, mock_fellows, mock_livingspaces, mock_offices): mock_offices.__iter__.return_value = ['oculus'] mock_livingspaces.__iter__.return_value = [] mock_fellows.__iter__.return_value = [] mock_staff.__iter__.return_value = [] msg = self.person.add_person('joan', 'ngugi', 'Fellow', 'Y') self.assertEqual(msg, "There are currently no livingspaces") @patch.dict( 'app.room.Room.total_rooms', { 'oculus': [1], 'green': [], 'mordor': [], 'shire': [23, 43, 52, 12, 32, 32], 'haskel': [], 'python': [1], 'ruby': [], 'blue': [98, 75, 45, 24, 76, 99] }) @patch.dict('app.person.Person.total_people', {1: 'Migwi', 2: 'jojo'}) @patch('app.room.Room.offices') @patch('app.room.Room.livingspaces') @patch('app.person.Person.fellows') @patch('app.person.Person.staff') def test_reallocation(self, mock_staff, mock_fellows, mock_livingspaces, mock_offices): mock_livingspaces.__iter__.return_value = [ 'haskel', 'python', 'ruby', 'blue' ] mock_offices.__iter__.return_value = ['oculus', 'mordor', 'shire'] mock_fellows.__iter__.return_value = ['Migwi'] mock_staff.__iter__.return_value = ['jojo'] fully_occupied = self.person.reallocate_person(1, 'shire') self.assertEqual(fully_occupied, "Sorry the office is occupied fully") already_present = self.person.reallocate_person(1, 'oculus') self.assertEqual( already_present, "The Person is already allocated in the requested room") msg = self.person.reallocate_person(1, 'mordor') self.assertEqual(msg, "Allocation to New office successfull!") person_msg = self.person.reallocate_person(3, 'mordor') self.assertEqual(person_msg, "The person ID does not exist!") room_msg = self.person.reallocate_person(1, 'shell') self.assertEqual(room_msg, "The room doesn't exist!") fellow_reallocate_livingspace = self.person.reallocate_person( 1, 'ruby') self.assertEqual(fellow_reallocate_livingspace, "Allocation to New livingSpace successful!") fully_occupied_ls = self.person.reallocate_person(1, 'blue') self.assertEqual(fully_occupied_ls, "Sorry the LivingSpace is currently fully occupied!") unallocated_room = self.person.reallocate_person(1, 'green') self.assertEqual(unallocated_room, "green was not Allocated") @patch('app.person.Person.add_person') def test_loads_people(self, mock_add_person): mock_add_person.return_value = '' sample_read_text = 'OLUWAFEMI SULE FELLOW Y' with patch("__builtin__.open", mock_open(read_data=sample_read_text)) as mock_file: self.person.load_people_data('list.txt') # assert open("people.txt", 'r').readlines() == sample_read_text mock_file.assert_called_with("list.txt") @patch.dict('app.room.Room.total_rooms', { 'oculus': [1], 'mordor': [2], 'python': [1] }) @patch.dict('app.person.Person.total_people', { 1: 'Migwi', 2: 'jojo', 3: 'jimbo' }) @patch('app.room.Room.offices') @patch('app.room.Room.livingspaces') @patch('app.person.Person.fellows') @patch('app.person.Person.staff') @patch('app.person.Person.unallocated_people') @patch('app.person.Person.fellows_not_allocated_office') @patch('app.person.Person.staff_not_allocated_office') def test_commit_people(self, mock_staff_not_allocated_office, mock_fellows_not_allocated_office, mock_unallocated_people, mock_staff, mock_fellows, mock_livingspaces, mock_offices): mock_livingspaces.__iter__.return_value = ['python'] mock_offices.__iter__.return_value = ['oculus', 'mordor'] mock_fellows.__iter__.return_value = ['Migwi'] mock_staff.__iter__.return_value = ['jojo', 'jimbo'] mock_staff_not_allocated_office.__iter__.return_value = ['jimbo'] mock_fellows_not_allocated_office.__iter__.return_value = [] mock_unallocated_people.__iter__.return_value = [] create_db('elsis.db') msg = self.person.commit_people('elsis.db') self.assertEqual(msg, 'Person data commit successfull') @patch.dict('app.room.Room.total_rooms', { 'oculus': [1], 'mordor': [2], 'python': [1] }) @patch.dict('app.person.Person.total_people', { 1: 'Migwi', 2: 'jojo', 3: 'jimbo' }) @patch('app.room.Room.offices') @patch('app.room.Room.livingspaces') @patch('app.person.Person.fellows') @patch('app.person.Person.staff') @patch('app.person.Person.unallocated_people') @patch('app.person.Person.fellows_not_allocated_office') @patch('app.person.Person.staff_not_allocated_office') def test_load_people(self, mock_staff_not_allocated_office, mock_fellows_not_allocated_office, mock_unallocated_people, mock_staff, mock_fellows, mock_livingspaces, mock_offices): mock_livingspaces.__iter__.return_value = ['python'] mock_offices.__iter__.return_value = ['oculus', 'mordor'] mock_fellows.__iter__.return_value = ['Migwi'] mock_staff.__iter__.return_value = ['jojo', 'jimbo'] mock_staff_not_allocated_office.__iter__.return_value = ['jimbo'] mock_fellows_not_allocated_office.__iter__.return_value = [] mock_unallocated_people.__iter__.return_value = [] msg = self.person.load_people('elsis.db') self.assertEqual(msg, 'People loaded successfully')