habsi/authentik
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

tests/integration: use channels test server

Browse Source 
Signed-off-by: Jens Langhammer <jens.langhammer@beryju.org>
This commit is contained in:
Jens Langhammer
2021-11-15 22:37:36 +01:00
parent 078633c2af
commit e4a0345231
2 changed files with 26 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
tests/integration/test_outpost_docker.py
View File

@ -4,22 +4,27 @@ from tempfile import mkdtemp
from time import sleep from time import sleep
import yaml import yaml
from django.test import TestCase from channels.testing import ChannelsLiveServerTestCase
from docker import DockerClient, from_env from docker import DockerClient, from_env
from docker.models.containers import Container from docker.models.containers import Container
from docker.types.healthcheck import Healthcheck from docker.types.healthcheck import Healthcheck
from authentik import __version__ from authentik import __version__
from authentik.crypto.models import CertificateKeyPair from authentik.crypto.models import CertificateKeyPair
from authentik.flows.models import Flow from authentik.flows.models import Flow, FlowDesignation
from authentik.outposts.controllers.docker import DockerController from authentik.outposts.controllers.docker import DockerController
from authentik.outposts.models import DockerServiceConnection, Outpost, OutpostType from authentik.outposts.models import (
DockerServiceConnection,
Outpost,
OutpostType,
default_outpost_config,
)
from authentik.outposts.tasks import outpost_local_connection from authentik.outposts.tasks import outpost_local_connection
from authentik.providers.proxy.models import ProxyProvider from authentik.providers.proxy.models import ProxyProvider
from tests.e2e.utils import get_docker_tag from tests.e2e.utils import get_docker_tag
class OutpostDockerTests(TestCase): class OutpostDockerTests(ChannelsLiveServerTestCase):
"""Test Docker Controllers""" """Test Docker Controllers"""
def _start_container(self, ssl_folder: str) -> Container: def _start_container(self, ssl_folder: str) -> Container:
@ -59,7 +64,9 @@ class OutpostDockerTests(TestCase):
name="test", name="test",
internal_host="http://localhost", internal_host="http://localhost",
external_host="http://localhost", external_host="http://localhost",
authorization_flow=Flow.objects.first(), authorization_flow=Flow.objects.create(
name="foo", slug="foo", designation=FlowDesignation.AUTHORIZATION
),
) )
authentication_kp = CertificateKeyPair.objects.create( authentication_kp = CertificateKeyPair.objects.create(
name="docker-authentication", name="docker-authentication",
@ -82,6 +89,7 @@ class OutpostDockerTests(TestCase):
name="test", name="test",
type=OutpostType.PROXY, type=OutpostType.PROXY,
service_connection=self.service_connection, service_connection=self.service_connection,
_config=default_outpost_config(self.live_server_url),
) )
self.outpost.providers.add(self.provider) self.outpost.providers.add(self.provider)
self.outpost.save() self.outpost.save()

18
tests/integration/test_proxy_docker.py
View File

@ -4,22 +4,27 @@ from tempfile import mkdtemp
from time import sleep from time import sleep
import yaml import yaml
from django.test import TestCase from channels.testing.live import ChannelsLiveServerTestCase
from docker import DockerClient, from_env from docker import DockerClient, from_env
from docker.models.containers import Container from docker.models.containers import Container
from docker.types.healthcheck import Healthcheck from docker.types.healthcheck import Healthcheck
from authentik import __version__ from authentik import __version__
from authentik.crypto.models import CertificateKeyPair from authentik.crypto.models import CertificateKeyPair
from authentik.flows.models import Flow from authentik.flows.models import Flow, FlowDesignation
from authentik.outposts.models import DockerServiceConnection, Outpost, OutpostType from authentik.outposts.models import (
DockerServiceConnection,
Outpost,
OutpostType,
default_outpost_config,
)
from authentik.outposts.tasks import outpost_local_connection from authentik.outposts.tasks import outpost_local_connection
from authentik.providers.proxy.controllers.docker import DockerController from authentik.providers.proxy.controllers.docker import DockerController
from authentik.providers.proxy.models import ProxyProvider from authentik.providers.proxy.models import ProxyProvider
from tests.e2e.utils import get_docker_tag from tests.e2e.utils import get_docker_tag
class TestProxyDocker(TestCase): class TestProxyDocker(ChannelsLiveServerTestCase):
"""Test Docker Controllers""" """Test Docker Controllers"""
def _start_container(self, ssl_folder: str) -> Container: def _start_container(self, ssl_folder: str) -> Container:
@ -59,7 +64,9 @@ class TestProxyDocker(TestCase):
name="test", name="test",
internal_host="http://localhost", internal_host="http://localhost",
external_host="http://localhost", external_host="http://localhost",
authorization_flow=Flow.objects.first(), authorization_flow=Flow.objects.create(
name="foo", slug="foo", designation=FlowDesignation.AUTHORIZATION
),
) )
authentication_kp = CertificateKeyPair.objects.create( authentication_kp = CertificateKeyPair.objects.create(
name="docker-authentication", name="docker-authentication",
@ -82,6 +89,7 @@ class TestProxyDocker(TestCase):
name="test", name="test",
type=OutpostType.PROXY, type=OutpostType.PROXY,
service_connection=self.service_connection, service_connection=self.service_connection,
_config=default_outpost_config(self.live_server_url),
) )
self.outpost.providers.add(self.provider) self.outpost.providers.add(self.provider)
self.outpost.save() self.outpost.save()