X-Git-Url: https://gerrit.fd.io/r/gitweb?a=blobdiff_plain;f=resources%2Ftools%2Ftopo_installation.py;h=0488bdae690d7f5da818340203eddca6dc543278;hb=66435e950bb418da98c4371a259fc4a19addda7e;hp=2931926b15ab97bb0e048c23a3570758a01f77f6;hpb=f8b2599f5a64ced7b5232df9a3305ece5d1b4c01;p=csit.git

diff --git a/resources/tools/topo_installation.py b/resources/tools/topo_installation.py
index 2931926b15..0488bdae69 100755
--- a/resources/tools/topo_installation.py
+++ b/resources/tools/topo_installation.py
@@ -22,6 +22,7 @@ from yaml import load
 
 from resources.libraries.python.ssh import SSH
 
+
 def ssh_no_error(ssh, cmd, sudo=False):
     """Execute a command over ssh channel, and log and exit if the command
     fails.
@@ -34,6 +35,31 @@ def ssh_no_error(ssh, cmd, sudo=False):
     :rtype: str
     """
 
+    if sudo:
+        ret, stdo, stde = ssh.exec_command_sudo(cmd, timeout=60)
+    else:
+        ret, stdo, stde = ssh.exec_command(cmd, timeout=60)
+
+    if ret != 0:
+        print 'Command execution failed: "{}"'.format(cmd)
+        print 'stdout: {0}'.format(stdo)
+        print 'stderr: {0}'.format(stde)
+        raise RuntimeError('Unexpected ssh command failure')
+
+    return stdo
+
+
+def ssh_ignore_error(ssh, cmd, sudo=False):
+    """Execute a command over ssh channel, ignore errors.
+
+    :param ssh: SSH() object connected to a node.
+    :param cmd: Command line to execute on remote node.
+    :type ssh: SSH() object
+    :type cmd: str
+    :return: stdout from the SSH command.
+    :rtype: str
+    """
+
     if sudo:
         ret, stdo, stde = ssh.exec_command_sudo(cmd)
     else:
@@ -43,7 +69,6 @@ def ssh_no_error(ssh, cmd, sudo=False):
         print 'Command execution failed: "{}"'.format(cmd)
         print 'stdout: {0}'.format(stdo)
         print 'stderr: {0}'.format(stde)
-        raise RuntimeError('Unexpected ssh command failure')
 
     return stdo
 
@@ -78,7 +103,7 @@ def main():
             if cancel_installation:
                 # Remove installation directory on DUT
                 cmd = "rm -r {}".format(install_dir)
-                stdout = ssh_no_error(ssh, cmd)
+                stdout = ssh_ignore_error(ssh, cmd)
                 print "###TI {}".format(stdout)
 
                 cmd = "dpkg -l | grep vpp"
@@ -94,7 +119,7 @@ def main():
                     print "###TI {}".format(stdout)
             else:
                 # Create installation directory on DUT
-                cmd = "mkdir {}".format(install_dir)
+                cmd = "rm -r {0}; mkdir {0}".format(install_dir)
                 stdout = ssh_no_error(ssh, cmd)
                 print "###TI {}".format(stdout)