Tests: handle ConnectionRefusedError in TestUnitHTTP.http().

2018-02-20 20:34:46 +03:00
parent 09f2009df5
commit 6d79c559b5
1 changed files with 6 additions and 4 deletions
--- a/test/unit.py
+++ b/test/unit.py
@@ -154,10 +154,12 @@ class TestUnitHTTP(TestUnit):
        if 'sock' not in kwargs:
            sock = socket.socket(sock_types[sock_type], socket.SOCK_STREAM)

-            if sock_type == 'unix':
-                sock.connect(addr)
-            else:
-                sock.connect((addr, port))
+            connect_args = addr if sock_type == 'unix' else (addr, port)
+            try:
+                sock.connect(connect_args)
+            except ConnectionRefusedError:
+                sock.close()
+                return None

        else:
            sock = kwargs['sock']