diff options
author | Max Romanov <max.romanov@nginx.com> | 2020-10-01 23:55:43 +0300 |
---|---|---|
committer | Max Romanov <max.romanov@nginx.com> | 2020-10-01 23:55:43 +0300 |
commit | 12f225a43acc6b5086b08c3d7df6f6ac2322efa1 (patch) | |
tree | b74841e98672e5fe9233175ec11cb9a63739f27d /test/python/delayed/asgi.py | |
parent | d97e3a3296db77f6a33ce010a66d2a0b2d4bac49 (diff) | |
download | unit-12f225a43acc6b5086b08c3d7df6f6ac2322efa1.tar.gz unit-12f225a43acc6b5086b08c3d7df6f6ac2322efa1.tar.bz2 |
Tests: added ASGI HTTP applications.
Diffstat (limited to 'test/python/delayed/asgi.py')
-rw-r--r-- | test/python/delayed/asgi.py | 51 |
1 files changed, 51 insertions, 0 deletions
diff --git a/test/python/delayed/asgi.py b/test/python/delayed/asgi.py new file mode 100644 index 00000000..d5cad929 --- /dev/null +++ b/test/python/delayed/asgi.py @@ -0,0 +1,51 @@ +import asyncio + +async def application(scope, receive, send): + assert scope['type'] == 'http' + + body = b'' + while True: + m = await receive() + body += m.get('body', b'') + if not m.get('more_body', False): + break + + headers = scope.get('headers', []) + + def get_header(n, v=None): + for h in headers: + if h[0] == n: + return h[1] + return v + + parts = int(get_header(b'x-parts', 1)) + delay = int(get_header(b'x-delay', 0)) + + loop = asyncio.get_event_loop() + + async def sleep(n): + future = loop.create_future() + loop.call_later(n, future.set_result, None) + await future + + await send({ + 'type': 'http.response.start', + 'status': 200, + 'headers': [ + (b'content-length', str(len(body)).encode()), + ] + }) + + if not body: + await sleep(delay) + return + + step = int(len(body) / parts) + for i in range(0, len(body), step): + await send({ + 'type': 'http.response.body', + 'body': body[i : i + step], + 'more_body': True, + }) + + await sleep(delay) |