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

fix websocket test cases

Browse Source
This commit is contained in:
jonaswinkler 2021-02-04 13:26:58 +01:00
parent 44ec3a3d9c
commit 8711a206a0
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -8,14 +8,16 @@ from paperless.asgi import application
class TestWebSockets(TestCase): 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/") 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")
@mock.patch("paperless.consumers.StatusConsumer._authenticated") @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 _authenticated.return_value = True
communicator = WebsocketCommunicator(application, "/ws/status/") communicator = WebsocketCommunicator(application, "/ws/status/")