[test] Work around random "Failed to establish a new connection" errors

mxmlnkn · Oct 12, 2024 · 06c81f5 · 06c81f5
1 parent e1d0f7c
commit 06c81f5
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tests/wait-for-port.py b/tests/wait-for-port.py
@@ -12,9 +12,12 @@
 with closing(socket.socket(socket.AF_INET, socket.SOCK_STREAM)) as sock:
     n = max(1, int(math.ceil(timeout / 5)))
     for i in range(n):
-        if sock.connect_ex(("127.0.0.1", int(sys.argv[1]))) == 0:
-            print(f"Weed port opened after {time.time() - t0:.1f} s.")
-            sys.exit(0)
+        try:
+            if sock.connect_ex(("127.0.0.1", int(sys.argv[1]))) == 0:
+                print(f"Weed port opened after {time.time() - t0:.1f} s.")
+                sys.exit(0)
+        except Exception:
+            pass
         # Do not wait after the last test.
         if i < n - 1:
             time.sleep(5)