Test: Additional changes to support cluster based test scenarios.
Change-Id: I8464d2c775809a723ef46c0f5d3625bbbe5d0d48
diff --git a/src/test/cli/onosclidriver.py b/src/test/cli/onosclidriver.py
index cc3d8c0..d9c6011 100644
--- a/src/test/cli/onosclidriver.py
+++ b/src/test/cli/onosclidriver.py
@@ -48,8 +48,11 @@
self.name = None
self.home = None
self.handle = None
- self.controller = os.getenv('ONOS_CONTROLLER_IP') or 'localhost'
- self.controller = self.controller.split(',')[0]
+ if controller is not None:
+ self.controller = controller
+ else:
+ self.controller = os.getenv('ONOS_CONTROLLER_IP') or 'localhost'
+ self.controller = self.controller.split(',')[0]
super( CLI, self ).__init__()
if connect == True:
self.connect_cli()