From 1cfd329b3d2746e73432e8007d429e565c6b5fa0 Mon Sep 17 00:00:00 2001 From: Andrey Zelenkov Date: Mon, 2 Sep 2019 18:03:33 +0300 Subject: Tests: fixed recv_bytes() in websockets.py. --- test/unit/applications/websockets.py | 12 +++--------- 1 file changed, 3 insertions(+), 9 deletions(-) (limited to 'test/unit/applications/websockets.py') diff --git a/test/unit/applications/websockets.py b/test/unit/applications/websockets.py index 3553af6e..5f9c3c51 100644 --- a/test/unit/applications/websockets.py +++ b/test/unit/applications/websockets.py @@ -61,15 +61,9 @@ class TestApplicationWebsocket(TestApplicationProto): def recv_bytes(sock, bytes): data = b'' while select.select([sock], [], [], read_timeout)[0]: - try: - if bytes < 65536: - data = sock.recv(bytes) - else: - data = self.recvall( - sock, read_timeout=read_timeout, buff_size=bytes - ) - break - except: + data += sock.recv(bytes - len(data)) + + if len(data) == bytes: break return data -- cgit