9bb0d04aeb
* root: add pytest-randomly to randomise tests Signed-off-by: Jens Langhammer <jens.langhammer@beryju.org> * *: generate flows for testing instead of relying on existing ones Signed-off-by: Jens Langhammer <jens.langhammer@beryju.org> * *: generate users for testing instead of relying on existing ones Signed-off-by: Jens Langhammer <jens.langhammer@beryju.org> * *: use generated certificate Signed-off-by: Jens Langhammer <jens.langhammer@beryju.org> * tests/e2e: keep containers Signed-off-by: Jens Langhammer <jens.langhammer@beryju.org> * tests/e2e: use websockets test case Signed-off-by: Jens Langhammer <jens.langhammer@beryju.org>
103 lines
3.5 KiB
Python
103 lines
3.5 KiB
Python
"""API flow tests"""
|
|
from django.urls import reverse
|
|
from rest_framework.test import APITestCase
|
|
|
|
from authentik.core.tests.utils import create_test_admin_user
|
|
from authentik.flows.api.stages import StageSerializer, StageViewSet
|
|
from authentik.flows.models import Flow, FlowDesignation, FlowStageBinding, Stage
|
|
from authentik.policies.dummy.models import DummyPolicy
|
|
from authentik.policies.models import PolicyBinding
|
|
from authentik.stages.dummy.models import DummyStage
|
|
|
|
DIAGRAM_EXPECTED = """st=>start: Start
|
|
stage_0=>operation: Stage
|
|
dummy1
|
|
stage_1_policy_0=>condition: Policy
|
|
None
|
|
stage_1=>operation: Stage
|
|
dummy2
|
|
e=>end: End|future
|
|
st(right)->stage_0
|
|
stage_0(bottom)->stage_1_policy_0
|
|
stage_1_policy_0(yes, right)->stage_1
|
|
stage_1_policy_0(no, bottom)->e
|
|
stage_1(bottom)->e"""
|
|
DIAGRAM_SHORT_EXPECTED = """st=>start: Start
|
|
e=>end: End|future
|
|
st(right)->e"""
|
|
|
|
|
|
class TestFlowsAPI(APITestCase):
|
|
"""API tests"""
|
|
|
|
def test_models(self):
|
|
"""Test that ui_user_settings returns none"""
|
|
self.assertIsNone(Stage().ui_user_settings)
|
|
|
|
def test_api_serializer(self):
|
|
"""Test that stage serializer returns the correct type"""
|
|
obj = DummyStage()
|
|
self.assertEqual(StageSerializer().get_component(obj), "ak-stage-dummy-form")
|
|
self.assertEqual(StageSerializer().get_verbose_name(obj), "Dummy Stage")
|
|
|
|
def test_api_viewset(self):
|
|
"""Test that stage serializer returns the correct type"""
|
|
dummy = DummyStage.objects.create()
|
|
self.assertIn(dummy, StageViewSet().get_queryset())
|
|
|
|
def test_api_diagram(self):
|
|
"""Test flow diagram."""
|
|
user = create_test_admin_user()
|
|
self.client.force_login(user)
|
|
|
|
flow = Flow.objects.create(
|
|
name="test-default-context",
|
|
slug="test-default-context",
|
|
designation=FlowDesignation.AUTHENTICATION,
|
|
)
|
|
false_policy = DummyPolicy.objects.create(result=False, wait_min=1, wait_max=2)
|
|
|
|
FlowStageBinding.objects.create(
|
|
target=flow, stage=DummyStage.objects.create(name="dummy1"), order=0
|
|
)
|
|
binding2 = FlowStageBinding.objects.create(
|
|
target=flow,
|
|
stage=DummyStage.objects.create(name="dummy2"),
|
|
order=1,
|
|
re_evaluate_policies=True,
|
|
)
|
|
|
|
PolicyBinding.objects.create(policy=false_policy, target=binding2, order=0)
|
|
|
|
response = self.client.get(
|
|
reverse("authentik_api:flow-diagram", kwargs={"slug": flow.slug})
|
|
)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertJSONEqual(response.content, {"diagram": DIAGRAM_EXPECTED})
|
|
|
|
def test_api_diagram_no_stages(self):
|
|
"""Test flow diagram with no stages."""
|
|
user = create_test_admin_user()
|
|
self.client.force_login(user)
|
|
|
|
flow = Flow.objects.create(
|
|
name="test-default-context",
|
|
slug="test-default-context",
|
|
designation=FlowDesignation.AUTHENTICATION,
|
|
)
|
|
response = self.client.get(
|
|
reverse("authentik_api:flow-diagram", kwargs={"slug": flow.slug})
|
|
)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertJSONEqual(response.content, {"diagram": DIAGRAM_SHORT_EXPECTED})
|
|
|
|
def test_types(self):
|
|
"""Test Stage's types endpoint"""
|
|
user = create_test_admin_user()
|
|
self.client.force_login(user)
|
|
|
|
response = self.client.get(
|
|
reverse("authentik_api:stage-types"),
|
|
)
|
|
self.assertEqual(response.status_code, 200)
|