From 152ad526f491e70905e2fec3b7d33a89cf23813b Mon Sep 17 00:00:00 2001
From: Andrei Zeliankou <zelenkov@nginx.com>
Date: Mon, 5 Oct 2020 11:05:00 +0100
Subject: Tests: added ASGI Lifespan.

---
 test/python/lifespan/empty/asgi.py      | 27 +++++++++++++++++++++++++++
 test/python/lifespan/error/asgi.py      |  3 +++
 test/python/lifespan/error_auto/asgi.py |  2 ++
 test/python/lifespan/failed/asgi.py     | 11 +++++++++++
 4 files changed, 43 insertions(+)
 create mode 100644 test/python/lifespan/empty/asgi.py
 create mode 100644 test/python/lifespan/error/asgi.py
 create mode 100644 test/python/lifespan/error_auto/asgi.py
 create mode 100644 test/python/lifespan/failed/asgi.py

(limited to 'test/python/lifespan')

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
-- 
cgit