Skip to content

Commit

Permalink
lint: fix flake8 errors
Browse files Browse the repository at this point in the history
  • Loading branch information
cacosandon committed Sep 4, 2024
1 parent 43a83c6 commit f9fceba
Showing 1 changed file with 13 additions and 3 deletions.
16 changes: 13 additions & 3 deletions tests/test_generic_websocket.py
Original file line number Diff line number Diff line change
Expand Up @@ -486,6 +486,7 @@ async def connect(self):
assert msg["code"] == 4007
assert msg["reason"] == "test reason"


@pytest.mark.django_db
@pytest.mark.asyncio
async def test_websocket_receive_with_none_text():
Expand All @@ -507,8 +508,15 @@ def receive(self, text_data=None, bytes_data=None):
connected, _ = await communicator.connect()
assert connected

# Simulate Hypercorn behavior (both 'text' and 'bytes' keys present, but 'text' is None)
await communicator.send_input({"type": "websocket.receive", "text": None, "bytes": b"test data"})
# Simulate Hypercorn behavior
# (both 'text' and 'bytes' keys present, but 'text' is None)
await communicator.send_input(
{
"type": "websocket.receive",
"text": None,
"bytes": b"test data",
}
)
response = await communicator.receive_output()
assert response["type"] == "websocket.send"
assert response["text"] == "Received bytes of length: 9"
Expand All @@ -520,7 +528,9 @@ def receive(self, text_data=None, bytes_data=None):
assert response["text"] == "Received bytes of length: 9"

# Test with valid text data
await communicator.send_input({"type": "websocket.receive", "text": "Hello, world!"})
await communicator.send_input(
{"type": "websocket.receive", "text": "Hello, world!"}
)
response = await communicator.receive_output()
assert response["type"] == "websocket.send"
assert response["text"] == "Received text: Hello, world!"
Expand Down

0 comments on commit f9fceba

Please sign in to comment.