commit 6a0c4de8ecd2084c7ac5000c550b947f1de5855e
author Pier <pierluigi.ventre@cnit.it> Sun Mar 18 16:01:30 2018 -0700
committer Pier <pierluigi.ventre@cnit.it> Fri Mar 23 11:11:00 2018 -0700
tree 77e4e63b34861cd96a1bbc0e864a6bd7d3e6b263
parent f7c218c9ecb1e38034c3be840b05fb8a3e8c546e

[CORD-2862] TestON drivers for OFDPA and Host

Change-Id: I5a0005c128c091aadee567a144c0acef55235184

TestON/drivers/common/cli/networkdriver.py

diff --git a/TestON/drivers/common/cli/networkdriver.py b/TestON/drivers/common/cli/networkdriver.py
index d1a7faf..38d7531 100755
--- a/TestON/drivers/common/cli/networkdriver.py
+++ b/TestON/drivers/common/cli/networkdriver.py
@@ -122,9 +122,9 @@
         try:
             for key, value in main.componentDictionary.items():
                 if hasattr( main, key ):
-                    if value[ 'type' ] in [ 'MininetSwitchDriver' ]:
+                    if value[ 'type' ] in [ 'MininetSwitchDriver', 'OFDPASwitchDriver' ]:
                         self.switches[ key ] = getattr( main, key )
-                    elif value[ 'type' ] in [ 'MininetHostDriver' ]:
+                    elif value[ 'type' ] in [ 'MininetHostDriver', 'HostDriver' ]:
                         self.hosts[ key ] = getattr( main, key )
             return main.TRUE
         except Exception:
@@ -138,7 +138,7 @@
         hosts = {}
         try:
             for hostComponent in self.hosts.values():
-                #TODO: return more host data
+                # TODO: return more host data
                 hosts[ hostComponent.options[ 'shortName' ] ] = {}
         except Exception:
             main.log.error( self.name + ": host component not as expected" )