policies: improve api test coverage

Signed-off-by: Jens Langhammer <jens.langhammer@beryju.org>
This commit is contained in:
Jens Langhammer 2022-07-05 23:20:45 +02:00
parent 9f736a9d99
commit d1c4818724
2 changed files with 36 additions and 2 deletions

View file

@ -158,7 +158,7 @@ class PolicyViewSet(
pk=test_params.validated_data["user"].pk
)
if not users.exists():
raise PermissionDenied()
return Response(status=400)
p_request = PolicyRequest(users.first())
p_request.debug = True

View file

@ -1,11 +1,13 @@
"""Test policies API"""
from json import loads
from unittest.mock import MagicMock, patch
from django.urls import reverse
from rest_framework.test import APITestCase
from authentik.core.tests.utils import create_test_admin_user
from authentik.policies.dummy.models import DummyPolicy
from authentik.policies.types import PolicyResult
class TestPoliciesAPI(APITestCase):
@ -17,8 +19,10 @@ class TestPoliciesAPI(APITestCase):
self.user = create_test_admin_user()
self.client.force_login(self.user)
def test_test_call(self):
@patch("authentik.policies.dummy.models.DummyPolicy.passes")
def test_test_call(self, passes_mock: MagicMock):
"""Test Policy's test endpoint"""
passes_mock.return_value = PolicyResult(True, "dummy")
response = self.client.post(
reverse("authentik_api:policy-test", kwargs={"pk": self.policy.pk}),
data={
@ -28,6 +32,22 @@ class TestPoliciesAPI(APITestCase):
body = loads(response.content.decode())
self.assertEqual(body["passing"], True)
self.assertEqual(body["messages"], ["dummy"])
self.assertEqual(body["log_messages"], [])
def test_test_call_invalid(self):
"""Test invalid policy test"""
response = self.client.post(
reverse("authentik_api:policy-test", kwargs={"pk": self.policy.pk}),
data={},
)
self.assertEqual(response.status_code, 400)
response = self.client.post(
reverse("authentik_api:policy-test", kwargs={"pk": self.policy.pk}),
data={
"user": self.user.pk + 1,
},
)
self.assertEqual(response.status_code, 400)
def test_types(self):
"""Test Policy's types endpoint"""
@ -35,3 +55,17 @@ class TestPoliciesAPI(APITestCase):
reverse("authentik_api:policy-types"),
)
self.assertEqual(response.status_code, 200)
def test_cache_info(self):
"""Test Policy's cache_info endpoint"""
response = self.client.get(
reverse("authentik_api:policy-cache-info"),
)
self.assertEqual(response.status_code, 200)
def test_cache_clear(self):
"""Test Policy's cache_clear endpoint"""
response = self.client.post(
reverse("authentik_api:policy-cache-clear"),
)
self.assertEqual(response.status_code, 204)