Custom socket connect timeout in Session()

SatelliteQE · Apr 30, 2023 · aa6296e · aa6296e
1 parent 33c9741
commit aa6296e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/broker/session.py b/broker/session.py
@@ -31,7 +31,8 @@ def __init__(self, **kwargs):
         sock.settimeout(kwargs.get("timeout"))
         port = kwargs.get("port", 22)
         key_filename = kwargs.get("key_filename")
-        helpers.simple_retry(sock.connect, [(host, port)])
+        socket_connect_timeout = kwargs.get("socket_connect_timeout", 60)
+        helpers.simple_retry(sock.connect, [(host, port)], max_timeout=socket_connect_timeout)
         self.session = ssh2_Session()
         self.session.handshake(sock)
         if key_filename: