[dts] [PATCH]framework: Delete NetDevice object after virtual machine shutdown

[Lijuan Tu]

Signed-off-by: Lijuan Tu <lijuanx.a.tu at intel.com>
---
 framework/virt_dut.py | 16 ++++++++++++++++
 nics/net_device.py    | 15 +++++++++++++++
 2 files changed, 31 insertions(+)

diff --git a/framework/virt_dut.py b/framework/virt_dut.py
index 9bf75dc..019a9f2 100644
--- a/framework/virt_dut.py
+++ b/framework/virt_dut.py
@@ -39,6 +39,7 @@ from settings import NICS, LOG_NAME_SEP, get_netdev
 from project_dpdk import DPDKdut
 from dut import Dut
 from net_device import GetNicObj
+from net_device import RemoveNicObj
 
 
 class VirtDut(DPDKdut):
@@ -75,10 +76,25 @@ class VirtDut(DPDKdut):
         self.logger.config_suite(self.host_dut.test_classname, 'virtdut')
 
     def close(self):
+        # Close ssh session
         if self.session:
             self.session.close()
+            self.session = None
         if self.alt_session:
             self.alt_session.close()
+            self.alt_session = None
+
+        # Remove Network object
+        vf_ports = self.get_vf_net_divce()
+        RemoveNicObj(self, vf_ports)
+
+    def get_vf_net_divce(self):
+        vf_port = []
+        for port_info in self.host_dut.ports_info:
+            if 'vfs_port' in port_info.keys():
+                for port in port_info['vfs_port']:
+                    vf_port.append(port)
+        return vf_port
 
     def set_nic_type(self, nic_type):
         """
diff --git a/nics/net_device.py b/nics/net_device.py
index 73750f5..8c967ae 100644
--- a/nics/net_device.py
+++ b/nics/net_device.py
@@ -830,6 +830,14 @@ def get_from_list(host, bus_id, devfun_id):
                 return nic['port']
     return None
 
+def remove_from_list(host, obj):
+    """
+    Remove network device object from global structure
+    Parameter will by host ip, netdevice object
+    """
+    for nic in NICS_LIST:
+        if host == nic['host']:
+            NICS_LIST.remove(nic)
 
 def GetNicObj(crb, bus_id, devfun_id):
     """
@@ -862,3 +870,10 @@ def GetNicObj(crb, bus_id, devfun_id):
 
     add_to_list(crb.crb['My IP'], obj)
     return obj
+
+def RemoveNicObj(crb, objs):
+    """
+    Remove network device object.
+    """
+    for obj in objs:
+        remove_from_list(crb.crb['My IP'], obj)
-- 
2.5.0