From 0fd8dd1442a781c67d3f3e144f1e4ae321dae7f7 Mon Sep 17 00:00:00 2001 From: Ofir Amir Date: Thu, 5 Jul 2018 17:21:08 +0300 Subject: [PATCH] Allow to pass connection timeout in ssh run script --- py/rackattack/ssh/run.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/py/rackattack/ssh/run.py b/py/rackattack/ssh/run.py index b188b44..f837153 100644 --- a/py/rackattack/ssh/run.py +++ b/py/rackattack/ssh/run.py @@ -14,9 +14,9 @@ def script(self, bashScript, verbose=True, outputTimeout=20 * 60): e.args += ('When running bash script "%s"' % bashScript), raise - def execute(self, command, outputTimeout=20 * 60, wrapCmd=True, verbose=True): + def execute(self, command, outputTimeout=20 * 60, wrapCmd=True, verbose=True, connectionTimeout=60): transport = self._sshClient.get_transport() - chan = transport.open_session() + chan = transport.open_session(timeout=connectionTimeout) commandToExecute = self._wrapCommand(command) if wrapCmd else command try: if verbose: