X-Git-Url: https://gerrit.opnfv.org/gerrit/gitweb?a=blobdiff_plain;f=doctor_tests%2Fcommon%2Futils.py;h=67ca4f4b764e1cf8b972274966ac32ae44897085;hb=d8eb12f4200c21f569df5bc01d378a846b4c0db0;hp=01889d9ae5c232e9a069822b542e0426638916fe;hpb=3e882aeb6b55ceef3bd3c20155f0014a7b8c058e;p=doctor.git

diff --git a/doctor_tests/common/utils.py b/doctor_tests/common/utils.py
index 01889d9a..67ca4f4b 100644
--- a/doctor_tests/common/utils.py
+++ b/doctor_tests/common/utils.py
@@ -10,6 +10,7 @@ import json
 import os
 import paramiko
 import re
+import subprocess
 
 
 def load_json_file(full_path):
@@ -67,7 +68,7 @@ class SSHClient(object):
     def __del__(self):
         self.client.close()
 
-    def ssh(self, command):
+    def ssh(self, command, raise_enabled=True):
         if self.log:
             self.log.info("Executing: %s" % command)
         stdin, stdout, stderr = self.client.exec_command(command)
@@ -75,9 +76,10 @@ class SSHClient(object):
         output = list()
         for line in stdout.read().splitlines():
             output.append(line.decode('utf-8'))
-        if ret:
+        if ret and raise_enabled:
             if self.log:
-                self.log.info("*** FAILED to run command %s (%s)" % (command, ret))
+                self.log.info("*** FAILED to run command %s (%s)"
+                              % (command, ret))
             raise Exception(
                 "Unable to run \ncommand: %s\nret: %s"
                 % (command, ret))
@@ -96,6 +98,27 @@ class SSHClient(object):
         ftp.close()
 
 
+class LocalSSH(object):
+
+    def __init__(self, log):
+        self.log = log
+        self.log.info('Init local ssh client')
+
+    def ssh(self, cmd):
+        ret = 0
+        output = "%s failed!!!" % cmd
+        try:
+            output = subprocess.check_output((cmd), shell=True,
+                                             universal_newlines=True)
+        except subprocess.CalledProcessError:
+            ret = 1
+        return ret, output
+
+    def scp(self, src_file, dst_file):
+        return subprocess.check_output("cp %s %s" % (src_file, dst_file),
+                                       shell=True)
+
+
 def run_async(func):
     from threading import Thread
     from functools import wraps