74 lines
2.0 KiB
Python
74 lines
2.0 KiB
Python
import pytest
|
|
import asyncio
|
|
import contextlib
|
|
|
|
from sanic.response import text, stream
|
|
|
|
|
|
async def test_request_cancel_when_connection_lost(loop, app, test_client):
|
|
app.still_serving_cancelled_request = False
|
|
|
|
@app.get('/')
|
|
async def handler(request):
|
|
await asyncio.sleep(1.0)
|
|
# at this point client is already disconnected
|
|
app.still_serving_cancelled_request = True
|
|
return text('OK')
|
|
|
|
test_cli = await test_client(app)
|
|
|
|
# schedule client call
|
|
task = loop.create_task(test_cli.get('/'))
|
|
loop.call_later(0.01, task)
|
|
await asyncio.sleep(0.5)
|
|
|
|
# cancelling request and closing connection after 0.5 sec
|
|
task.cancel()
|
|
|
|
with contextlib.suppress(asyncio.CancelledError):
|
|
await task
|
|
|
|
# Wait for server and check if it's still serving the cancelled request
|
|
await asyncio.sleep(1.0)
|
|
|
|
assert app.still_serving_cancelled_request is False
|
|
|
|
|
|
async def test_stream_request_cancel_when_connection_lost(loop, app, test_client):
|
|
app.still_serving_cancelled_request = False
|
|
|
|
@app.post('/post/<id>', stream=True)
|
|
async def post(request, id):
|
|
assert isinstance(request.stream, asyncio.Queue)
|
|
|
|
async def streaming(response):
|
|
while True:
|
|
body = await request.stream.get()
|
|
if body is None:
|
|
break
|
|
await response.write(body.decode('utf-8'))
|
|
|
|
await asyncio.sleep(1.0)
|
|
# at this point client is already disconnected
|
|
app.still_serving_cancelled_request = True
|
|
|
|
return stream(streaming)
|
|
|
|
test_cli = await test_client(app)
|
|
|
|
# schedule client call
|
|
task = loop.create_task(test_cli.post('/post/1'))
|
|
loop.call_later(0.01, task)
|
|
await asyncio.sleep(0.5)
|
|
|
|
# cancelling request and closing connection after 0.5 sec
|
|
task.cancel()
|
|
|
|
with contextlib.suppress(asyncio.CancelledError):
|
|
await task
|
|
|
|
# Wait for server and check if it's still serving the cancelled request
|
|
await asyncio.sleep(1.0)
|
|
|
|
assert app.still_serving_cancelled_request is False
|