commit b0c977c22028d68c76346275c11806fc05212d23
author Yulia Portnova <yportnova@mirantis.com> Fri Dec 11 19:23:28 2015 +0200
committer Yulia Portnova <yportnova@mirantis.com> Fri Dec 11 19:23:28 2015 +0200
tree 3a85c9dc1c2e7c5de77b85fe432f62838f08b014
parent 5ea9df0a51c30657b56406e531f0973029873192

fixes for two scripts; omgbenchmark sample

wally/run_test.py

diff --git a/wally/run_test.py b/wally/run_test.py
index 2cd397e..af317b3 100755
--- a/wally/run_test.py
+++ b/wally/run_test.py
@@ -26,12 +26,14 @@
 from wally.suits.itest import TestConfig
 from wally.suits.io.fio import IOPerfTest
 from wally.suits.postgres import PgBenchTest
+from wally.suits.omgbench import OmgTest
 
 
 TOOL_TYPE_MAPPER = {
     "io": IOPerfTest,
     "pgbench": PgBenchTest,
     "mysql": MysqlTest,
+    "omg": OmgTest,
 }
 
 
@@ -540,6 +542,8 @@
                 rep = "\n\n".join(rep_lst)
             elif tp in ['mysql', 'pgbench'] and data is not None:
                 rep = MysqlTest.format_for_console(data)
+            elif tp == 'omg':
+                rep = OmgTest.format_for_console(data)
             else:
                 logger.warning("Can't generate text report for " + tp)
                 continue