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]
)
