def test_update_employee_in_file(self): """ Test update() method in persistence file. """ self.employee.name = 'Updated name' self.manager.update(self.employee) manager2 = get_manager() self.assertEqual([self.employee], manager2.get_employees(employee_id=1))
def test_add_employee_persistence_update(self): """ Test persistence file after add() method. """ new_employee = self.employee new_employee.name = 'New employee' new_employee.id_sse = None self.manager.add(new_employee) manager2 = get_manager() self.assertEqual([new_employee], manager2.get_employees(name='New employee'))
def setUp(self): """ Create EmployeeManagerDummy instance. Create object of employee with id = 1. """ self.manager = get_manager() self.employee = Employee( id_sse=1, name='Flo Zavala', position='Abiliton Application Architect', project=r'IT\TRSY\TRSY Management\Software Architecture', vertical='TRSY', workplaces=[ Workplace(location='Kyiv1', room='321', table=4), ])
def test_equal_sse(self): self.loggining() request = self._client.get('/api/v1/workplaces/?office=Kyiv1') floor = json.loads(request.content.decode())[0] workplace1 = floor['workplaces'][0] workplace2 = floor['workplaces'][1] self.assertEqual(workplace1['employee'], None) employee2 = sse.get_manager().get_employees( employee_id=workplace2['employee']['id_sse'] )[0] self.assertIn( (workplace2['room'], workplace2['table']), [(wp.room, wp.table) for wp in employee2.workplaces] )