add column for simple_queue workload (ydb-platform#13263)

uzhastik · Jan 15, 2025 · 71c60ed · 71c60ed
1 parent 834b057
commit 71c60ed
Showing 1 changed file with 19 additions and 14 deletions.
diff --git a/ydb/tests/workloads/simple_queue/tests/test_workload.py b/ydb/tests/workloads/simple_queue/tests/test_workload.py
@@ -9,23 +9,28 @@
 class TestYdbWorkload(object):
     @classmethod
     def setup_class(cls):
-        cls.cluster = KiKiMR(KikimrConfigGenerator(erasure=Erasure.MIRROR_3_DC))
+        config_generator = KikimrConfigGenerator(erasure=Erasure.MIRROR_3_DC)
+        config_generator.yaml_config["table_service_config"]["allow_olap_data_query"] = True
+        cls.cluster = KiKiMR(config_generator)
         cls.cluster.start()
+        workload_path = yatest.common.build_path("ydb/tests/workloads/simple_queue/simple_queue")
+        cls.workload_command_prefix = [
+            workload_path,
+            "--endpoint", "grpc://localhost:%d" % cls.cluster.nodes[1].grpc_port,
+            "--database=/Root",
+            "--duration", "60",
+        ]
 
     @classmethod
     def teardown_class(cls):
         cls.cluster.stop()
 
-    def test(self):
-        workload_path = yatest.common.build_path("ydb/tests/workloads/simple_queue/simple_queue")
-        store = "row"  # or "column"
-        yatest.common.execute(
-            [
-                workload_path,
-                "--endpoint", "grpc://localhost:%d" % self.cluster.nodes[1].grpc_port,
-                "--database=/Root",
-                "--duration", "60",
-                "--mode", store,
-            ],
-            wait=True
-        )
+    def test_row(self):
+        command = self.workload_command_prefix
+        command.extend(["--mode", "row"])
+        yatest.common.execute(command, wait=True)
+
+    def test_column(self):
+        command = self.workload_command_prefix
+        command.extend(["--mode", "column"])
+        yatest.common.execute(command, wait=True)