Skip to content

Commit 569fa13

Browse files
[PR #9799/d40b1839 backport][3.10] Add compressed WebSocketWriter benchmark (#9801)
Co-authored-by: J. Nick Koston <[email protected]>
1 parent 1ce9953 commit 569fa13

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

tests/test_benchmarks_http_websocket.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -121,3 +121,19 @@ async def _send_one_hundred_websocket_text_messages() -> None:
121121
@benchmark
122122
def _run() -> None:
123123
loop.run_until_complete(_send_one_hundred_websocket_text_messages())
124+
125+
126+
def test_send_one_hundred_websocket_compressed_messages(
127+
loop: asyncio.AbstractEventLoop, benchmark: BenchmarkFixture
128+
) -> None:
129+
"""Benchmark sending 100 WebSocket compressed messages."""
130+
writer = WebSocketWriter(MockProtocol(loop=loop), MockTransport(), compress=15)
131+
raw_message = b"Hello, World!" * 100
132+
133+
async def _send_one_hundred_websocket_compressed_messages() -> None:
134+
for _ in range(100):
135+
await writer._send_frame(raw_message, WSMsgType.BINARY)
136+
137+
@benchmark
138+
def _run() -> None:
139+
loop.run_until_complete(_send_one_hundred_websocket_compressed_messages())

0 commit comments

Comments
 (0)