summaryrefslogtreecommitdiffhomepage
path: root/test/python/lifespan
diff options
context:
space:
mode:
Diffstat (limited to 'test/python/lifespan')
-rw-r--r--test/python/lifespan/empty/asgi.py27
-rw-r--r--test/python/lifespan/error/asgi.py3
-rw-r--r--test/python/lifespan/error_auto/asgi.py2
-rw-r--r--test/python/lifespan/failed/asgi.py11
4 files changed, 43 insertions, 0 deletions
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