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

websocket testing

Browse Source
This commit is contained in:
jonaswinkler 2021-02-04 14:17:18 +01:00
parent 8711a206a0
commit 866c8fc848
1 changed files with 38 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
src/paperless/tests/test_websockets.py
View File

@ -1,23 +1,31 @@
from unittest import mock from unittest import mock
from channels.layers import get_channel_layer
from channels.testing import WebsocketCommunicator from channels.testing import WebsocketCommunicator
from django.test import TestCase from django.test import TestCase, override_settings
from paperless.asgi import application from paperless.asgi import application
TEST_CHANNEL_LAYERS = {
'default': {
'BACKEND': 'channels.layers.InMemoryChannelLayer',
},
}
class TestWebSockets(TestCase): class TestWebSockets(TestCase):
@mock.patch("paperless.consumers.async_to_sync") @override_settings(CHANNEL_LAYERS=TEST_CHANNEL_LAYERS)
async def test_no_auth(self, async_to_sync): async def test_no_auth(self):
communicator = WebsocketCommunicator(application, "/ws/status/") communicator = WebsocketCommunicator(application, "/ws/status/")
connected, subprotocol = await communicator.connect() connected, subprotocol = await communicator.connect()
self.assertFalse(connected) self.assertFalse(connected)
await communicator.disconnect() await communicator.disconnect()
@mock.patch("paperless.consumers.async_to_sync") @override_settings(CHANNEL_LAYERS=TEST_CHANNEL_LAYERS)
@mock.patch("paperless.consumers.StatusConsumer._authenticated") @mock.patch("paperless.consumers.StatusConsumer._authenticated")
async def test_auth(self, _authenticated, async_to_sync): async def test_auth(self, _authenticated):
_authenticated.return_value = True _authenticated.return_value = True
communicator = WebsocketCommunicator(application, "/ws/status/") communicator = WebsocketCommunicator(application, "/ws/status/")
@ -25,3 +33,28 @@ class TestWebSockets(TestCase):
self.assertTrue(connected) self.assertTrue(connected)
await communicator.disconnect() await communicator.disconnect()
@override_settings(CHANNEL_LAYERS=TEST_CHANNEL_LAYERS)
@mock.patch("paperless.consumers.StatusConsumer._authenticated")
async def test_receive(self, _authenticated):
_authenticated.return_value = True
communicator = WebsocketCommunicator(application, "/ws/status/")
connected, subprotocol = await communicator.connect()
self.assertTrue(connected)
message = {
"task_id": "test"
}
channel_layer = get_channel_layer()
await channel_layer.group_send("status_updates", {
"type": "status_update",
"data": message
})
response = await communicator.receive_json_from()
self.assertEqual(response, message)
await communicator.disconnect()