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