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