def setUp(self): """Set the environment for the tests up.""" super().setUp() self.overtime_credits = OvertimeCreditFactory.create_batch( 5, user=self.user) OvertimeCreditFactory.create_batch(5)
def test_overtime_credit_get_superuser(superadmin_client): OvertimeCreditFactory.create_batch(2) OvertimeCreditFactory.create(user=superadmin_client.user) url = reverse("overtime-credit-list") result = superadmin_client.get(url) assert result.status_code == status.HTTP_200_OK json = result.json() assert len(json["data"]) == 3
def test_overtime_credit_get_authenticated(auth_client): OvertimeCreditFactory.create_batch(2) overtime_credit = OvertimeCreditFactory.create(user=auth_client.user) url = reverse("overtime-credit-list") result = auth_client.get(url) assert result.status_code == status.HTTP_200_OK json = result.json() assert len(json["data"]) == 1 assert json["data"][0]["id"] == str(overtime_credit.id)
def test_overtime_credit_get_supervisor(auth_client): user = UserFactory.create() auth_client.user.supervisees.add(user) OvertimeCreditFactory.create_batch(1) OvertimeCreditFactory.create(user=auth_client.user) OvertimeCreditFactory.create(user=user) url = reverse("overtime-credit-list") result = auth_client.get(url) assert result.status_code == status.HTTP_200_OK json = result.json() assert len(json["data"]) == 2