[dts] [PATCH V1] framework: add set_env_variable when reconnet dut

[lihong <lihongx.ma@intel.com>]

When dts lost the session with DUT, it will try to reconnet to DUT,
but it will lose the env config, so add the env config when reconnet
to DUT.

Signed-off-by: lihong <lihongx.ma@intel.com>
---
 framework/project_dpdk.py | 11 +++++++----
 framework/test_case.py    |  1 +
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/framework/project_dpdk.py b/framework/project_dpdk.py
index 6035c68..f113f3e 100644
--- a/framework/project_dpdk.py
+++ b/framework/project_dpdk.py
@@ -64,10 +64,7 @@ class DPDKdut(Dut):
         self.set_toolchain(target)
 
         # set env variable
-        # These have to be setup all the time. Some tests need to compile
-        # example apps by themselves and will fail otherwise.
-        self.send_expect("export RTE_TARGET=" + target, "#")
-        self.send_expect("export RTE_SDK=`pwd`", "#")
+        self.set_env_variable()
 
         self.set_rxtx_mode()
 
@@ -88,6 +85,12 @@ class DPDKdut(Dut):
             self.bind_interfaces_linux(drivername)
         self.extra_nic_setup()
 
+    def set_env_variable(self):
+        # These have to be setup all the time. Some tests need to compile
+        # example apps by themselves and will fail otherwise.
+        self.send_expect("export RTE_TARGET=" + self.target, "#")
+        self.send_expect("export RTE_SDK=`pwd`", "#")
+
     def setup_modules(self, target, drivername, drivermode):
         """
         Install DPDK required kernel module on DUT.
diff --git a/framework/test_case.py b/framework/test_case.py
index 27d236b..c028492 100644
--- a/framework/test_case.py
+++ b/framework/test_case.py
@@ -135,6 +135,7 @@ class TestCase(object):
             crb.reconnect_session()
             if 'dut' in str(type(crb)):
                 crb.send_expect("cd %s" % crb.base_dir, "#")
+                crb.set_env_variable()
 
         try:
             result = crb.alt_session.check_available()
-- 
2.7.4