def test_callingGetEmployeesReturnsListOfDicts(self): Employee.objects.filter = mock.MagicMock() company = Company() company.pk = 98 employee_one = Employee() employee_one.pk = 12 employee_one.user = User() employee_one.user.first_name = 'Fornavn' employee_one.user.last_name = 'Efternavn' employee_one.company = company employee_two = Employee() employee_two.pk = 13 employee_two.user = User() employee_two.user.first_name = 'Fornavn' employee_two.user.last_name = 'Efternavn' employee_two.company = company Employee.objects.filter.return_value = [employee_one, employee_two] employee_one.getMyEmployees = mock.MagicMock() employee_one.getMyEmployees.return_value = [] employee_list = employee_one.getEmployees() self.assertEquals(2, len(employee_list)) self.assertTrue(isinstance(employee_list[0], dict)) self.assertTrue(isinstance(employee_list[0]['manager'], Employee)) self.assertEquals(employee_list[0]['manager'].pk, 12) self.assertTrue(isinstance(employee_list[0]['employees'], list)) self.assertEquals(employee_list[0]['manager'].user.first_name, 'Fornavn') self.assertEquals(employee_list[0]['manager'].user.last_name, 'Efternavn')
def test_callingGetEmployeesReturnsListOfDicts(self):
Employee.objects.filter = mock.MagicMock()
company = Company()
company.pk = 98
employee_one = Employee()
employee_one.pk = 12
employee_one.user = User()
employee_one.user.first_name = 'Fornavn'
employee_one.user.last_name = 'Efternavn'
employee_one.company = company
employee_two = Employee()
employee_two.pk = 13
employee_two.user = User()
employee_two.user.first_name = 'Fornavn'
employee_two.user.last_name = 'Efternavn'
employee_two.company = company
Employee.objects.filter.return_value = [employee_one, employee_two]
employee_one.getMyEmployees = mock.MagicMock()
employee_one.getMyEmployees.return_value = []
employee_list = employee_one.getEmployees()
self.assertEquals(2, len(employee_list))
self.assertTrue(isinstance(employee_list[0], dict))
self.assertTrue(isinstance(employee_list[0]['manager'], Employee))
self.assertEquals(employee_list[0]['manager'].pk, 12)
self.assertTrue(isinstance(employee_list[0]['employees'], list))
self.assertEquals(employee_list[0]['manager'].user.first_name,
'Fornavn')
self.assertEquals(employee_list[0]['manager'].user.last_name,
'Efternavn')
def test_callingGetEmployeesCallsFilterOnEmployeeWithCompanyPkAsArgumentAndIsMangerEqualToTrue(self): employee = Employee() company = Company() company.pk = 8 employee.company = company with mock.patch.object(Employee, 'objects') as query_mock: employee.getEmployees() query_mock.filter.assert_called_with(company__pk=company.pk, is_manager=True)
def test_callingGetAvailableSchemesCallsFilterOnCompetenceFieldCollectionToCompanyRelationWithCompanyPkAsArgument(
self):
company = Company()
company.pk = 8
with mock.patch.object(CompetenceFieldCollectionToCompanyRelation,
'objects') as query_mock:
company.getAvailableSchemes()
query_mock.filter.assert_called_with(company__pk=company.pk)
def test_callingGetEmployeesCallsFilterOnEmployeeWithCompanyPkAsArgumentAndIsMangerEqualToTrue(
self):
employee = Employee()
company = Company()
company.pk = 8
employee.company = company
with mock.patch.object(Employee, 'objects') as query_mock:
employee.getEmployees()
query_mock.filter.assert_called_with(company__pk=company.pk,
is_manager=True)
def test_callingGetEmployeesCallsGetMyEmployeesOneTimePerManagerInSameCompanyAsCaller(self): Employee.objects.filter = mock.MagicMock() company = Company() company.pk = 98 employee_one = Employee() employee_one.pk = 12 employee_one.company = company employee_one.user = User() employee_one.user.first_name = 'Fornavn' employee_one.user.last_name = 'Efternavn' employee_two = Employee() employee_two.pk = 12 employee_two.company = company employee_two.user = User() employee_two.user.first_name = 'Fornavn' employee_two.user.last_name = 'Efternavn' Employee.objects.filter.return_value = [employee_one, employee_two] employee_one.getMyEmployees = mock.MagicMock() employee_one.getEmployees() self.assertEquals(2, employee_one.getMyEmployees.call_count)
def test_callingGetEmployeesCallsGetMyEmployeesOneTimePerManagerInSameCompanyAsCaller(
self):
Employee.objects.filter = mock.MagicMock()
company = Company()
company.pk = 98
employee_one = Employee()
employee_one.pk = 12
employee_one.company = company
employee_one.user = User()
employee_one.user.first_name = 'Fornavn'
employee_one.user.last_name = 'Efternavn'
employee_two = Employee()
employee_two.pk = 12
employee_two.company = company
employee_two.user = User()
employee_two.user.first_name = 'Fornavn'
employee_two.user.last_name = 'Efternavn'
Employee.objects.filter.return_value = [employee_one, employee_two]
employee_one.getMyEmployees = mock.MagicMock()
employee_one.getEmployees()
self.assertEquals(2, employee_one.getMyEmployees.call_count)
def test_callingGetAvailableSchemesCallsFilterOnCompetenceFieldCollectionToCompanyRelationWithCompanyPkAsArgument(self): company = Company() company.pk = 8 with mock.patch.object(CompetenceFieldCollectionToCompanyRelation, 'objects') as query_mock: company.getAvailableSchemes() query_mock.filter.assert_called_with(company__pk=company.pk)
