Skip to content

Commit

Permalink
[PR #9885/76b0d734 backport][3.11] Add benchmarks for web middleware (#…
Browse files Browse the repository at this point in the history
…9889)

Co-authored-by: J. Nick Koston <nick@koston.org>
  • Loading branch information
patchback[bot] and bdraco authored Nov 14, 2024
1 parent 581390c commit 00978de
Showing 1 changed file with 44 additions and 0 deletions.
44 changes: 44 additions & 0 deletions tests/test_benchmarks_web_middleware.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,44 @@
"""codspeed benchmarks for web middlewares."""

import asyncio

from pytest_codspeed import BenchmarkFixture

from aiohttp import web
from aiohttp.pytest_plugin import AiohttpClient
from aiohttp.typedefs import Handler


def test_ten_web_middlewares(
benchmark: BenchmarkFixture,
loop: asyncio.AbstractEventLoop,
aiohttp_client: AiohttpClient,
) -> None:
"""Benchmark 100 requests with 10 middlewares."""
message_count = 100

async def handler(request: web.Request) -> web.Response:
return web.Response()

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

class MiddlewareClass:
@web.middleware
async def call(
self, request: web.Request, handler: Handler
) -> web.StreamResponse:
return await handler(request)

for _ in range(10):
app.middlewares.append(MiddlewareClass().call)

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

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

0 comments on commit 00978de

Please sign in to comment.