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)