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

tests/e2e: fix IdentificationStage not having sources set 

Signed-off-by: Jens Langhammer <jens.langhammer@beryju.org>
This commit is contained in:
Jens Langhammer 2021-05-25 17:35:24 +02:00
parent c4453f38a2
commit 02411bb543
2 changed files with 21 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/e2e/test_source_oauth.py
View File

@ -25,6 +25,7 @@ from authentik.providers.oauth2.generators import (
from authentik.sources.oauth.models import OAuthSource from authentik.sources.oauth.models import OAuthSource
from authentik.sources.oauth.types.manager import SourceType from authentik.sources.oauth.types.manager import SourceType
from authentik.sources.oauth.types.twitter import TwitterOAuthCallback from authentik.sources.oauth.types.twitter import TwitterOAuthCallback
from authentik.stages.identification.models import IdentificationStage
from tests.e2e.utils import SeleniumTestCase, apply_migration, object_manager, retry from tests.e2e.utils import SeleniumTestCase, apply_migration, object_manager, retry
CONFIG_PATH = "/tmp/dex.yml" # nosec CONFIG_PATH = "/tmp/dex.yml" # nosec
@ -115,7 +116,7 @@ class TestSourceOAuth2(SeleniumTestCase):
authentication_flow = Flow.objects.get(slug="default-source-authentication") authentication_flow = Flow.objects.get(slug="default-source-authentication")
enrollment_flow = Flow.objects.get(slug="default-source-enrollment") enrollment_flow = Flow.objects.get(slug="default-source-enrollment")
OAuthSource.objects.create( # nosec source = OAuthSource.objects.create( # nosec
name="dex", name="dex",
slug="dex", slug="dex",
authentication_flow=authentication_flow, authentication_flow=authentication_flow,
@ -127,6 +128,9 @@ class TestSourceOAuth2(SeleniumTestCase):
consumer_key="example-app", consumer_key="example-app",
consumer_secret=self.client_secret, consumer_secret=self.client_secret,
) )
ident_stage = IdentificationStage.objects.first()
ident_stage.sources.set([source])
ident_stage.save()
@retry() @retry()
@apply_migration("authentik_core", "0003_default_user") @apply_migration("authentik_core", "0003_default_user")
@ -308,7 +312,7 @@ class TestSourceOAuth1(SeleniumTestCase):
authentication_flow = Flow.objects.get(slug="default-source-authentication") authentication_flow = Flow.objects.get(slug="default-source-authentication")
enrollment_flow = Flow.objects.get(slug="default-source-enrollment") enrollment_flow = Flow.objects.get(slug="default-source-enrollment")
OAuthSource.objects.create( # nosec source = OAuthSource.objects.create( # nosec
name="oauth1", name="oauth1",
slug=self.source_slug, slug=self.source_slug,
authentication_flow=authentication_flow, authentication_flow=authentication_flow,
@ -317,6 +321,9 @@ class TestSourceOAuth1(SeleniumTestCase):
consumer_key=self.client_id, consumer_key=self.client_id,
consumer_secret=self.client_secret, consumer_secret=self.client_secret,
) )
ident_stage = IdentificationStage.objects.first()
ident_stage.sources.set([source])
ident_stage.save()
@retry() @retry()
@apply_migration("authentik_core", "0003_default_user") @apply_migration("authentik_core", "0003_default_user")

14
tests/e2e/test_source_saml.py
View File

@ -16,6 +16,7 @@ from authentik.core.models import User
from authentik.crypto.models import CertificateKeyPair from authentik.crypto.models import CertificateKeyPair
from authentik.flows.models import Flow from authentik.flows.models import Flow
from authentik.sources.saml.models import SAMLBindingTypes, SAMLSource from authentik.sources.saml.models import SAMLBindingTypes, SAMLSource
from authentik.stages.identification.models import IdentificationStage
from tests.e2e.utils import SeleniumTestCase, apply_migration, object_manager, retry from tests.e2e.utils import SeleniumTestCase, apply_migration, object_manager, retry
LOGGER = get_logger() LOGGER = get_logger()
@ -119,7 +120,7 @@ class TestSourceSAML(SeleniumTestCase):
key_data=IDP_KEY, key_data=IDP_KEY,
) )
SAMLSource.objects.create( source = SAMLSource.objects.create(
name="saml-idp-test", name="saml-idp-test",
slug="saml-idp-test", slug="saml-idp-test",
authentication_flow=authentication_flow, authentication_flow=authentication_flow,
@ -130,6 +131,9 @@ class TestSourceSAML(SeleniumTestCase):
binding_type=SAMLBindingTypes.REDIRECT, binding_type=SAMLBindingTypes.REDIRECT,
signing_kp=keypair, signing_kp=keypair,
) )
ident_stage = IdentificationStage.objects.first()
ident_stage.sources.set([source])
ident_stage.save()
self.driver.get(self.live_server_url) self.driver.get(self.live_server_url)
@ -199,6 +203,9 @@ class TestSourceSAML(SeleniumTestCase):
binding_type=SAMLBindingTypes.POST, binding_type=SAMLBindingTypes.POST,
signing_kp=keypair, signing_kp=keypair,
) )
ident_stage = IdentificationStage.objects.first()
ident_stage.sources.set([source])
ident_stage.save()
self.driver.get(self.live_server_url) self.driver.get(self.live_server_url)
@ -270,7 +277,7 @@ class TestSourceSAML(SeleniumTestCase):
key_data=IDP_KEY, key_data=IDP_KEY,
) )
SAMLSource.objects.create( source = SAMLSource.objects.create(
name="saml-idp-test", name="saml-idp-test",
slug="saml-idp-test", slug="saml-idp-test",
authentication_flow=authentication_flow, authentication_flow=authentication_flow,
@ -281,6 +288,9 @@ class TestSourceSAML(SeleniumTestCase):
binding_type=SAMLBindingTypes.POST_AUTO, binding_type=SAMLBindingTypes.POST_AUTO,
signing_kp=keypair, signing_kp=keypair,
) )
ident_stage = IdentificationStage.objects.first()
ident_stage.sources.set([source])
ident_stage.save()
self.driver.get(self.live_server_url) self.driver.get(self.live_server_url)