Skip to content

Commit 0615314

Browse files
[PR #10752/07590cd2 backport][3.11] Add a test to the WebSocket parser for sending one byte at a time (#10754)
Co-authored-by: J. Nick Koston <[email protected]>
1 parent 8b9888d commit 0615314

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

tests/test_websocket_parser.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -344,6 +344,17 @@ def test_simple_binary(
344344
assert res == ((WSMsgType.BINARY, b"binary", ""), 6)
345345

346346

347+
def test_one_byte_at_a_time(
348+
out: WebSocketDataQueue, parser: PatchableWebSocketReader
349+
) -> None:
350+
"""Send one byte at a time to the parser."""
351+
data = build_frame(b"binary", WSMsgType.BINARY)
352+
for i in range(len(data)):
353+
parser._feed_data(data[i : i + 1])
354+
res = out._buffer[0]
355+
assert res == ((WSMsgType.BINARY, b"binary", ""), 6)
356+
357+
347358
def test_fragmentation_header(
348359
out: WebSocketDataQueue, parser: PatchableWebSocketReader
349360
) -> None:

0 commit comments

Comments
 (0)