Merge pull request #7 from ionut-arm/socket-poll

Make the mock service poll

Author: ionut-arm

parsec_mock/parsec_mock.py

@@ -41,18 +41,41 @@ def run_test(test_folder, parsec_socket):
     sock.listen(1)
 
     while True:
+        print("Waiting for connections", flush=True)
         connection, client_addr = sock.accept()
         try:
+            print("Connection received from {}".format(client_addr), flush=True)
+            # Optionally reduce the default connection timeout
+            # connection.settimeout(10)
             received_data = connection.recv(4096)
-            b64_received_data = base64.b64encode(received_data).decode("ascii")
-            if b64_received_data in test_cases:
-                (name, test_case) = test_cases[b64_received_data]
-                print("Received expected request for test case {}".format(name))
-                bin_response = base64.b64decode(test_case.test_data.response)
-                connection.sendall(bin_response)
-            else:
-                print("Received unexpected request {}".format(b64_received_data))
+            all_received_data = bytearray()
+            # Read the socket in cycle until we either
+            # - receive a correct command or
+            # - fail to receive any data
+            while received_data:
+                all_received_data += received_data
+                b64_received_data = base64.b64encode(all_received_data).decode("ascii")
+                if b64_received_data in test_cases:
+                    (name, test_case) = test_cases[b64_received_data]
+                    print(
+                        "Received expected request for test case {}".format(name),
+                        flush=True,
+                    )
+                    bin_response = base64.b64decode(test_case.test_data.response)
+                    connection.sendall(bin_response)
+                    break
+
+                print(
+                    "Received unexpected request {}".format(b64_received_data),
+                    flush=True,
+                )
+                print("Continue receiving", flush=True)
+                received_data = connection.recv(4096)
+
+            if not received_data:
+                print("No correct command received. No more data to read", flush=True)
         finally:
+            print("Closing connection", flush=True)
             connection.close()