trustchain-oc1-orchestral
/
authentik
Archived
10
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

flows/tests: update transfer tests to use same rollback function as importer

This commit is contained in:
Jens Langhammer 2020-09-09 17:22:44 +02:00
parent 6d27408a10
commit cddc9bc1b7
1 changed files with 78 additions and 84 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

86
passbook/flows/tests/test_transfer.py
View File

@ -1,15 +1,15 @@
"""Test flow transfer""" """Test flow transfer"""
from json import dumps from json import dumps
from django.db import transaction
from django.test import TransactionTestCase from django.test import TransactionTestCase
from passbook.flows.models import Flow, FlowDesignation, FlowStageBinding from passbook.flows.models import Flow, FlowDesignation, FlowStageBinding
from passbook.flows.transfer.common import DataclassEncoder from passbook.flows.transfer.common import DataclassEncoder
from passbook.flows.transfer.exporter import FlowExporter from passbook.flows.transfer.exporter import FlowExporter
from passbook.flows.transfer.importer import FlowImporter from passbook.flows.transfer.importer import FlowImporter, transaction_rollback
from passbook.policies.expression.models import ExpressionPolicy from passbook.policies.expression.models import ExpressionPolicy
from passbook.policies.models import PolicyBinding from passbook.policies.models import PolicyBinding
from passbook.providers.oauth2.generators import generate_client_id
from passbook.stages.prompt.models import FieldTypes, Prompt, PromptStage from passbook.stages.prompt.models import FieldTypes, Prompt, PromptStage
from passbook.stages.user_login.models import UserLoginStage from passbook.stages.user_login.models import UserLoginStage
@ -28,14 +28,15 @@ class TestFlowTransfer(TransactionTestCase):
def test_export_validate_import(self): def test_export_validate_import(self):
"""Test export and validate it""" """Test export and validate it"""
sid = transaction.savepoint() flow_slug = generate_client_id()
login_stage = UserLoginStage.objects.create(name="default-authentication-login") with transaction_rollback():
login_stage = UserLoginStage.objects.create(name=generate_client_id())
flow = Flow.objects.create( flow = Flow.objects.create(
slug="test", slug=flow_slug,
designation=FlowDesignation.AUTHENTICATION, designation=FlowDesignation.AUTHENTICATION,
name="Welcome to passbook!", name=generate_client_id(),
title="test", title=generate_client_id(),
) )
FlowStageBinding.objects.update_or_create( FlowStageBinding.objects.update_or_create(
target=flow, stage=login_stage, order=0, target=flow, stage=login_stage, order=0,
@ -43,62 +44,60 @@ class TestFlowTransfer(TransactionTestCase):
exporter = FlowExporter(flow) exporter = FlowExporter(flow)
export = exporter.export() export = exporter.export()
transaction.savepoint_rollback(sid)
self.assertEqual(len(export.entries), 3) self.assertEqual(len(export.entries), 3)
export_json = dumps(export, cls=DataclassEncoder) export_json = exporter.export_to_string()
importer = FlowImporter(export_json) importer = FlowImporter(export_json)
self.assertTrue(importer.validate()) self.assertTrue(importer.validate())
self.assertTrue(importer.apply()) self.assertTrue(importer.apply())
self.assertTrue(Flow.objects.filter(slug="test").exists()) self.assertTrue(Flow.objects.filter(slug=flow_slug).exists())
def test_export_validate_import_policies(self): def test_export_validate_import_policies(self):
"""Test export and validate it""" """Test export and validate it"""
sid = transaction.savepoint() flow_slug = generate_client_id()
stage_name = generate_client_id()
with transaction_rollback():
flow_policy = ExpressionPolicy.objects.create( flow_policy = ExpressionPolicy.objects.create(
name="default-source-authentication-if-sso", expression="return True", name=generate_client_id(), expression="return True",
) )
flow = Flow.objects.create( flow = Flow.objects.create(
slug="default-source-authentication-test", slug=flow_slug,
designation=FlowDesignation.AUTHENTICATION, designation=FlowDesignation.AUTHENTICATION,
name="Welcome to passbook!", name=generate_client_id(),
title="test", title=generate_client_id(),
) )
PolicyBinding.objects.create(policy=flow_policy, target=flow, order=0) PolicyBinding.objects.create(policy=flow_policy, target=flow, order=0)
user_login = UserLoginStage.objects.create( user_login = UserLoginStage.objects.create(name=stage_name)
name="default-source-authentication-login-test" fsb = FlowStageBinding.objects.create(
target=flow, stage=user_login, order=0
) )
FlowStageBinding.objects.create(target=flow, stage=user_login, order=0) PolicyBinding.objects.create(policy=flow_policy, target=fsb, order=0)
exporter = FlowExporter(flow) exporter = FlowExporter(flow)
export = exporter.export() export = exporter.export()
transaction.savepoint_rollback(sid)
export_json = dumps(export, cls=DataclassEncoder) export_json = dumps(export, cls=DataclassEncoder)
importer = FlowImporter(export_json) importer = FlowImporter(export_json)
self.assertTrue(importer.validate()) self.assertTrue(importer.validate())
self.assertTrue(importer.apply()) self.assertTrue(importer.apply())
self.assertTrue( self.assertTrue(UserLoginStage.objects.filter(name=stage_name).exists())
UserLoginStage.objects.filter( self.assertTrue(Flow.objects.filter(slug=flow_slug).exists())
name="default-source-authentication-login-test"
).exists()
)
def test_export_validate_import_prompt(self): def test_export_validate_import_prompt(self):
"""Test export and validate it""" """Test export and validate it"""
sid = transaction.savepoint() with transaction_rollback():
# First stage fields # First stage fields
username_prompt = Prompt.objects.create( username_prompt = Prompt.objects.create(
field_key="username", label="Username", order=0, type=FieldTypes.TEXT field_key="username", label="Username", order=0, type=FieldTypes.TEXT
) )
password = Prompt.objects.create( password = Prompt.objects.create(
field_key="password", label="Password", order=1, type=FieldTypes.PASSWORD field_key="password",
label="Password",
order=1,
type=FieldTypes.PASSWORD,
) )
password_repeat = Prompt.objects.create( password_repeat = Prompt.objects.create(
field_key="password_repeat", field_key="password_repeat",
@ -106,25 +105,22 @@ class TestFlowTransfer(TransactionTestCase):
order=2, order=2,
type=FieldTypes.PASSWORD, type=FieldTypes.PASSWORD,
) )
# Stages
first_stage = PromptStage.objects.create(name="prompt-stage-first-test")
first_stage.fields.set([username_prompt, password, password_repeat])
first_stage.save()
# Password checking policy # Password checking policy
password_policy = ExpressionPolicy.objects.create( password_policy = ExpressionPolicy.objects.create(
name="policy-enrollment-password-equals-test", name=generate_client_id(), expression="return True",
expression="return request.context['password'] == request.context['password_repeat']",
)
PolicyBinding.objects.create(
target=first_stage, policy=password_policy, order=0
) )
# Stages
first_stage = PromptStage.objects.create(name=generate_client_id())
first_stage.fields.set([username_prompt, password, password_repeat])
first_stage.validation_policies.set([password_policy])
first_stage.save()
flow = Flow.objects.create( flow = Flow.objects.create(
name="default-enrollment-flow", name=generate_client_id(),
slug="default-enrollment-flow-test", slug=generate_client_id(),
designation=FlowDesignation.ENROLLMENT, designation=FlowDesignation.ENROLLMENT,
title="test", title=generate_client_id(),
) )
FlowStageBinding.objects.create(target=flow, stage=first_stage, order=0) FlowStageBinding.objects.create(target=flow, stage=first_stage, order=0)
@ -133,8 +129,6 @@ class TestFlowTransfer(TransactionTestCase):
export = exporter.export() export = exporter.export()
export_json = dumps(export, cls=DataclassEncoder) export_json = dumps(export, cls=DataclassEncoder)
transaction.savepoint_rollback(sid)
importer = FlowImporter(export_json) importer = FlowImporter(export_json)
self.assertTrue(importer.validate()) self.assertTrue(importer.validate())