diff options
| author | Maxim Yurchuk <[email protected]> | 2025-01-15 10:05:49 +0000 |
|---|---|---|
| committer | GitHub <[email protected]> | 2025-01-15 13:05:49 +0300 |
| commit | 71c60ed7f13183b2d52808dd7aebcba767dc964a (patch) | |
| tree | 0d36aec74cbbb12754a60cd5d7ad6a882d2a34e8 | |
| parent | 834b05718637b9383153ac34b5279f8f220a5c0f (diff) | |
add column for simple_queue workload (#13263)
| -rw-r--r-- | ydb/tests/workloads/simple_queue/tests/test_workload.py | 33 |
1 files changed, 19 insertions, 14 deletions
diff --git a/ydb/tests/workloads/simple_queue/tests/test_workload.py b/ydb/tests/workloads/simple_queue/tests/test_workload.py index f83a9c462e6..38b7827b74b 100644 --- a/ydb/tests/workloads/simple_queue/tests/test_workload.py +++ b/ydb/tests/workloads/simple_queue/tests/test_workload.py @@ -9,23 +9,28 @@ from ydb.tests.library.common.types import Erasure 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) |