def create(self, validated_data): """ Create and return complete instance of team Entity based on validated data :param validated_data: :return: """ return request_data_models.CreateTeamLeaderOrEmployeeRequestData( **validated_data)
def test_remove_team_employee(self): # first add employee to another team ( Employee must be in at least one team.) request_data = request_data_models.CreateTeamLeaderOrEmployeeRequestData( team_id=2, employee_id=1) new_team_employee = self.use_case.add_team_employee(request_data) deleted_team_employee = self.use_case.remove_team_employee(1) self.assertEqual(deleted_team_employee.id, 1) with self.assertRaises(domain_exceptions.ObjectEntityDoesNotExist): self.use_case.remove_team_employee(1)
def test_assign_team_leader_employee(self): """test adding team leader to a team without a team leader""" request_data = request_data_models.CreateTeamLeaderOrEmployeeRequestData( team_id=2, employee_id=2) new_team_leader = self.use_case.assign_team_leader(request_data) self.assertEqual(len(self.use_case.retrieve_all_teams_leaders()), 2) self.assertIsInstance(new_team_leader, TeamLeaderEntity) self.assertEqual(new_team_leader.team.name, "Greys") self.assertEqual(new_team_leader.leader.id, 2) self.assertEqual(new_team_leader.leader.name, "Scox")
def test_assign_team_leader_invalid_requests(self): # assign team leader with employee that doesnt exist with self.assertRaises(domain_exceptions.EmployeeDoesNotExist): request_data = request_data_models.CreateTeamLeaderOrEmployeeRequestData( team_id=2, employee_id=3, ) self.use_case.assign_team_leader(request_data) # assign team leader with team that doesn't exist with self.assertRaises(domain_exceptions.TeamDoesNotExist): request_data = request_data_models.CreateTeamLeaderOrEmployeeRequestData( team_id=3, employee_id=3, ) self.use_case.assign_team_leader(request_data) # assign team leader to a team that already have a leader with self.assertRaises(domain_exceptions.TeamHasALeader): request_data = request_data_models.CreateTeamLeaderOrEmployeeRequestData( team_id=1, employee_id=2, ) self.use_case.assign_team_leader(request_data)
def test_add_team_employee_invalid_requests(self): # add team employee with employee that doesnt exist with self.assertRaises(domain_exceptions.EmployeeDoesNotExist): request_data = request_data_models.CreateTeamLeaderOrEmployeeRequestData( team_id=2, employee_id=3, ) self.use_case.add_team_employee(request_data) # add team employee with team that doesn't exist with self.assertRaises(domain_exceptions.TeamDoesNotExist): request_data = request_data_models.CreateTeamLeaderOrEmployeeRequestData( team_id=3, employee_id=3, ) self.use_case.add_team_employee(request_data) # add team employee to a team that already have a member of the team with self.assertRaises(domain_exceptions.EmployeeIsATeamMember): request_data = request_data_models.CreateTeamLeaderOrEmployeeRequestData( team_id=1, employee_id=1, ) self.use_case.add_team_employee(request_data)
def setUp(self): self.request_data = request_data_models.CreateTeamLeaderOrEmployeeRequestData( team_id=1, employee_id=2)