Skip to content

Commit

Permalink
[PR #9840/cc5fa316 backport][3.10] Add benchmark for sending compress…
Browse files Browse the repository at this point in the history
…ed payload with chunks (#9841)

Co-authored-by: J. Nick Koston <nick@koston.org>
  • Loading branch information
patchback[bot] and bdraco authored Nov 12, 2024
1 parent 4f4b90f commit 68a1f42
Showing 1 changed file with 35 additions and 0 deletions.
35 changes: 35 additions & 0 deletions tests/test_benchmarks_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -120,6 +120,41 @@ def _run() -> None:
loop.run_until_complete(run_client_benchmark())


def test_five_get_requests_with_567296_compressed_chunked_payload(
loop: asyncio.AbstractEventLoop,
aiohttp_client: AiohttpClient,
benchmark: BenchmarkFixture,
) -> None:
"""Benchmark 5 compressed GET requests with a payload of 567296."""
message_count = 5
# This payload compresses poorly to ~567296 bytes.
payload = (
bytes(range(0, 256))
+ bytes(range(255, 0, -1))
+ bytes(range(0, 128))
+ bytes(range(255, 0, -1))
) * 1024

async def handler(request: web.Request) -> web.Response:
resp = web.Response(body=payload)
resp.enable_compression()
return resp

app = web.Application()
app.router.add_route("GET", "/", handler)

async def run_client_benchmark() -> None:
client = await aiohttp_client(app)
for _ in range(message_count):
resp = await client.get("/")
await resp.read()
await client.close()

@benchmark
def _run() -> None:
loop.run_until_complete(run_client_benchmark())


def test_one_hundred_get_requests_with_1024_content_length_payload(
loop: asyncio.AbstractEventLoop,
aiohttp_client: AiohttpClient,
Expand Down

0 comments on commit 68a1f42

Please sign in to comment.