simple_queue workload in pytest (#12895)

Author: maximyurchuk

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
+        )

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()

ydb/tests/workloads/simple_queue/ya.make

@@ -11,3 +11,7 @@ PEERDIR(
 
 END()
 
+RECURSE_FOR_TESTS(
+    tests
+)
+