1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
import asyncio
import contextlib
import gc
import sys
from aiohttp import ClientError, ClientSession, web
from aiohttp.test_utils import get_unused_port_socket
gc.set_debug(gc.DEBUG_LEAK)
async def main() -> None:
app = web.Application()
async def stream_handler(request: web.Request) -> web.Response:
assert request.transport is not None
request.transport.close() # Forcefully closing connection
return web.Response()
app.router.add_get("/stream", stream_handler)
sock = get_unused_port_socket("127.0.0.1")
port = sock.getsockname()[1]
runner = web.AppRunner(app)
await runner.setup()
site = web.SockSite(runner, sock)
await site.start()
session = ClientSession()
async def fetch_stream(url: str) -> None:
"""Fetch a stream and read a few bytes from it."""
with contextlib.suppress(ClientError):
await session.get(url)
client_task = asyncio.create_task(fetch_stream(f"http://localhost:{port}/stream"))
await client_task
gc.collect()
client_response_present = any(
type(obj).__name__ == "ClientResponse" for obj in gc.garbage
)
await session.close()
await runner.cleanup()
sys.exit(1 if client_response_present else 0)
asyncio.run(main())
|
