Tests: get rid of classes in test files.
Class usage came from the unittest framework and it was always redundant after migration to the pytest. This commit removes classes from files containing tests to make them more readable and understandable.
This commit is contained in:
Andrei Zeliankou
@@ -3,23 +3,22 @@ import time
|
||||
|
||||
import pytest
|
||||
from packaging import version
|
||||
from unit.applications.lang.python import TestApplicationPython
|
||||
from unit.applications.lang.python import ApplicationPython
|
||||
|
||||
prerequisites = {
|
||||
'modules': {'python': lambda v: version.parse(v) >= version.parse('3.5')}
|
||||
}
|
||||
|
||||
client = ApplicationPython(load_module='asgi')
|
||||
|
||||
class TestASGIApplication(TestApplicationPython):
|
||||
load_module = 'asgi'
|
||||
|
||||
def test_asgi_application_variables(self, date_to_sec_epoch, sec_epoch):
|
||||
self.load('variables')
|
||||
def test_asgi_application_variables(date_to_sec_epoch, sec_epoch):
|
||||
client.load('variables')
|
||||
|
||||
body = 'Test body string.'
|
||||
body = 'Test body string.'
|
||||
|
||||
resp = self.http(
|
||||
f"""POST / HTTP/1.1
|
||||
resp = client.http(
|
||||
f"""POST / HTTP/1.1
|
||||
Host: localhost
|
||||
Content-Length: {len(body)}
|
||||
Custom-Header: blah
|
||||
@@ -29,190 +28,230 @@ Connection: close
|
||||
custom-header: BLAH
|
||||
|
||||
{body}""".encode(),
|
||||
raw=True,
|
||||
)
|
||||
raw=True,
|
||||
)
|
||||
|
||||
assert resp['status'] == 200, 'status'
|
||||
headers = resp['headers']
|
||||
header_server = headers.pop('Server')
|
||||
assert re.search(r'Unit/[\d\.]+', header_server), 'server header'
|
||||
assert resp['status'] == 200, 'status'
|
||||
headers = resp['headers']
|
||||
header_server = headers.pop('Server')
|
||||
assert re.search(r'Unit/[\d\.]+', header_server), 'server header'
|
||||
|
||||
date = headers.pop('Date')
|
||||
assert date[-4:] == ' GMT', 'date header timezone'
|
||||
assert abs(date_to_sec_epoch(date) - sec_epoch) < 5, 'date header'
|
||||
date = headers.pop('Date')
|
||||
assert date[-4:] == ' GMT', 'date header timezone'
|
||||
assert abs(date_to_sec_epoch(date) - sec_epoch) < 5, 'date header'
|
||||
|
||||
assert headers == {
|
||||
'Connection': 'close',
|
||||
'content-length': str(len(body)),
|
||||
'content-type': 'text/html',
|
||||
'request-method': 'POST',
|
||||
'request-uri': '/',
|
||||
'http-host': 'localhost',
|
||||
'http-version': '1.1',
|
||||
'custom-header': 'blah, Blah, BLAH',
|
||||
'asgi-version': '3.0',
|
||||
'asgi-spec-version': '2.1',
|
||||
'scheme': 'http',
|
||||
}, 'headers'
|
||||
assert resp['body'] == body, 'body'
|
||||
assert headers == {
|
||||
'Connection': 'close',
|
||||
'content-length': str(len(body)),
|
||||
'content-type': 'text/html',
|
||||
'request-method': 'POST',
|
||||
'request-uri': '/',
|
||||
'http-host': 'localhost',
|
||||
'http-version': '1.1',
|
||||
'custom-header': 'blah, Blah, BLAH',
|
||||
'asgi-version': '3.0',
|
||||
'asgi-spec-version': '2.1',
|
||||
'scheme': 'http',
|
||||
}, 'headers'
|
||||
assert resp['body'] == body, 'body'
|
||||
|
||||
def test_asgi_application_ipv6(self):
|
||||
self.load('empty')
|
||||
|
||||
assert 'success' in self.conf(
|
||||
{"[::1]:7080": {"pass": "applications/empty"}}, 'listeners'
|
||||
)
|
||||
def test_asgi_application_ipv6():
|
||||
client.load('empty')
|
||||
|
||||
assert self.get(sock_type='ipv6')['status'] == 200
|
||||
assert 'success' in client.conf(
|
||||
{"[::1]:7080": {"pass": "applications/empty"}}, 'listeners'
|
||||
)
|
||||
|
||||
def test_asgi_application_unix(self, temp_dir):
|
||||
self.load('empty')
|
||||
assert client.get(sock_type='ipv6')['status'] == 200
|
||||
|
||||
addr = f'{temp_dir}/sock'
|
||||
assert 'success' in self.conf(
|
||||
{f"unix:{addr}": {"pass": "applications/empty"}}, 'listeners'
|
||||
)
|
||||
|
||||
assert self.get(sock_type='unix', addr=addr)['status'] == 200
|
||||
def test_asgi_application_unix(temp_dir):
|
||||
client.load('empty')
|
||||
|
||||
def test_asgi_application_query_string(self):
|
||||
self.load('query_string')
|
||||
addr = f'{temp_dir}/sock'
|
||||
assert 'success' in client.conf(
|
||||
{f"unix:{addr}": {"pass": "applications/empty"}}, 'listeners'
|
||||
)
|
||||
|
||||
resp = self.get(url='/?var1=val1&var2=val2')
|
||||
assert client.get(sock_type='unix', addr=addr)['status'] == 200
|
||||
|
||||
assert (
|
||||
resp['headers']['query-string'] == 'var1=val1&var2=val2'
|
||||
), 'query-string header'
|
||||
|
||||
def test_asgi_application_prefix(self):
|
||||
self.load('prefix', prefix='/api/rest')
|
||||
def test_asgi_application_query_string():
|
||||
client.load('query_string')
|
||||
|
||||
def set_prefix(prefix):
|
||||
self.conf(f'"{prefix}"', 'applications/prefix/prefix')
|
||||
resp = client.get(url='/?var1=val1&var2=val2')
|
||||
|
||||
def check_prefix(url, prefix):
|
||||
resp = self.get(url=url)
|
||||
assert resp['status'] == 200
|
||||
assert resp['headers']['prefix'] == prefix
|
||||
assert (
|
||||
resp['headers']['query-string'] == 'var1=val1&var2=val2'
|
||||
), 'query-string header'
|
||||
|
||||
check_prefix('/ap', 'NULL')
|
||||
check_prefix('/api', 'NULL')
|
||||
check_prefix('/api/', 'NULL')
|
||||
check_prefix('/api/res', 'NULL')
|
||||
check_prefix('/api/restful', 'NULL')
|
||||
check_prefix('/api/rest', '/api/rest')
|
||||
check_prefix('/api/rest/', '/api/rest')
|
||||
check_prefix('/api/rest/get', '/api/rest')
|
||||
check_prefix('/api/rest/get/blah', '/api/rest')
|
||||
|
||||
set_prefix('/api/rest/')
|
||||
check_prefix('/api/rest', '/api/rest')
|
||||
check_prefix('/api/restful', 'NULL')
|
||||
check_prefix('/api/rest/', '/api/rest')
|
||||
check_prefix('/api/rest/blah', '/api/rest')
|
||||
def test_asgi_application_prefix():
|
||||
client.load('prefix', prefix='/api/rest')
|
||||
|
||||
set_prefix('/app')
|
||||
check_prefix('/ap', 'NULL')
|
||||
check_prefix('/app', '/app')
|
||||
check_prefix('/app/', '/app')
|
||||
check_prefix('/application/', 'NULL')
|
||||
def set_prefix(prefix):
|
||||
client.conf(f'"{prefix}"', 'applications/prefix/prefix')
|
||||
|
||||
set_prefix('/')
|
||||
check_prefix('/', 'NULL')
|
||||
check_prefix('/app', 'NULL')
|
||||
def check_prefix(url, prefix):
|
||||
resp = client.get(url=url)
|
||||
assert resp['status'] == 200
|
||||
assert resp['headers']['prefix'] == prefix
|
||||
|
||||
def test_asgi_application_query_string_space(self):
|
||||
self.load('query_string')
|
||||
check_prefix('/ap', 'NULL')
|
||||
check_prefix('/api', 'NULL')
|
||||
check_prefix('/api/', 'NULL')
|
||||
check_prefix('/api/res', 'NULL')
|
||||
check_prefix('/api/restful', 'NULL')
|
||||
check_prefix('/api/rest', '/api/rest')
|
||||
check_prefix('/api/rest/', '/api/rest')
|
||||
check_prefix('/api/rest/get', '/api/rest')
|
||||
check_prefix('/api/rest/get/blah', '/api/rest')
|
||||
|
||||
resp = self.get(url='/ ?var1=val1&var2=val2')
|
||||
assert (
|
||||
resp['headers']['query-string'] == 'var1=val1&var2=val2'
|
||||
), 'query-string space'
|
||||
set_prefix('/api/rest/')
|
||||
check_prefix('/api/rest', '/api/rest')
|
||||
check_prefix('/api/restful', 'NULL')
|
||||
check_prefix('/api/rest/', '/api/rest')
|
||||
check_prefix('/api/rest/blah', '/api/rest')
|
||||
|
||||
resp = self.get(url='/ %20?var1=val1&var2=val2')
|
||||
assert (
|
||||
resp['headers']['query-string'] == 'var1=val1&var2=val2'
|
||||
), 'query-string space 2'
|
||||
set_prefix('/app')
|
||||
check_prefix('/ap', 'NULL')
|
||||
check_prefix('/app', '/app')
|
||||
check_prefix('/app/', '/app')
|
||||
check_prefix('/application/', 'NULL')
|
||||
|
||||
resp = self.get(url='/ %20 ?var1=val1&var2=val2')
|
||||
assert (
|
||||
resp['headers']['query-string'] == 'var1=val1&var2=val2'
|
||||
), 'query-string space 3'
|
||||
set_prefix('/')
|
||||
check_prefix('/', 'NULL')
|
||||
check_prefix('/app', 'NULL')
|
||||
|
||||
resp = self.get(url='/blah %20 blah? var1= val1 & var2=val2')
|
||||
assert (
|
||||
resp['headers']['query-string'] == ' var1= val1 & var2=val2'
|
||||
), 'query-string space 4'
|
||||
|
||||
def test_asgi_application_query_string_empty(self):
|
||||
self.load('query_string')
|
||||
def test_asgi_application_query_string_space():
|
||||
client.load('query_string')
|
||||
|
||||
resp = self.get(url='/?')
|
||||
resp = client.get(url='/ ?var1=val1&var2=val2')
|
||||
assert (
|
||||
resp['headers']['query-string'] == 'var1=val1&var2=val2'
|
||||
), 'query-string space'
|
||||
|
||||
assert resp['status'] == 200, 'query string empty status'
|
||||
assert resp['headers']['query-string'] == '', 'query string empty'
|
||||
resp = client.get(url='/ %20?var1=val1&var2=val2')
|
||||
assert (
|
||||
resp['headers']['query-string'] == 'var1=val1&var2=val2'
|
||||
), 'query-string space 2'
|
||||
|
||||
def test_asgi_application_query_string_absent(self):
|
||||
self.load('query_string')
|
||||
resp = client.get(url='/ %20 ?var1=val1&var2=val2')
|
||||
assert (
|
||||
resp['headers']['query-string'] == 'var1=val1&var2=val2'
|
||||
), 'query-string space 3'
|
||||
|
||||
resp = self.get()
|
||||
resp = client.get(url='/blah %20 blah? var1= val1 & var2=val2')
|
||||
assert (
|
||||
resp['headers']['query-string'] == ' var1= val1 & var2=val2'
|
||||
), 'query-string space 4'
|
||||
|
||||
assert resp['status'] == 200, 'query string absent status'
|
||||
assert resp['headers']['query-string'] == '', 'query string absent'
|
||||
|
||||
@pytest.mark.skip('not yet')
|
||||
def test_asgi_application_server_port(self):
|
||||
self.load('server_port')
|
||||
def test_asgi_application_query_string_empty():
|
||||
client.load('query_string')
|
||||
|
||||
assert (
|
||||
self.get()['headers']['Server-Port'] == '7080'
|
||||
), 'Server-Port header'
|
||||
resp = client.get(url='/?')
|
||||
|
||||
@pytest.mark.skip('not yet')
|
||||
def test_asgi_application_working_directory_invalid(self):
|
||||
self.load('empty')
|
||||
assert resp['status'] == 200, 'query string empty status'
|
||||
assert resp['headers']['query-string'] == '', 'query string empty'
|
||||
|
||||
assert 'success' in self.conf(
|
||||
'"/blah"', 'applications/empty/working_directory'
|
||||
), 'configure invalid working_directory'
|
||||
|
||||
assert self.get()['status'] == 500, 'status'
|
||||
def test_asgi_application_query_string_absent():
|
||||
client.load('query_string')
|
||||
|
||||
def test_asgi_application_204_transfer_encoding(self):
|
||||
self.load('204_no_content')
|
||||
resp = client.get()
|
||||
|
||||
assert (
|
||||
'Transfer-Encoding' not in self.get()['headers']
|
||||
), '204 header transfer encoding'
|
||||
assert resp['status'] == 200, 'query string absent status'
|
||||
assert resp['headers']['query-string'] == '', 'query string absent'
|
||||
|
||||
def test_asgi_application_shm_ack_handle(self):
|
||||
# Minimum possible limit
|
||||
shm_limit = 10 * 1024 * 1024
|
||||
|
||||
self.load('mirror', limits={"shm": shm_limit})
|
||||
@pytest.mark.skip('not yet')
|
||||
def test_asgi_application_server_port():
|
||||
client.load('server_port')
|
||||
|
||||
# Should exceed shm_limit
|
||||
max_body_size = 12 * 1024 * 1024
|
||||
assert (
|
||||
client.get()['headers']['Server-Port'] == '7080'
|
||||
), 'Server-Port header'
|
||||
|
||||
assert 'success' in self.conf(
|
||||
f'{{"http":{{"max_body_size": {max_body_size} }}}}',
|
||||
'settings',
|
||||
)
|
||||
|
||||
assert self.get()['status'] == 200, 'init'
|
||||
@pytest.mark.skip('not yet')
|
||||
def test_asgi_application_working_directory_invalid():
|
||||
client.load('empty')
|
||||
|
||||
body = '0123456789AB' * 1024 * 1024 # 12 Mb
|
||||
resp = self.post(body=body, read_buffer_size=1024 * 1024)
|
||||
assert 'success' in client.conf(
|
||||
'"/blah"', 'applications/empty/working_directory'
|
||||
), 'configure invalid working_directory'
|
||||
|
||||
assert resp['body'] == body, 'keep-alive 1'
|
||||
assert client.get()['status'] == 500, 'status'
|
||||
|
||||
def test_asgi_keepalive_body(self):
|
||||
self.load('mirror')
|
||||
|
||||
assert self.get()['status'] == 200, 'init'
|
||||
def test_asgi_application_204_transfer_encoding():
|
||||
client.load('204_no_content')
|
||||
|
||||
body = '0123456789' * 500
|
||||
(resp, sock) = self.post(
|
||||
assert (
|
||||
'Transfer-Encoding' not in client.get()['headers']
|
||||
), '204 header transfer encoding'
|
||||
|
||||
|
||||
def test_asgi_application_shm_ack_handle():
|
||||
# Minimum possible limit
|
||||
shm_limit = 10 * 1024 * 1024
|
||||
|
||||
client.load('mirror', limits={"shm": shm_limit})
|
||||
|
||||
# Should exceed shm_limit
|
||||
max_body_size = 12 * 1024 * 1024
|
||||
|
||||
assert 'success' in client.conf(
|
||||
f'{{"http":{{"max_body_size": {max_body_size} }}}}',
|
||||
'settings',
|
||||
)
|
||||
|
||||
assert client.get()['status'] == 200, 'init'
|
||||
|
||||
body = '0123456789AB' * 1024 * 1024 # 12 Mb
|
||||
resp = client.post(body=body, read_buffer_size=1024 * 1024)
|
||||
|
||||
assert resp['body'] == body, 'keep-alive 1'
|
||||
|
||||
|
||||
def test_asgi_keepalive_body():
|
||||
client.load('mirror')
|
||||
|
||||
assert client.get()['status'] == 200, 'init'
|
||||
|
||||
body = '0123456789' * 500
|
||||
(resp, sock) = client.post(
|
||||
headers={
|
||||
'Host': 'localhost',
|
||||
'Connection': 'keep-alive',
|
||||
},
|
||||
start=True,
|
||||
body=body,
|
||||
read_timeout=1,
|
||||
)
|
||||
|
||||
assert resp['body'] == body, 'keep-alive 1'
|
||||
|
||||
body = '0123456789'
|
||||
resp = client.post(sock=sock, body=body)
|
||||
|
||||
assert resp['body'] == body, 'keep-alive 2'
|
||||
|
||||
|
||||
def test_asgi_keepalive_reconfigure():
|
||||
client.load('mirror')
|
||||
|
||||
assert client.get()['status'] == 200, 'init'
|
||||
|
||||
body = '0123456789'
|
||||
conns = 3
|
||||
socks = []
|
||||
|
||||
for i in range(conns):
|
||||
(resp, sock) = client.post(
|
||||
headers={
|
||||
'Host': 'localhost',
|
||||
'Connection': 'keep-alive',
|
||||
@@ -222,235 +261,216 @@ custom-header: BLAH
|
||||
read_timeout=1,
|
||||
)
|
||||
|
||||
assert resp['body'] == body, 'keep-alive 1'
|
||||
assert resp['body'] == body, 'keep-alive open'
|
||||
|
||||
body = '0123456789'
|
||||
resp = self.post(sock=sock, body=body)
|
||||
client.load('mirror', processes=i + 1)
|
||||
|
||||
assert resp['body'] == body, 'keep-alive 2'
|
||||
socks.append(sock)
|
||||
|
||||
def test_asgi_keepalive_reconfigure(self):
|
||||
self.load('mirror')
|
||||
|
||||
assert self.get()['status'] == 200, 'init'
|
||||
|
||||
body = '0123456789'
|
||||
conns = 3
|
||||
socks = []
|
||||
|
||||
for i in range(conns):
|
||||
(resp, sock) = self.post(
|
||||
headers={
|
||||
'Host': 'localhost',
|
||||
'Connection': 'keep-alive',
|
||||
},
|
||||
start=True,
|
||||
body=body,
|
||||
read_timeout=1,
|
||||
)
|
||||
|
||||
assert resp['body'] == body, 'keep-alive open'
|
||||
|
||||
self.load('mirror', processes=i + 1)
|
||||
|
||||
socks.append(sock)
|
||||
|
||||
for i in range(conns):
|
||||
(resp, sock) = self.post(
|
||||
headers={
|
||||
'Host': 'localhost',
|
||||
'Connection': 'keep-alive',
|
||||
},
|
||||
start=True,
|
||||
sock=socks[i],
|
||||
body=body,
|
||||
read_timeout=1,
|
||||
)
|
||||
|
||||
assert resp['body'] == body, 'keep-alive request'
|
||||
|
||||
self.load('mirror', processes=i + 1)
|
||||
|
||||
for i in range(conns):
|
||||
resp = self.post(sock=socks[i], body=body)
|
||||
|
||||
assert resp['body'] == body, 'keep-alive close'
|
||||
|
||||
self.load('mirror', processes=i + 1)
|
||||
|
||||
def test_asgi_keepalive_reconfigure_2(self):
|
||||
self.load('mirror')
|
||||
|
||||
assert self.get()['status'] == 200, 'init'
|
||||
|
||||
body = '0123456789'
|
||||
|
||||
(resp, sock) = self.post(
|
||||
for i in range(conns):
|
||||
(resp, sock) = client.post(
|
||||
headers={
|
||||
'Host': 'localhost',
|
||||
'Connection': 'keep-alive',
|
||||
},
|
||||
start=True,
|
||||
sock=socks[i],
|
||||
body=body,
|
||||
read_timeout=1,
|
||||
)
|
||||
|
||||
assert resp['body'] == body, 'reconfigure 2 keep-alive 1'
|
||||
assert resp['body'] == body, 'keep-alive request'
|
||||
|
||||
self.load('empty')
|
||||
client.load('mirror', processes=i + 1)
|
||||
|
||||
assert self.get()['status'] == 200, 'init'
|
||||
for i in range(conns):
|
||||
resp = client.post(sock=socks[i], body=body)
|
||||
|
||||
(resp, sock) = self.post(start=True, sock=sock, body=body)
|
||||
assert resp['body'] == body, 'keep-alive close'
|
||||
|
||||
assert resp['status'] == 200, 'reconfigure 2 keep-alive 2'
|
||||
assert resp['body'] == '', 'reconfigure 2 keep-alive 2 body'
|
||||
client.load('mirror', processes=i + 1)
|
||||
|
||||
assert 'success' in self.conf(
|
||||
{"listeners": {}, "applications": {}}
|
||||
), 'reconfigure 2 clear configuration'
|
||||
|
||||
resp = self.get(sock=sock)
|
||||
def test_asgi_keepalive_reconfigure_2():
|
||||
client.load('mirror')
|
||||
|
||||
assert resp == {}, 'reconfigure 2 keep-alive 3'
|
||||
assert client.get()['status'] == 200, 'init'
|
||||
|
||||
def test_asgi_keepalive_reconfigure_3(self):
|
||||
self.load('empty')
|
||||
body = '0123456789'
|
||||
|
||||
assert self.get()['status'] == 200, 'init'
|
||||
(resp, sock) = client.post(
|
||||
headers={
|
||||
'Host': 'localhost',
|
||||
'Connection': 'keep-alive',
|
||||
},
|
||||
start=True,
|
||||
body=body,
|
||||
read_timeout=1,
|
||||
)
|
||||
|
||||
sock = self.http(
|
||||
b"""GET / HTTP/1.1
|
||||
assert resp['body'] == body, 'reconfigure 2 keep-alive 1'
|
||||
|
||||
client.load('empty')
|
||||
|
||||
assert client.get()['status'] == 200, 'init'
|
||||
|
||||
(resp, sock) = client.post(start=True, sock=sock, body=body)
|
||||
|
||||
assert resp['status'] == 200, 'reconfigure 2 keep-alive 2'
|
||||
assert resp['body'] == '', 'reconfigure 2 keep-alive 2 body'
|
||||
|
||||
assert 'success' in client.conf(
|
||||
{"listeners": {}, "applications": {}}
|
||||
), 'reconfigure 2 clear configuration'
|
||||
|
||||
resp = client.get(sock=sock)
|
||||
|
||||
assert resp == {}, 'reconfigure 2 keep-alive 3'
|
||||
|
||||
|
||||
def test_asgi_keepalive_reconfigure_3():
|
||||
client.load('empty')
|
||||
|
||||
assert client.get()['status'] == 200, 'init'
|
||||
|
||||
sock = client.http(
|
||||
b"""GET / HTTP/1.1
|
||||
""",
|
||||
raw=True,
|
||||
no_recv=True,
|
||||
)
|
||||
raw=True,
|
||||
no_recv=True,
|
||||
)
|
||||
|
||||
assert self.get()['status'] == 200
|
||||
assert client.get()['status'] == 200
|
||||
|
||||
assert 'success' in self.conf(
|
||||
{"listeners": {}, "applications": {}}
|
||||
), 'reconfigure 3 clear configuration'
|
||||
assert 'success' in client.conf(
|
||||
{"listeners": {}, "applications": {}}
|
||||
), 'reconfigure 3 clear configuration'
|
||||
|
||||
resp = self.http(
|
||||
b"""Host: localhost
|
||||
resp = client.http(
|
||||
b"""Host: localhost
|
||||
Connection: close
|
||||
|
||||
""",
|
||||
sock=sock,
|
||||
raw=True,
|
||||
)
|
||||
sock=sock,
|
||||
raw=True,
|
||||
)
|
||||
|
||||
assert resp['status'] == 200, 'reconfigure 3'
|
||||
assert resp['status'] == 200, 'reconfigure 3'
|
||||
|
||||
def test_asgi_process_switch(self):
|
||||
self.load('delayed', processes=2)
|
||||
|
||||
self.get(
|
||||
def test_asgi_process_switch():
|
||||
client.load('delayed', processes=2)
|
||||
|
||||
client.get(
|
||||
headers={
|
||||
'Host': 'localhost',
|
||||
'Content-Length': '0',
|
||||
'X-Delay': '5',
|
||||
'Connection': 'close',
|
||||
},
|
||||
no_recv=True,
|
||||
)
|
||||
|
||||
headers_delay_1 = {
|
||||
'Connection': 'close',
|
||||
'Host': 'localhost',
|
||||
'Content-Length': '0',
|
||||
'X-Delay': '1',
|
||||
}
|
||||
|
||||
client.get(headers=headers_delay_1, no_recv=True)
|
||||
|
||||
time.sleep(0.5)
|
||||
|
||||
for _ in range(10):
|
||||
client.get(headers=headers_delay_1, no_recv=True)
|
||||
|
||||
client.get(headers=headers_delay_1)
|
||||
|
||||
|
||||
def test_asgi_application_loading_error(skip_alert):
|
||||
skip_alert(r'Python failed to import module "blah"')
|
||||
|
||||
client.load('empty', module="blah")
|
||||
|
||||
assert client.get()['status'] == 503, 'loading error'
|
||||
|
||||
|
||||
def test_asgi_application_threading(wait_for_record):
|
||||
"""wait_for_record() timeouts after 5s while every thread works at
|
||||
least 3s. So without releasing GIL test should fail.
|
||||
"""
|
||||
|
||||
client.load('threading')
|
||||
|
||||
for _ in range(10):
|
||||
client.get(no_recv=True)
|
||||
|
||||
assert (
|
||||
wait_for_record(r'\(5\) Thread: 100', wait=50) is not None
|
||||
), 'last thread finished'
|
||||
|
||||
|
||||
def test_asgi_application_threads():
|
||||
client.load('threads', threads=2)
|
||||
|
||||
socks = []
|
||||
|
||||
for _ in range(2):
|
||||
sock = client.get(
|
||||
headers={
|
||||
'Host': 'localhost',
|
||||
'Content-Length': '0',
|
||||
'X-Delay': '5',
|
||||
'X-Delay': '3',
|
||||
'Connection': 'close',
|
||||
},
|
||||
no_recv=True,
|
||||
)
|
||||
|
||||
headers_delay_1 = {
|
||||
'Connection': 'close',
|
||||
socks.append(sock)
|
||||
|
||||
time.sleep(1.0) # required to avoid greedy request reading
|
||||
|
||||
threads = set()
|
||||
|
||||
for sock in socks:
|
||||
resp = client.recvall(sock).decode('utf-8')
|
||||
|
||||
client.log_in(resp)
|
||||
|
||||
resp = client._resp_to_dict(resp)
|
||||
|
||||
assert resp['status'] == 200, 'status'
|
||||
|
||||
threads.add(resp['headers']['x-thread'])
|
||||
|
||||
sock.close()
|
||||
|
||||
assert len(socks) == len(threads), 'threads differs'
|
||||
|
||||
|
||||
def test_asgi_application_legacy():
|
||||
client.load('legacy')
|
||||
|
||||
resp = client.get(
|
||||
headers={
|
||||
'Host': 'localhost',
|
||||
'Content-Length': '0',
|
||||
'X-Delay': '1',
|
||||
}
|
||||
'Connection': 'close',
|
||||
},
|
||||
)
|
||||
|
||||
self.get(headers=headers_delay_1, no_recv=True)
|
||||
assert resp['status'] == 200, 'status'
|
||||
|
||||
time.sleep(0.5)
|
||||
|
||||
for _ in range(10):
|
||||
self.get(headers=headers_delay_1, no_recv=True)
|
||||
def test_asgi_application_legacy_force():
|
||||
client.load('legacy_force', protocol='asgi')
|
||||
|
||||
self.get(headers=headers_delay_1)
|
||||
resp = client.get(
|
||||
headers={
|
||||
'Host': 'localhost',
|
||||
'Content-Length': '0',
|
||||
'Connection': 'close',
|
||||
},
|
||||
)
|
||||
|
||||
def test_asgi_application_loading_error(self, skip_alert):
|
||||
skip_alert(r'Python failed to import module "blah"')
|
||||
|
||||
self.load('empty', module="blah")
|
||||
|
||||
assert self.get()['status'] == 503, 'loading error'
|
||||
|
||||
def test_asgi_application_threading(self, wait_for_record):
|
||||
"""wait_for_record() timeouts after 5s while every thread works at
|
||||
least 3s. So without releasing GIL test should fail.
|
||||
"""
|
||||
|
||||
self.load('threading')
|
||||
|
||||
for _ in range(10):
|
||||
self.get(no_recv=True)
|
||||
|
||||
assert (
|
||||
wait_for_record(r'\(5\) Thread: 100', wait=50) is not None
|
||||
), 'last thread finished'
|
||||
|
||||
def test_asgi_application_threads(self):
|
||||
self.load('threads', threads=2)
|
||||
|
||||
socks = []
|
||||
|
||||
for _ in range(2):
|
||||
sock = self.get(
|
||||
headers={
|
||||
'Host': 'localhost',
|
||||
'X-Delay': '3',
|
||||
'Connection': 'close',
|
||||
},
|
||||
no_recv=True,
|
||||
)
|
||||
|
||||
socks.append(sock)
|
||||
|
||||
time.sleep(1.0) # required to avoid greedy request reading
|
||||
|
||||
threads = set()
|
||||
|
||||
for sock in socks:
|
||||
resp = self.recvall(sock).decode('utf-8')
|
||||
|
||||
self.log_in(resp)
|
||||
|
||||
resp = self._resp_to_dict(resp)
|
||||
|
||||
assert resp['status'] == 200, 'status'
|
||||
|
||||
threads.add(resp['headers']['x-thread'])
|
||||
|
||||
sock.close()
|
||||
|
||||
assert len(socks) == len(threads), 'threads differs'
|
||||
|
||||
def test_asgi_application_legacy(self):
|
||||
self.load('legacy')
|
||||
|
||||
resp = self.get(
|
||||
headers={
|
||||
'Host': 'localhost',
|
||||
'Content-Length': '0',
|
||||
'Connection': 'close',
|
||||
},
|
||||
)
|
||||
|
||||
assert resp['status'] == 200, 'status'
|
||||
|
||||
def test_asgi_application_legacy_force(self):
|
||||
self.load('legacy_force', protocol='asgi')
|
||||
|
||||
resp = self.get(
|
||||
headers={
|
||||
'Host': 'localhost',
|
||||
'Content-Length': '0',
|
||||
'Connection': 'close',
|
||||
},
|
||||
)
|
||||
|
||||
assert resp['status'] == 200, 'status'
|
||||
assert resp['status'] == 200, 'status'
|
||||
|
||||
Reference in New Issue
Block a user