Add pull to SSH and SDB in tct/run/common_test.py

Change-Id: I5f549d1fa9dff3bcc075066521a3cd38f2335a1f
Signed-off-by: Aleksander Mistewicz <a.mistewicz@samsung.com>

1 files changed, 13 insertions, 0 deletions

diff --git a/tct/run/common_test.py b/tct/run/common_test.py
index cb83afd..e8fa68f 100755
--- a/tct/run/common_test.py
+++ b/tct/run/common_test.py
@@ -18,6 +18,13 @@ class SSH(object):
         return subprocess.Popen(["sshpass", "-p", "tizen", "ssh", "-F", self.ssh_config, "root@" + self.ip] + cmd,
                 stdout=subprocess.PIPE)
 
+    def pull(self, src, dest):
+        """
+        src, dest -> returncode (int)
+        """
+        return subprocess.call(["sshpass", "-p", "tizen", "scp", "-F", self.ssh_config,
+            "root@" + self.ip + ":" + src, dest], stdout=subprocess.PIPE)
+
 class SDB(object):
     def __init__(self, serial):
         self.serial = serial
@@ -28,6 +35,12 @@ class SDB(object):
         """
         return subprocess.Popen(["sdb", "-s", self.serial, "shell"] + cmd, stdout=subprocess.PIPE)
 
+    def pull(self, src, dest):
+        """
+        src, dest -> returncode (int)
+        """
+        return subprocess.call(["sdb", "-s", self.serial, "pull", src, dest], stdout=subprocess.PIPE)
+
 class CommonTest(Test):
     default_params = {'timeout': 60.0}