Allow listing hypervisors in VMTP

[functest.git] / functest / opnfv_tests / openstack / vmtp / vmtp.py

diff --git a/functest/opnfv_tests/openstack/vmtp/vmtp.py b/functest/opnfv_tests/openstack/vmtp/vmtp.py
index 822c1b6..edff99c 100644 (file)
--- a/functest/opnfv_tests/openstack/vmtp/vmtp.py
+++ b/functest/opnfv_tests/openstack/vmtp/vmtp.py
@@ -33,6 +33,7 @@ from xtesting.core import testcase
 
 from functest.core import singlevm
 from functest.utils import env
+from functest.utils import functest_utils
 
 
 class Vmtp(singlevm.VmReady2):
@@ -148,8 +149,15 @@ class Vmtp(singlevm.VmReady2):
             del new_env['OS_TENANT_ID']
         except Exception:  # pylint: disable=broad-except
             pass
+        extra_args = ""
+        if env.get("VMTP_HYPERVISORS"):
+            hypervisors = functest_utils.convert_ini_to_list(
+                env.get("VMTP_HYPERVISORS"))
+            for hypervisor in hypervisors:
+                extra_args = "{} --hypervisor {} ".format(
+                    extra_args, hypervisor)
         cmd = ['vmtp', '-d', '--json', '{}/vmtp.json'.format(self.res_dir),
-               '-c', self.config]
+               '-c', self.config, extra_args]
         output = subprocess.check_output(
             cmd, stderr=subprocess.STDOUT, env=new_env).decode("utf-8")
         self.__logger.info("%s\n%s", " ".join(cmd), output)