summaryrefslogtreecommitdiffhomepage
path: root/test/python/delayed/asgi.py
diff options
context:
space:
mode:
authorMax Romanov <max.romanov@nginx.com>2020-10-01 23:55:43 +0300
committerMax Romanov <max.romanov@nginx.com>2020-10-01 23:55:43 +0300
commit12f225a43acc6b5086b08c3d7df6f6ac2322efa1 (patch)
treeb74841e98672e5fe9233175ec11cb9a63739f27d /test/python/delayed/asgi.py
parentd97e3a3296db77f6a33ce010a66d2a0b2d4bac49 (diff)
downloadunit-12f225a43acc6b5086b08c3d7df6f6ac2322efa1.tar.gz
unit-12f225a43acc6b5086b08c3d7df6f6ac2322efa1.tar.bz2
Tests: added ASGI HTTP applications.
Diffstat (limited to '')
-rw-r--r--test/python/delayed/asgi.py51
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)