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

Tests: increased read_timeout for websockets tests.

Browse Source
This commit is contained in:
Andrey Zelenkov
2019-09-03 21:23:32 +03:00
parent daadc2e00b
commit 3e23afb0d2
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -57,7 +57,7 @@ class TestApplicationWebsocket(TestApplicationProto):
def serialize_close(self, code=1000, reason=''): def serialize_close(self, code=1000, reason=''):
return struct.pack('!H', code) + reason.encode('utf-8') return struct.pack('!H', code) + reason.encode('utf-8')
def frame_read(self, sock, read_timeout=10): def frame_read(self, sock, read_timeout=30):
def recv_bytes(sock, bytes): def recv_bytes(sock, bytes):
data = b'' data = b''
while select.select([sock], [], [], read_timeout)[0]: while select.select([sock], [], [], read_timeout)[0]:
@@ -91,7 +91,11 @@ class TestApplicationWebsocket(TestApplicationProto):
if frame['mask']: if frame['mask']:
mask_bits = recv_bytes(sock, 4) mask_bits = recv_bytes(sock, 4)
data = recv_bytes(sock, length) data = b''
if length != 0:
data = recv_bytes(sock, length)
if frame['mask']: if frame['mask']:
data = self.apply_mask(data, mask_bits) data = self.apply_mask(data, mask_bits)