From d12e24017e3dd06f61f1907315b4305a315d5724 Mon Sep 17 00:00:00 2001
From: Jens Langhammer <jens.langhammer@beryju.org>
Date: Mon, 15 Nov 2021 23:58:19 +0100
Subject: [PATCH] outposts: add websocket tests

Signed-off-by: Jens Langhammer <jens.langhammer@beryju.org>
---
 authentik/outposts/tests/test_ws.py | 77 +++++++++++++++++++++++++++++
 authentik/root/test_runner.py       |  3 +-
 2 files changed, 78 insertions(+), 2 deletions(-)
 create mode 100644 authentik/outposts/tests/test_ws.py

diff --git a/authentik/outposts/tests/test_ws.py b/authentik/outposts/tests/test_ws.py
new file mode 100644
index 000000000..6b4d04c2d
--- /dev/null
+++ b/authentik/outposts/tests/test_ws.py
@@ -0,0 +1,77 @@
+"""Websocket tests"""
+from dataclasses import asdict
+
+from channels.routing import URLRouter
+from channels.testing import WebsocketCommunicator
+from django.test import TransactionTestCase
+
+from authentik import __version__
+from authentik.flows.models import Flow, FlowDesignation
+from authentik.outposts.channels import WebsocketMessage, WebsocketMessageInstruction
+from authentik.outposts.models import Outpost, OutpostType
+from authentik.providers.proxy.models import ProxyProvider
+from authentik.root import websocket
+
+
+class TestOutpostWS(TransactionTestCase):
+    """Websocket tests"""
+
+    def setUp(self) -> None:
+        self.provider: ProxyProvider = ProxyProvider.objects.create(
+            name="test",
+            internal_host="http://localhost",
+            external_host="http://localhost",
+            authorization_flow=Flow.objects.create(
+                name="foo", slug="foo", designation=FlowDesignation.AUTHORIZATION
+            ),
+        )
+        self.outpost: Outpost = Outpost.objects.create(
+            name="test",
+            type=OutpostType.PROXY,
+        )
+        self.outpost.providers.add(self.provider)
+        self.token = self.outpost.token.key
+
+    async def test_auth(self):
+        """Test auth without token"""
+        communicator = WebsocketCommunicator(
+            URLRouter(websocket.websocket_urlpatterns), f"/ws/outpost/{self.outpost.pk}/"
+        )
+        connected, _ = await communicator.connect()
+        self.assertFalse(connected)
+
+    async def test_auth_valid(self):
+        """Test auth with token"""
+        communicator = WebsocketCommunicator(
+            URLRouter(websocket.websocket_urlpatterns),
+            f"/ws/outpost/{self.outpost.pk}/",
+            {b"authorization": f"Bearer {self.token}".encode()},
+        )
+        connected, _ = await communicator.connect()
+        self.assertTrue(connected)
+
+    async def test_send(self):
+        """Test sending of Hello"""
+        communicator = WebsocketCommunicator(
+            URLRouter(websocket.websocket_urlpatterns),
+            f"/ws/outpost/{self.outpost.pk}/",
+            {b"authorization": f"Bearer {self.token}".encode()},
+        )
+        connected, _ = await communicator.connect()
+        self.assertTrue(connected)
+        await communicator.send_json_to(
+            asdict(
+                WebsocketMessage(
+                    instruction=WebsocketMessageInstruction.HELLO,
+                    args={
+                        "version": __version__,
+                        "buildHash": "foo",
+                        "uuid": "123",
+                    },
+                )
+            )
+        )
+        response = await communicator.receive_json_from()
+        self.assertEqual(
+            response, asdict(WebsocketMessage(instruction=WebsocketMessageInstruction.ACK, args={}))
+        )
diff --git a/authentik/root/test_runner.py b/authentik/root/test_runner.py
index 459bb4b64..db9e499f7 100644
--- a/authentik/root/test_runner.py
+++ b/authentik/root/test_runner.py
@@ -37,8 +37,7 @@ class PytestTestRunner:  # pragma: no cover
             argv.append("-vv")
         if self.failfast:
             argv.append("--exitfirst")
-        if self.keepdb:
-            argv.append("--reuse-db")
+        argv.append("--reuse-db")
 
         argv.extend(test_labels)
         return pytest.main(argv)