From 8711a206a0eb7dafa40ba172c51f0d6ce8821845 Mon Sep 17 00:00:00 2001 From: jonaswinkler Date: Thu, 4 Feb 2021 13:26:58 +0100 Subject: [PATCH] fix websocket test cases --- src/paperless/tests/test_websockets.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/paperless/tests/test_websockets.py b/src/paperless/tests/test_websockets.py index e5745ce77..4f98fe2b5 100644 --- a/src/paperless/tests/test_websockets.py +++ b/src/paperless/tests/test_websockets.py @@ -8,14 +8,16 @@ from paperless.asgi import application class TestWebSockets(TestCase): - async def test_no_auth(self): + @mock.patch("paperless.consumers.async_to_sync") + async def test_no_auth(self, async_to_sync): communicator = WebsocketCommunicator(application, "/ws/status/") connected, subprotocol = await communicator.connect() self.assertFalse(connected) await communicator.disconnect() + @mock.patch("paperless.consumers.async_to_sync") @mock.patch("paperless.consumers.StatusConsumer._authenticated") - async def test_auth(self, _authenticated): + async def test_auth(self, _authenticated, async_to_sync): _authenticated.return_value = True communicator = WebsocketCommunicator(application, "/ws/status/")