technocloud-public/nginx-unit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Tests: added notification on "read_timeout" expiration.

Browse Source
This commit is contained in:
Andrei Zeliankou
2020-04-03 01:46:59 +01:00
parent 2bb8b3d88a
commit d7aa514d6a
2 changed files with 30 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
test/unit/applications/websockets.py
View File

@@ -52,7 +52,11 @@ class TestApplicationWebsocket(TestApplicationProto):
)
resp = ''
while select.select([sock], [], [], 60)[0]:
while True:
rlist = select.select([sock], [], [], 60)[0]
if not rlist:
self.fail('Can\'t read response from server.')
resp += sock.recv(4096).decode()
if (
@@ -73,7 +77,15 @@ class TestApplicationWebsocket(TestApplicationProto):
def frame_read(self, sock, read_timeout=60):
def recv_bytes(sock, bytes):
data = b''
while select.select([sock], [], [], read_timeout)[0]:
while True:
rlist = select.select([sock], [], [], read_timeout)[0]
if not rlist:
# For all current cases if the "read_timeout" was changed
# than test do not expect to get a response from server.
if read_timeout == 60:
self.fail('Can\'t read response from server.')
break
data += sock.recv(bytes - len(data))
if len(data) == bytes: