summaryrefslogtreecommitdiffhomepage
path: root/test/python
diff options
context:
space:
mode:
Diffstat (limited to 'test/python')
-rw-r--r--test/python/body_bytearray/asgi.py20
-rw-r--r--test/python/body_generate/wsgi.py4
-rw-r--r--test/python/delayed/asgi.py4
-rw-r--r--test/python/environment/wsgi.py6
-rw-r--r--test/python/iter_exception/wsgi.py4
-rw-r--r--test/python/legacy/asgi.py4
-rw-r--r--test/python/legacy_force/asgi.py9
-rw-r--r--test/python/lifespan/empty/asgi.py2
-rw-r--r--test/python/lifespan/failed/asgi.py2
-rw-r--r--test/python/restart/longstart.py1
-rw-r--r--test/python/unicode/wsgi.py2
-rw-r--r--test/python/user_group/wsgi.py7
12 files changed, 48 insertions, 17 deletions
diff --git a/test/python/body_bytearray/asgi.py b/test/python/body_bytearray/asgi.py
new file mode 100644
index 00000000..6d2f402f
--- /dev/null
+++ b/test/python/body_bytearray/asgi.py
@@ -0,0 +1,20 @@
+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):
+ body = bytearray(body)
+ 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/body_generate/wsgi.py b/test/python/body_generate/wsgi.py
index 73462be6..3d12ac60 100644
--- a/test/python/body_generate/wsgi.py
+++ b/test/python/body_generate/wsgi.py
@@ -1,6 +1,6 @@
def application(env, start_response):
length = env.get('HTTP_X_LENGTH', '10')
- bytes = b'X' * int(length)
+ body = b'X' * int(length)
start_response('200', [('Content-Length', length)])
- return [bytes]
+ return [body]
diff --git a/test/python/delayed/asgi.py b/test/python/delayed/asgi.py
index 1cb15a92..3c25e49a 100644
--- a/test/python/delayed/asgi.py
+++ b/test/python/delayed/asgi.py
@@ -33,7 +33,9 @@ async def application(scope, receive, send):
{
'type': 'http.response.start',
'status': 200,
- 'headers': [(b'content-length', str(len(body)).encode()),],
+ 'headers': [
+ (b'content-length', str(len(body)).encode()),
+ ],
}
)
diff --git a/test/python/environment/wsgi.py b/test/python/environment/wsgi.py
index 91e0ba49..622b8bc0 100644
--- a/test/python/environment/wsgi.py
+++ b/test/python/environment/wsgi.py
@@ -2,9 +2,11 @@ import os
def application(env, start_response):
- vars = env.get('HTTP_X_VARIABLES').split(',')
+ variables = env.get('HTTP_X_VARIABLES').split(',')
- body = ','.join([str(os.environ[var]) for var in vars if var in os.environ])
+ body = ','.join(
+ [str(os.environ[var]) for var in variables if var in os.environ]
+ )
body = body.encode()
start_response('200', [('Content-Length', str(len(body)))])
diff --git a/test/python/iter_exception/wsgi.py b/test/python/iter_exception/wsgi.py
index 2779a845..66a09af7 100644
--- a/test/python/iter_exception/wsgi.py
+++ b/test/python/iter_exception/wsgi.py
@@ -8,9 +8,7 @@ class application:
def __iter__(self):
self.__i = 0
self._skip_level = int(self.environ.get('HTTP_X_SKIP', 0))
- self._not_skip_close = int(
- self.environ.get('HTTP_X_NOT_SKIP_CLOSE', 0)
- )
+ self._not_skip_close = int(self.environ.get('HTTP_X_NOT_SKIP_CLOSE', 0))
self._is_chunked = self.environ.get('HTTP_X_CHUNKED')
headers = [(('Content-Length', '10'))]
diff --git a/test/python/legacy/asgi.py b/test/python/legacy/asgi.py
index 1d45cc4f..8be8932e 100644
--- a/test/python/legacy/asgi.py
+++ b/test/python/legacy/asgi.py
@@ -9,6 +9,8 @@ async def app_http(receive, send):
{
'type': 'http.response.start',
'status': 200,
- 'headers': [(b'content-length', b'0'),],
+ 'headers': [
+ (b'content-length', b'0'),
+ ],
}
)
diff --git a/test/python/legacy_force/asgi.py b/test/python/legacy_force/asgi.py
index ad2785f2..56c7061d 100644
--- a/test/python/legacy_force/asgi.py
+++ b/test/python/legacy_force/asgi.py
@@ -1,11 +1,10 @@
def application(scope, receive=None, send=None):
assert scope['type'] == 'http'
- if receive == None and send == None:
+ if receive is None and send is None:
return app_http
- else:
- return app_http(receive, send)
+ return app_http(receive, send)
async def app_http(receive, send):
@@ -13,6 +12,8 @@ async def app_http(receive, send):
{
'type': 'http.response.start',
'status': 200,
- 'headers': [(b'content-length', b'0'),],
+ 'headers': [
+ (b'content-length', b'0'),
+ ],
}
)
diff --git a/test/python/lifespan/empty/asgi.py b/test/python/lifespan/empty/asgi.py
index 27395a28..2071e6e0 100644
--- a/test/python/lifespan/empty/asgi.py
+++ b/test/python/lifespan/empty/asgi.py
@@ -3,7 +3,7 @@ import os
async def handler(prefix, scope, receive, send):
if scope['type'] == 'lifespan':
- with open(f'{prefix}version', 'w+') as f:
+ with open(f'{prefix}version', 'w+', encoding='utf-8') as f:
f.write(
f"{scope['asgi']['version']} {scope['asgi']['spec_version']}"
)
diff --git a/test/python/lifespan/failed/asgi.py b/test/python/lifespan/failed/asgi.py
index 8f315f70..fbd006f9 100644
--- a/test/python/lifespan/failed/asgi.py
+++ b/test/python/lifespan/failed/asgi.py
@@ -6,6 +6,6 @@ async def application(scope, receive, send):
await send({"type": "lifespan.startup.failed"})
raise Exception('Exception blah')
- elif message['type'] == 'lifespan.shutdown':
+ if message['type'] == 'lifespan.shutdown':
await send({'type': 'lifespan.shutdown.complete'})
return
diff --git a/test/python/restart/longstart.py b/test/python/restart/longstart.py
index 777398ac..3bc383b7 100644
--- a/test/python/restart/longstart.py
+++ b/test/python/restart/longstart.py
@@ -3,6 +3,7 @@ import time
time.sleep(2)
+
def application(environ, start_response):
body = str(os.getpid()).encode()
diff --git a/test/python/unicode/wsgi.py b/test/python/unicode/wsgi.py
index f2f85f5d..8c9a59dd 100644
--- a/test/python/unicode/wsgi.py
+++ b/test/python/unicode/wsgi.py
@@ -1,7 +1,7 @@
def application(environ, start_response):
temp_dir = environ.get('HTTP_TEMP_DIR')
- with open(f'{temp_dir}/tempfile', 'w') as f:
+ with open(f'{temp_dir}/tempfile', 'w', encoding='utf-8') as f:
f.write('\u26a0\ufe0f')
start_response('200', [('Content-Length', '0')])
diff --git a/test/python/user_group/wsgi.py b/test/python/user_group/wsgi.py
index 4003c064..8f3ba50d 100644
--- a/test/python/user_group/wsgi.py
+++ b/test/python/user_group/wsgi.py
@@ -6,7 +6,12 @@ def application(environ, start_response):
uid = os.geteuid()
gid = os.getegid()
- out = json.dumps({'UID': uid, 'GID': gid,}).encode('utf-8')
+ out = json.dumps(
+ {
+ 'UID': uid,
+ 'GID': gid,
+ }
+ ).encode('utf-8')
start_response(
'200 OK',