Tests: handle ConnectionRefusedError in TestUnitHTTP.http().

This commit is contained in:
Andrey Zelenkov
2018-02-20 20:34:46 +03:00
parent 09f2009df5
commit 6d79c559b5

View File

@@ -154,10 +154,12 @@ class TestUnitHTTP(TestUnit):
if 'sock' not in kwargs: if 'sock' not in kwargs:
sock = socket.socket(sock_types[sock_type], socket.SOCK_STREAM) sock = socket.socket(sock_types[sock_type], socket.SOCK_STREAM)
if sock_type == 'unix': connect_args = addr if sock_type == 'unix' else (addr, port)
sock.connect(addr) try:
else: sock.connect(connect_args)
sock.connect((addr, port)) except ConnectionRefusedError:
sock.close()
return None
else: else:
sock = kwargs['sock'] sock = kwargs['sock']