def test_callingGetMyEmployeesCallsOrderByOnFilterResultWithLastNameThenFirstNameAsArguments(self):
employee = Employee()
employee.pk = 2
with mock.patch.object(Employee, 'objects') as query_mock:
return_employee = Employee()
return_employee.pk = 4
return_employee.user = User()
return_employee.user.first_name = 'Joachim'
return_employee.user.last_name = 'Andersen'
return_employee.manager = employee
employee.getMyEmployees()
query_mock.filter.return_value.order_by.assert_called_with('user__last_name', 'user__first_name')
def test_callingGetMyEmployeesCallsOrderByOnFilterResultWithLastNameThenFirstNameAsArguments(
self):
employee = Employee()
employee.pk = 2
with mock.patch.object(Employee, 'objects') as query_mock:
return_employee = Employee()
return_employee.pk = 4
return_employee.user = User()
return_employee.user.first_name = 'Joachim'
return_employee.user.last_name = 'Andersen'
return_employee.manager = employee
employee.getMyEmployees()
query_mock.filter.return_value.order_by.assert_called_with(
'user__last_name', 'user__first_name')
def test_callingGetMyEmployeesReturnsListOfEmployees(self):
employee = Employee()
employee.pk = 2
with mock.patch.object(Employee, 'objects') as employee_mock:
return_employee = Employee()
return_employee.pk = 4
return_employee.user = User()
return_employee.user.first_name = 'Joachim'
return_employee.user.last_name = 'Andersen'
return_employee.manager = employee
query_set_mock = mock.MagicMock()
query_set_mock.order_by.return_value = iter([return_employee, ])
employee_mock.filter.return_value = query_set_mock
result = employee.getMyEmployees()
self.assertTrue(isinstance(result[0], Employee))
self.assertEqual(1, len(result))
self.assertEqual(4, result[0].pk)
self.assertEqual('Joachim', result[0].user.first_name)
self.assertEqual('Andersen', result[0].user.last_name)
def test_callingGetMyEmployeesReturnsListOfEmployees(self):
employee = Employee()
employee.pk = 2
with mock.patch.object(Employee, 'objects') as employee_mock:
return_employee = Employee()
return_employee.pk = 4
return_employee.user = User()
return_employee.user.first_name = 'Joachim'
return_employee.user.last_name = 'Andersen'
return_employee.manager = employee
query_set_mock = mock.MagicMock()
query_set_mock.order_by.return_value = iter([
return_employee,
])
employee_mock.filter.return_value = query_set_mock
result = employee.getMyEmployees()
self.assertTrue(isinstance(result[0], Employee))
self.assertEqual(1, len(result))
self.assertEqual(4, result[0].pk)
self.assertEqual('Joachim', result[0].user.first_name)
self.assertEqual('Andersen', result[0].user.last_name)
