diff options
| author | Maxim Yurchuk <maxim-yurchuk@ydb.tech> | 2024-12-25 13:04:52 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-12-25 16:04:52 +0300 |
| commit | b46a3cc13f7b807cc7c8b8e43ae50b0617d8b2fe (patch) | |
| tree | 48bed425dfb6476e62daf68baef7c81a21c9b488 | |
| parent | 57f6f6331a567e252cb00980c880e1e32ac25af1 (diff) | |
| download | ydb-b46a3cc13f7b807cc7c8b8e43ae50b0617d8b2fe.tar.gz | |
simple_queue workload in pytest (#12895)
| -rw-r--r-- | ydb/tests/workloads/simple_queue/tests/test_workload.py | 31 | ||||
| -rw-r--r-- | ydb/tests/workloads/simple_queue/tests/ya.make | 25 | ||||
| -rw-r--r-- | ydb/tests/workloads/simple_queue/ya.make | 4 |
3 files changed, 60 insertions, 0 deletions
diff --git a/ydb/tests/workloads/simple_queue/tests/test_workload.py b/ydb/tests/workloads/simple_queue/tests/test_workload.py new file mode 100644 index 00000000000..f83a9c462e6 --- /dev/null +++ b/ydb/tests/workloads/simple_queue/tests/test_workload.py @@ -0,0 +1,31 @@ +# -*- coding: utf-8 -*- +import yatest + +from ydb.tests.library.harness.kikimr_runner import KiKiMR +from ydb.tests.library.harness.kikimr_config import KikimrConfigGenerator +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)) + cls.cluster.start() + + @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 + ) diff --git a/ydb/tests/workloads/simple_queue/tests/ya.make b/ydb/tests/workloads/simple_queue/tests/ya.make new file mode 100644 index 00000000000..dd5bff65633 --- /dev/null +++ b/ydb/tests/workloads/simple_queue/tests/ya.make @@ -0,0 +1,25 @@ +PY3TEST() +ENV(YDB_DRIVER_BINARY="ydb/apps/ydbd/ydbd") + +TEST_SRCS( + test_workload.py +) + +IF (SANITIZER_TYPE) + REQUIREMENTS(ram:32) +ENDIF() + +SIZE(MEDIUM) + +DEPENDS( + ydb/apps/ydbd + ydb/apps/ydb + ydb/tests/workloads/simple_queue +) + +PEERDIR( + ydb/tests/library +) + + +END() diff --git a/ydb/tests/workloads/simple_queue/ya.make b/ydb/tests/workloads/simple_queue/ya.make index c90c63ced4a..cd4c38ee24a 100644 --- a/ydb/tests/workloads/simple_queue/ya.make +++ b/ydb/tests/workloads/simple_queue/ya.make @@ -11,3 +11,7 @@ PEERDIR( END() +RECURSE_FOR_TESTS( + tests +) + |