little modification in consume_socket_content

1 files 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:
-        new_content = sock.recv(chunks)
+    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