Tests: added ASGI HTTP applications.

This commit is contained in:
Max Romanov
2020-10-01 23:55:43 +03:00
parent d97e3a3296
commit 12f225a43a
9 changed files with 598 additions and 0 deletions

View File

@@ -0,0 +1,8 @@
async def application(scope, receive, send):
assert scope['type'] == 'http'
await send({
'type': 'http.response.start',
'status': 204,
'headers': [],
})

View File

@@ -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)

10
test/python/empty/asgi.py Normal file
View File

@@ -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'),
]
})

View File

@@ -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,
})

View File

@@ -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']),
]
})

View File

@@ -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()),
]
})

View File

@@ -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'),
]
})

View File

@@ -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,
})