[dts] [PATCH V1] framework/dts: close session if init pktgen failed

[lihong]

Signed-off-by: lihong <lihongx.ma at intel.com>
---
 framework/dts.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/framework/dts.py b/framework/dts.py
index f1e57bd..634a569 100644
--- a/framework/dts.py
+++ b/framework/dts.py
@@ -304,7 +304,14 @@ def dts_crbs_init(crbInsts, skip_setup, read_cache, project, base_dir, serialize
     tester.duts = duts
     show_speedup_options_messages(read_cache, skip_setup)
     tester.set_speedup_options(read_cache, skip_setup)
-    tester.init_ext_gen()
+    try:
+        tester.init_ext_gen()
+    except Exception as e:
+        log_handler.error(str(e))
+        tester.close()
+        for dutobj in duts:
+            dutobj.close()
+        raise e
 
     nic = settings.load_global_setting(settings.HOST_NIC_SETTING)
     for dutobj in duts:
-- 
2.7.4