[dts] [PATCH] framework: support create addtional sessions on crb

Yong Liu yong.liu at intel.com
Tue Dec 1 09:51:44 CET 2015


Add session create/destroy functions in crb module.
Suite can create addtional session in dut by this function.
When session useless, suite should close the session definitely.
Below is sample code addtional session:

    session = self.dut.new_session()
    session.send_expect("ls -al", "# ")
    self.dut.close_session(session=session)

Signed-off-by: Marvin Liu <yong.liu at intel.com>

diff --git a/framework/crb.py b/framework/crb.py
index 3c158ec..68fe399 100644
--- a/framework/crb.py
+++ b/framework/crb.py
@@ -56,6 +56,7 @@ class Crb(object):
         self.skip_setup = False
         self.serializer = serializer
         self.ports_info = None
+        self.sessions = []
 
         self.logger = getLogger(name)
         self.session = SSHConnection(self.get_ip_address(), name,
@@ -81,6 +82,24 @@ class Crb(object):
 
         return self.session.send_expect(cmds, expected, timeout, verify)
 
+    def create_session(self, name=""):
+        """
+        Create new session for addtional useage. This session will not enable log.
+        """
+        session = SSHConnection(self.get_ip_address(), name,
+                                     self.get_password())
+        self.sessions.append(session)
+        return session.session
+
+    def destroy_session(self, session=None):
+        """
+        Destroy addtional session.
+        """
+        for save_session in self.sessions:
+            if save_session.session == session:
+                save_session.close()
+            self.sessions.remove(save_session)
+
     def send_command(self, cmds, timeout=TIMEOUT, alt_session=False):
         """
         Send commands to crb and return string before timeout.
diff --git a/framework/dut.py b/framework/dut.py
index c75f325..2cea7fb 100644
--- a/framework/dut.py
+++ b/framework/dut.py
@@ -41,6 +41,7 @@ from crb import Crb
 from net_device import NetDevice
 from virt_resource import VirtResource
 from utils import RED
+from uuid import uuid4
 
 
 class Dut(Crb):
@@ -85,6 +86,21 @@ class Dut(Crb):
             self.host_session.init_log(self.logger)
             self.host_init_flag = True
 
+    def new_session(self):
+        """
+        Create new session for dut instance. Session name will be unique.
+        """
+        session_name = str(uuid4())
+        session = self.create_session(name=session_name)
+        session.send_expect("cd %s" % self.base_dir, "# ")
+        return session
+
+    def close_session(self, session):
+        """
+        close new session in dut instance
+        """
+        self.destroy_session(session)
+
     def change_config_option(self, target, parameter, value):
         """
         This function change option in the config file
-- 
1.9.3



More information about the dts mailing list