paradizelost/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2025-04-02 13:45:10 -05:00
Code Issues Packages Projects Releases Wiki Activity

added test for web socket authentication

Browse Source
This commit is contained in:
jonaswinkler 2021-02-03 23:18:28 +01:00
parent c51fa60172
commit b9e1ed7d7e
1 changed files with 25 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
src/paperless/tests/test_websockets.py Normal file
View File

@ -0,0 +1,25 @@
from unittest import mock
from channels.testing import WebsocketCommunicator
from django.test import TestCase
from paperless.asgi import application
class TestWebSockets(TestCase):
async def test_no_auth(self):
communicator = WebsocketCommunicator(application, "/ws/status/")
connected, subprotocol = await communicator.connect()
self.assertFalse(connected)
await communicator.disconnect()
@mock.patch("paperless.consumers.StatusConsumer._authenticated")
async def test_auth(self, _authenticated):
_authenticated.return_value = True
communicator = WebsocketCommunicator(application, "/ws/status/")
connected, subprotocol = await communicator.connect()
self.assertTrue(connected)
await communicator.disconnect()