Tests: read_timeout option introduced.

Also, increased default select() timeout from 1s to 5s.
This commit is contained in:
Andrey Zelenkov
2019-02-27 19:43:14 +03:00
parent 4de2c8b567
commit bd77c9a4d2
3 changed files with 16 additions and 13 deletions

View File

@@ -226,7 +226,7 @@ class TestUnitPythonProcman(unit.TestUnitApplicationPython):
(resp, sock) = self.get(headers={
'Host': 'localhost',
'Connection': 'keep-alive'
}, start=True)
}, start=True, read_timeout=1)
self.assertEqual(len(self.pids_for_process()), 1,
'keepalive connection 1')

View File

@@ -14,7 +14,7 @@ class TestUnitSettings(unit.TestUnitApplicationPython):
self.conf({'http': { 'header_read_timeout': 2 }}, 'settings')
(resp, sock) = self.http(b"""GET / HTTP/1.1
""", start=True, raw=True)
""", start=True, read_timeout=1, raw=True)
time.sleep(3)
@@ -31,17 +31,17 @@ Connection: close
self.conf({'http': { 'header_read_timeout': 4 }}, 'settings')
(resp, sock) = self.http(b"""GET / HTTP/1.1
""", start=True, raw=True, no_recv=True)
""", start=True, read_timeout=1, raw=True, no_recv=True)
time.sleep(2)
(resp, sock) = self.http(b"""Host: localhost
""", start=True, sock=sock, raw=True, no_recv=True)
""", start=True, sock=sock, read_timeout=1, raw=True, no_recv=True)
time.sleep(2)
(resp, sock) = self.http(b"""X-Blah: blah
""", start=True, sock=sock, raw=True)
""", start=True, sock=sock, read_timeout=1, raw=True)
if len(resp) != 0:
sock.close()
@@ -66,7 +66,7 @@ Host: localhost
Content-Length: 10
Connection: close
""", start=True, raw_resp=True, raw=True)
""", start=True, raw_resp=True, read_timeout=1, raw=True)
time.sleep(3)
@@ -84,15 +84,17 @@ Host: localhost
Content-Length: 10
Connection: close
""", start=True, raw=True)
""", start=True, read_timeout=1, raw=True)
time.sleep(2)
(resp, sock) = self.http(b"""012""", start=True, sock=sock, raw=True)
(resp, sock) = self.http(b"""012""", start=True, sock=sock,
read_timeout=1, raw=True)
time.sleep(2)
(resp, sock) = self.http(b"""345""", start=True, sock=sock, raw=True)
(resp, sock) = self.http(b"""345""", start=True, sock=sock,
read_timeout=1, raw=True)
time.sleep(2)
@@ -143,7 +145,7 @@ Connection: close
(resp, sock) = self.get(headers={
'Host': 'localhost',
'Connection': 'keep-alive'
}, start=True)
}, start=True, read_timeout=1)
time.sleep(3)

View File

@@ -357,7 +357,8 @@ class TestUnitHTTP(TestUnit):
if 'no_recv' not in kwargs:
enc = 'utf-8' if 'encoding' not in kwargs else kwargs['encoding']
resp = self.recvall(sock).decode(enc)
read_timeout = 5 if 'read_timeout' not in kwargs else kwargs['read_timeout']
resp = self.recvall(sock, read_timeout=read_timeout).decode(enc)
if TestUnit.detailed:
print('<<<', resp.encode('utf-8'), sep='\n')
@@ -383,9 +384,9 @@ class TestUnitHTTP(TestUnit):
def put(self, **kwargs):
return self.http('PUT', **kwargs)
def recvall(self, sock, buff_size=4096):
def recvall(self, sock, read_timeout=5, buff_size=4096):
data = b''
while select.select([sock], [], [], 1)[0]:
while select.select([sock], [], [], read_timeout)[0]:
try:
part = sock.recv(buff_size)
except: