summaryrefslogtreecommitdiffhomepage
path: root/test/test_settings.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/test_settings.py')
-rw-r--r--test/test_settings.py199
1 files changed, 126 insertions, 73 deletions
diff --git a/test/test_settings.py b/test/test_settings.py
index c59ca8b9..49041b62 100644
--- a/test/test_settings.py
+++ b/test/test_settings.py
@@ -3,7 +3,9 @@ import socket
import time
import pytest
+
from unit.applications.lang.python import TestApplicationPython
+from unit.utils import sysctl
class TestSettings(TestApplicationPython):
@@ -12,33 +14,42 @@ class TestSettings(TestApplicationPython):
def test_settings_header_read_timeout(self):
self.load('empty')
- self.conf({'http': {'header_read_timeout': 2}}, 'settings')
-
- (resp, sock) = self.http(
- b"""GET / HTTP/1.1
+ def req():
+ (resp, sock) = self.http(
+ b"""GET / HTTP/1.1
""",
- start=True,
- read_timeout=1,
- raw=True,
- )
+ start=True,
+ read_timeout=1,
+ raw=True,
+ )
- time.sleep(3)
+ time.sleep(3)
- resp = self.http(
- b"""Host: localhost
+ return self.http(
+ b"""Host: localhost
Connection: close
-""",
- sock=sock,
- raw=True,
+ """,
+ sock=sock,
+ raw=True,
+ )
+
+ assert 'success' in self.conf(
+ {'http': {'header_read_timeout': 2}}, 'settings'
)
+ assert req()['status'] == 408, 'status header read timeout'
- assert resp['status'] == 408, 'status header read timeout'
+ assert 'success' in self.conf(
+ {'http': {'header_read_timeout': 7}}, 'settings'
+ )
+ assert req()['status'] == 200, 'status header read timeout 2'
def test_settings_header_read_timeout_update(self):
self.load('empty')
- self.conf({'http': {'header_read_timeout': 4}}, 'settings')
+ assert 'success' in self.conf(
+ {'http': {'header_read_timeout': 4}}, 'settings'
+ )
(resp, sock) = self.http(
b"""GET / HTTP/1.1
@@ -89,31 +100,40 @@ Connection: close
def test_settings_body_read_timeout(self):
self.load('empty')
- self.conf({'http': {'body_read_timeout': 2}}, 'settings')
-
- (resp, sock) = self.http(
- b"""POST / HTTP/1.1
+ def req():
+ (resp, sock) = self.http(
+ b"""POST / HTTP/1.1
Host: localhost
Content-Length: 10
Connection: close
""",
- start=True,
- raw_resp=True,
- read_timeout=1,
- raw=True,
- )
+ start=True,
+ raw_resp=True,
+ read_timeout=1,
+ raw=True,
+ )
- time.sleep(3)
+ time.sleep(3)
- resp = self.http(b"""0123456789""", sock=sock, raw=True)
+ return self.http(b"""0123456789""", sock=sock, raw=True)
- assert resp['status'] == 408, 'status body read timeout'
+ assert 'success' in self.conf(
+ {'http': {'body_read_timeout': 2}}, 'settings'
+ )
+ assert req()['status'] == 408, 'status body read timeout'
+
+ assert 'success' in self.conf(
+ {'http': {'body_read_timeout': 7}}, 'settings'
+ )
+ assert req()['status'] == 200, 'status body read timeout 2'
def test_settings_body_read_timeout_update(self):
self.load('empty')
- self.conf({'http': {'body_read_timeout': 4}}, 'settings')
+ assert 'success' in self.conf(
+ {'http': {'body_read_timeout': 4}}, 'settings'
+ )
(resp, sock) = self.http(
b"""POST / HTTP/1.1
@@ -146,85 +166,116 @@ Connection: close
assert resp['status'] == 200, 'status body read timeout update'
def test_settings_send_timeout(self, temp_dir):
- self.load('mirror')
+ self.load('body_generate')
- data_len = 1048576
+ def req(addr, data_len):
+ sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
+ sock.connect(addr)
- self.conf({'http': {'send_timeout': 1}}, 'settings')
+ req = (
+ """GET / HTTP/1.1
+Host: localhost
+X-Length: %d
+Connection: close
- addr = temp_dir + '/sock'
+"""
+ % data_len
+ )
- self.conf({"unix:" + addr: {'application': 'mirror'}}, 'listeners')
+ sock.sendall(req.encode())
- sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
- sock.connect(addr)
+ data = sock.recv(16).decode()
- req = """POST / HTTP/1.1
-Host: localhost
-Content-Type: text/html
-Content-Length: %d
-Connection: close
+ time.sleep(3)
-""" % data_len + (
- 'X' * data_len
+ data += self.recvall(sock).decode()
+
+ sock.close()
+
+ return data
+
+ sysctl_out = sysctl()
+ values = re.findall(
+ r'net.core.[rw]mem_(?:max|default).*?(\d+)', sysctl_out
)
+ values = [int(v) for v in values]
- sock.sendall(req.encode())
+ data_len = 1048576 if len(values) == 0 else 10 * max(values)
- data = sock.recv(16).decode()
+ addr = temp_dir + '/sock'
- time.sleep(3)
+ assert 'success' in self.conf(
+ {"unix:" + addr: {'application': 'body_generate'}}, 'listeners'
+ )
- data += self.recvall(sock).decode()
+ assert 'success' in self.conf(
+ {'http': {'send_timeout': 1}}, 'settings'
+ )
- sock.close()
+ data = req(addr, data_len)
+ assert re.search(r'200 OK', data), 'send timeout status'
+ assert len(data) < data_len, 'send timeout data '
- assert re.search(r'200 OK', data), 'status send timeout'
- assert len(data) < data_len, 'data send timeout'
+ self.conf({'http': {'send_timeout': 7}}, 'settings')
+
+ data = req(addr, data_len)
+ assert re.search(r'200 OK', data), 'send timeout status 2'
+ assert len(data) > data_len, 'send timeout data 2'
def test_settings_idle_timeout(self):
self.load('empty')
- assert self.get()['status'] == 200, 'init'
+ def req():
+ (resp, sock) = self.get(
+ headers={'Host': 'localhost', 'Connection': 'keep-alive'},
+ start=True,
+ read_timeout=1,
+ )
- self.conf({'http': {'idle_timeout': 2}}, 'settings')
+ time.sleep(3)
- (resp, sock) = self.get(
- headers={'Host': 'localhost', 'Connection': 'keep-alive'},
- start=True,
- read_timeout=1,
- )
+ return self.get(sock=sock)
- time.sleep(3)
+ assert self.get()['status'] == 200, 'init'
- resp = self.get(
- headers={'Host': 'localhost', 'Connection': 'close'}, sock=sock
+ assert 'success' in self.conf(
+ {'http': {'idle_timeout': 2}}, 'settings'
)
+ assert req()['status'] == 408, 'status idle timeout'
- assert resp['status'] == 408, 'status idle timeout'
+ assert 'success' in self.conf(
+ {'http': {'idle_timeout': 7}}, 'settings'
+ )
+ assert req()['status'] == 200, 'status idle timeout 2'
def test_settings_idle_timeout_2(self):
self.load('empty')
- assert self.get()['status'] == 200, 'init'
+ def req():
+ _, sock = self.http(b'', start=True, raw=True, no_recv=True)
- self.conf({'http': {'idle_timeout': 1}}, 'settings')
+ time.sleep(3)
- _, sock = self.http(b'', start=True, raw=True, no_recv=True)
+ return self.get(sock=sock)
- time.sleep(3)
+ assert self.get()['status'] == 200, 'init'
- assert (
- self.get(
- headers={'Host': 'localhost', 'Connection': 'close'}, sock=sock
- )['status']
- == 408
- ), 'status idle timeout'
+ assert 'success' in self.conf(
+ {'http': {'idle_timeout': 1}}, 'settings'
+ )
+ assert req()['status'] == 408, 'status idle timeout'
+
+ assert 'success' in self.conf(
+ {'http': {'idle_timeout': 7}}, 'settings'
+ )
+ assert req()['status'] == 200, 'status idle timeout 2'
def test_settings_max_body_size(self):
self.load('empty')
- self.conf({'http': {'max_body_size': 5}}, 'settings')
+ assert 'success' in self.conf(
+ {'http': {'max_body_size': 5}}, 'settings'
+ )
assert self.post(body='01234')['status'] == 200, 'status size'
assert self.post(body='012345')['status'] == 413, 'status size max'
@@ -232,7 +283,9 @@ Connection: close
def test_settings_max_body_size_large(self):
self.load('mirror')
- self.conf({'http': {'max_body_size': 32 * 1024 * 1024}}, 'settings')
+ assert 'success' in self.conf(
+ {'http': {'max_body_size': 32 * 1024 * 1024}}, 'settings'
+ )
body = '0123456789abcdef' * 4 * 64 * 1024
resp = self.post(body=body, read_buffer_size=1024 * 1024)
generated by cgit (git 2.34.1) at 2024-11-16 21:35:15 +0000