From 6d79c559b5cebeb2ad7c5dbdb779e3bec26fc08a Mon Sep 17 00:00:00 2001 From: Andrey Zelenkov Date: Tue, 20 Feb 2018 20:34:46 +0300 Subject: [PATCH] Tests: handle ConnectionRefusedError in TestUnitHTTP.http(). --- test/unit.py | 10 ++++++---- 1 file 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']