Merged with the default branch.1.20.0-1

15 files changed, 288 insertions, 0 deletions

diff --git a/test/python/204_no_content/asgi.py b/test/python/204_no_content/asgi.py
new file mode 100644
index 00000000..634facc2
--- /dev/null
+++ b/test/python/204_no_content/asgi.py
@@ -0,0 +1,8 @@
+async def application(scope, receive, send):
+    assert scope['type'] == 'http'
+
+    await send({
+        'type': 'http.response.start',
+        'status': 204,
+        'headers': [],
+    })
diff --git a/test/python/callable/wsgi.py b/test/python/callable/wsgi.py
new file mode 100644
index 00000000..365f82fa
--- /dev/null
+++ b/test/python/callable/wsgi.py
@@ -0,0 +1,7 @@
+def application(env, start_response):
+    start_response('204', [('Content-Length', '0')])
+    return []
+
+def app(env, start_response):
+    start_response('200', [('Content-Length', '0')])
+    return []
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)
diff --git a/test/python/empty/asgi.py b/test/python/empty/asgi.py
new file mode 100644
index 00000000..58b7c1f2
--- /dev/null
+++ b/test/python/empty/asgi.py
@@ -0,0 +1,10 @@
+async def application(scope, receive, send):
+    assert scope['type'] == 'http'
+
+    await send({
+        'type': 'http.response.start',
+        'status': 200,
+        'headers': [
+            (b'content-length', b'0'),
+        ]
+    })
diff --git a/test/python/lifespan/empty/asgi.py b/test/python/lifespan/empty/asgi.py
new file mode 100644
index 00000000..ea43af13
--- /dev/null
+++ b/test/python/lifespan/empty/asgi.py
@@ -0,0 +1,27 @@
+import os
+
+
+async def application(scope, receive, send):
+    if scope['type'] == 'lifespan':
+        with open('version', 'w+') as f:
+            f.write(
+                scope['asgi']['version'] + ' ' + scope['asgi']['spec_version']
+            )
+        while True:
+            message = await receive()
+            if message['type'] == 'lifespan.startup':
+                os.remove('startup')
+                await send({'type': 'lifespan.startup.complete'})
+            elif message['type'] == 'lifespan.shutdown':
+                os.remove('shutdown')
+                await send({'type': 'lifespan.shutdown.complete'})
+                return
+
+    if scope['type'] == 'http':
+        await send(
+            {
+                'type': 'http.response.start',
+                'status': 204,
+                'headers': [(b'content-length', b'0'),],
+            }
+        )
diff --git a/test/python/lifespan/error/asgi.py b/test/python/lifespan/error/asgi.py
new file mode 100644
index 00000000..509cb3ee
--- /dev/null
+++ b/test/python/lifespan/error/asgi.py
@@ -0,0 +1,3 @@
+async def application(scope, receive, send):
+    if scope['type'] != 'http':
+        raise Exception('Exception blah')
diff --git a/test/python/lifespan/error_auto/asgi.py b/test/python/lifespan/error_auto/asgi.py
new file mode 100644
index 00000000..90d506a1
--- /dev/null
+++ b/test/python/lifespan/error_auto/asgi.py
@@ -0,0 +1,2 @@
+async def application(scope, receive, send):
+    assert scope['type'] == 'http'
diff --git a/test/python/lifespan/failed/asgi.py b/test/python/lifespan/failed/asgi.py
new file mode 100644
index 00000000..8f315f70
--- /dev/null
+++ b/test/python/lifespan/failed/asgi.py
@@ -0,0 +1,11 @@
+async def application(scope, receive, send):
+    if scope['type'] == 'lifespan':
+        while True:
+            message = await receive()
+            if message['type'] == 'lifespan.startup':
+                await send({"type": "lifespan.startup.failed"})
+                raise Exception('Exception blah')
+
+            elif message['type'] == 'lifespan.shutdown':
+                await send({'type': 'lifespan.shutdown.complete'})
+                return
diff --git a/test/python/mirror/asgi.py b/test/python/mirror/asgi.py
new file mode 100644
index 00000000..7088e893
--- /dev/null
+++ b/test/python/mirror/asgi.py
@@ -0,0 +1,22 @@
+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
+
+    await send({
+        'type': 'http.response.start',
+        'status': 200,
+        'headers': [
+            (b'content-length', str(len(body)).encode()),
+        ]
+    })
+
+    await send({
+        'type': 'http.response.body',
+        'body': body,
+    })
diff --git a/test/python/query_string/asgi.py b/test/python/query_string/asgi.py
new file mode 100644
index 00000000..28f4d107
--- /dev/null
+++ b/test/python/query_string/asgi.py
@@ -0,0 +1,11 @@
+async def application(scope, receive, send):
+    assert scope['type'] == 'http'
+
+    await send({
+        'type': 'http.response.start',
+        'status': 200,
+        'headers': [
+            (b'content-length', b'0'),
+            (b'query-string', scope['query_string']),
+        ]
+    })
diff --git a/test/python/server_port/asgi.py b/test/python/server_port/asgi.py
new file mode 100644
index 00000000..e79ced00
--- /dev/null
+++ b/test/python/server_port/asgi.py
@@ -0,0 +1,11 @@
+async def application(scope, receive, send):
+    assert scope['type'] == 'http'
+
+    await send({
+        'type': 'http.response.start',
+        'status': 200,
+        'headers': [
+            (b'content-length', b'0'),
+            (b'server-port', str(scope['server'][1]).encode()),
+        ]
+    })
diff --git a/test/python/threading/asgi.py b/test/python/threading/asgi.py
new file mode 100644
index 00000000..3c978e50
--- /dev/null
+++ b/test/python/threading/asgi.py
@@ -0,0 +1,42 @@
+import asyncio
+import sys
+import time
+import threading
+
+
+class Foo(threading.Thread):
+    num = 10
+
+    def __init__(self, x):
+        self.__x = x
+        threading.Thread.__init__(self)
+
+    def log_index(self, index):
+        sys.stderr.write(
+            "(" + str(index) + ") Thread: " + str(self.__x) + "\n"
+        )
+        sys.stderr.flush()
+
+    def run(self):
+        i = 0
+        for _ in range(3):
+            self.log_index(i)
+            i += 1
+            time.sleep(1)
+            self.log_index(i)
+            i += 1
+
+
+async def application(scope, receive, send):
+    assert scope['type'] == 'http'
+
+    Foo(Foo.num).start()
+    Foo.num += 10
+
+    await send({
+        'type': 'http.response.start',
+        'status': 200,
+        'headers': [
+            (b'content-length', b'0'),
+        ]
+    })
diff --git a/test/python/variables/asgi.py b/test/python/variables/asgi.py
new file mode 100644
index 00000000..dd1cca72
--- /dev/null
+++ b/test/python/variables/asgi.py
@@ -0,0 +1,40 @@
+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):
+        res = []
+        for h in headers:
+            if h[0] == n:
+                res.append(h[1])
+        return b', '.join(res)
+
+    await send({
+        'type': 'http.response.start',
+        'status': 200,
+        'headers': [
+            (b'content-type', get_header(b'content-type')),
+            (b'content-length', str(len(body)).encode()),
+            (b'request-method', scope['method'].encode()),
+            (b'request-uri', scope['path'].encode()),
+            (b'http-host', get_header(b'host')),
+            (b'http-version', scope['http_version'].encode()),
+            (b'asgi-version', scope['asgi']['version'].encode()),
+            (b'asgi-spec-version', scope['asgi']['spec_version'].encode()),
+            (b'scheme', scope['scheme'].encode()),
+            (b'custom-header', get_header(b'custom-header')),
+        ]
+    })
+
+    await send({
+        'type': 'http.response.body',
+        'body': body,
+    })
diff --git a/test/python/websockets/mirror/asgi.py b/test/python/websockets/mirror/asgi.py
new file mode 100644
index 00000000..0f1d9953
--- /dev/null
+++ b/test/python/websockets/mirror/asgi.py
@@ -0,0 +1,18 @@
+async def application(scope, receive, send):
+    if scope['type'] == 'websocket':
+        while True:
+            m = await receive()
+            if m['type'] == 'websocket.connect':
+                await send({
+                  'type': 'websocket.accept',
+                })
+
+            if m['type'] == 'websocket.receive':
+                await send({
+                  'type': 'websocket.send',
+                  'bytes': m.get('bytes', None),
+                  'text': m.get('text', None),
+                })
+
+            if m['type'] == 'websocket.disconnect':
+                break;
diff --git a/test/python/websockets/subprotocol/asgi.py b/test/python/websockets/subprotocol/asgi.py
new file mode 100644
index 00000000..92263dd7
--- /dev/null
+++ b/test/python/websockets/subprotocol/asgi.py
@@ -0,0 +1,25 @@
+async def application(scope, receive, send):
+    assert scope['type'] == 'websocket'
+
+    while True:
+        m = await receive()
+        if m['type'] == 'websocket.connect':
+            subprotocols = scope['subprotocols']
+
+            await send({
+              'type': 'websocket.accept',
+              'headers': [
+                  (b'x-subprotocols', str(subprotocols).encode()),
+              ],
+              'subprotocol': subprotocols[0],
+            })
+
+        if m['type'] == 'websocket.receive':
+            await send({
+              'type': 'websocket.send',
+              'bytes': m.get('bytes', None),
+              'text': m.get('text', None),
+            })
+
+        if m['type'] == 'websocket.disconnect':
+            break;