From d58753344638992a63f4fe8e516c9f55a9a7f027 Mon Sep 17 00:00:00 2001 From: lovedboy Date: Fri, 21 Oct 2016 16:37:46 +0800 Subject: [PATCH] little modification in consume_socket_content --- tests/testserver/server.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/testserver/server.py b/tests/testserver/server.py index 93b6522a..6a1dcaa5 100644 --- a/tests/testserver/server.py +++ b/tests/testserver/server.py @@ -8,17 +8,17 @@ import select def consume_socket_content(sock, timeout=0.5): chunks = 65536 content = b'' - more_to_read = select.select([sock], [], [], timeout)[0] - while more_to_read: + while True: + more_to_read = select.select([sock], [], [], timeout)[0] + if not more_to_read: + break + new_content = sock.recv(chunks) - if not new_content: break content += new_content - # stop reading if no new data is received for a while - more_to_read = select.select([sock], [], [], timeout)[0] return content