How to test send_json() contents when passing close=True

[MaZZly]

How to test send_json() contents when passing close=True

[MaZZly]

Okay, so I managed to figure out when/why this happens...

The scenario is that the connection is closed during establishing of connection in AsyncJsonWebsocketConsumer.connect()

When we open the connection with WebsocketCommunicator.connect(), the first message is interpreted as "all good, we are connected", and then the second/remaining message in the pipeline is the websocket.close-one

communicator = WebsocketCommunicator(application, ws_url, headers)
connected, subprotocol = await communicator.connect()  # This one consumes the "jwt-expired"-message
msg = await communicator.receive_output()
self.assertEqual(msg, {"type": "websocket.close"})

To test that the jwt-expired-message is there during connect we have to manually send the websocket.connect-message:

communicator = WebsocketCommunicator(application, ws_url, headers)
await communicator.send_input({"type": "websocket.connect"})
msg = await communicator.receive_json_from()
self.assertEqual(msg, {"type": "jwt-expired"})
msg = await communicator.receive_output()
self.assertEqual(msg, {"type": "websocket.close"})
self.assertTrue(await communicator.receive_nothing())

[carltongibson]

What's your test code look like?

With WebsocketCommunicator you'd use receive_json_from to check the messages as they come in...