diff options
Diffstat (limited to 'test')
-rw-r--r-- | test/unit.py | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/test/unit.py b/test/unit.py index 2d50d64b..48b5497d 100644 --- 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'] |