Tests: added ASGI Lifespan.

5 files changed, 114 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
diff --git a/test/test_asgi_lifespan.py b/test/test_asgi_lifespan.py
new file mode 100644
index 00000000..2fbc68f8
--- /dev/null
+++ b/test/test_asgi_lifespan.py
@@ -0,0 +1,71 @@
+import os
+import pytest
+from distutils.version import LooseVersion
+
+from unit.applications.lang.python import TestApplicationPython
+from conftest import option
+
+
+class TestASGILifespan(TestApplicationPython):
+    prerequisites = {
+        'modules': {'python': lambda v: LooseVersion(v) >= LooseVersion('3.5')}
+    }
+    load_module = 'asgi'
+
+    def test_asgi_lifespan(self):
+        self.load('lifespan/empty')
+
+        startup_path = option.test_dir + '/python/lifespan/empty/startup'
+        shutdown_path = option.test_dir + '/python/lifespan/empty/shutdown'
+        version_path = option.test_dir + '/python/lifespan/empty/version'
+
+        open(startup_path, 'a').close()
+        open(shutdown_path, 'a').close()
+        open(version_path, 'a').close()
+
+        assert self.get()['status'] == 204
+
+        self.stop()
+
+        is_startup = os.path.isfile(startup_path)
+        is_shutdown = os.path.isfile(shutdown_path)
+
+        if is_startup:
+            os.remove(startup_path)
+
+        if is_shutdown:
+            os.remove(shutdown_path)
+
+        with open(version_path, 'r') as f:
+            version = f.read()
+
+        os.remove(version_path)
+
+        assert not is_startup, 'startup'
+        assert not is_shutdown, 'shutdown'
+        assert version == '3.0 2.0', 'version'
+
+    def test_asgi_lifespan_failed(self):
+        self.load('lifespan/failed')
+
+        assert self.get()['status'] == 503
+
+        assert (
+            self.wait_for_record(r'\[error\].*Application startup failed')
+            is not None
+        ), 'error message'
+        assert self.wait_for_record(r'Exception blah') is not None, 'exception'
+
+    def test_asgi_lifespan_error(self):
+        self.load('lifespan/error')
+
+        self.get()
+
+        assert self.wait_for_record(r'Exception blah') is not None, 'exception'
+
+    def test_asgi_lifespan_error_auto(self):
+        self.load('lifespan/error_auto')
+
+        self.get()
+
+        assert self.wait_for_record(r'AssertionError') is not None, 'assertion'