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

[Tu, Lijuan]

Applied, thanks

> -----Original Message-----
> From: dts [mailto:dts-bounces at dpdk.org] On Behalf Of lihong
> Sent: Thursday, May 30, 2019 8:01 AM
> To: dts at dpdk.org
> Cc: Chen, Zhaoyan <zhaoyan.chen at intel.com>; Ma, LihongX
> <lihongx.ma at intel.com>
> Subject: [dts] [PATCH V1] framework: add set_env_variable when reconnet
> dut
> 
> 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 at 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