def delete_employees(self, request, *args, **kwargs):
company = self.get_object()
if 'employees' in request.data:
existing_employees = User.get_existing_by_pk(
request.data.get('employees'))
company.delete_employees(existing_employees)
serializer = self.get_serializer(company)
return Response(serializer.data)
def test_clean_employees_to_company(self):
numbers_of_employees = 3
employee_list = []
for i in range(1, numbers_of_employees + 1):
user = User.objects.create_user(
username=f'usertest{i}',
first_name='User',
last_name=f'Test {i}',
email=f'test{i}@test.com',
)
employee_list.append(user.pk)
test_company = Company.objects.get(pk=self.company_list[0])
test_company.add_employees(User.get_existing_by_pk(employee_list))
response = self.client.delete(
reverse('company-clean-employees', kwargs={'pk': test_company.pk}))
self.assertEqual(response.status_code, status.HTTP_200_OK)
self.assertEqual(test_company.get_employees().count(), 0)
