Forwarding flavor parameters in stress test

[bottlenecks.git] / utils / infra_setup / runner / yardstick.py

diff --git a/utils/infra_setup/runner/yardstick.py b/utils/infra_setup/runner/yardstick.py
index 616bcc5..6071585 100644 (file)
--- a/utils/infra_setup/runner/yardstick.py
+++ b/utils/infra_setup/runner/yardstick.py
@@ -45,7 +45,14 @@ def yardstick_command_parser(debug, cidr, outfile, parameter):
     cmd += " --output-file " + outfile
     image_name = config.bottlenecks_config["yardstick_image_name"]
     parameter["image_name"] = image_name
-    print parameter
+    DEPLOY_SCENARIO = os.getenv("DEPLOY_SCENARIO")
+    RAM_NUM = os.getenv("RAM_NUM")
+    if DEPLOY_SCENARIO:
+        if "ovs" in DEPLOY_SCENARIO:
+            parameter["dpdk_enabled"] = True
+            if RAM_NUM:
+                parameter["ram_num"] = RAM_NUM
+    LOG.info(parameter)
     if parameter is not None:
         cmd += " --task-args " + '"' + str(parameter) + '"'
     return cmd