feat: add statistics workload (#24798)

Added statistics workload to nemesis CI

Refs: #24043

Author: PavelEfarinov

ydb/tests/olap/load/lib/workload_statistics.py

@@ -0,0 +1,47 @@
+import time
+import pytest
+from .workload_executor import WorkloadTestBase
+from ydb.tests.olap.lib.ydb_cluster import YdbCluster
+from ydb.tests.olap.lib.utils import get_external_param
+
+import logging
+LOGGER = logging.getLogger(__name__)
+
+
+class WorkloadStatisticsBase(WorkloadTestBase):
+    workload_binary_name = 'statistics_workload'
+    workload_env_var = 'STATISTICS_WORKLOAD_BINARY'
+
+    @pytest.mark.parametrize(
+        'nemesis_enabled', [True, False],
+        ids=['nemesis_true', 'nemesis_false']
+    )
+    def test_workload_statistics(self, nemesis_enabled: bool):
+        command_args_template = (
+            "--host {node_host} "
+            "--port 2135 "
+            f"--database /{YdbCluster.ydb_database} "
+            "--prefix statistics_workload_{node_host}_iter_{iteration_num}_{uuid} "
+        )
+
+        additional_stats = {
+            "workload_type": "statistics",
+            "statistics_template": "workload_statistics_{node_host}_iter_{iteration_num}_{uuid}",
+            "nemesis": nemesis_enabled,
+            "test_timestamp": int(time.time()),
+        }
+
+        self.execute_workload_test(
+            workload_name=f"StatisticsWorkload_nemesis_{nemesis_enabled}",
+            command_args=command_args_template,
+            duration_value=self.timeout,
+            additional_stats=additional_stats,
+            use_chunks=True,
+            duration_param="--duration",
+            nemesis=nemesis_enabled,
+            nodes_percentage=100
+        )
+
+
+class TestWorkloadStatistics(WorkloadStatisticsBase):
+    timeout = int(get_external_param('workload_duration', 120))

ydb/tests/olap/load/lib/ya.make

@@ -24,6 +24,7 @@ PY3_LIBRARY()
         workload_reconfig_state_storage.py
         workload_show_create.py
         workload_cdc.py
+        workload_statistics.py
         upload.py
 
     )

ydb/tests/olap/load/test_workload_statistics.py

@@ -0,0 +1 @@
+from ydb.tests.olap.load.lib.workload_statistics import * # noqa

ydb/tests/olap/load/ya.make

@@ -20,6 +20,7 @@ PY3TEST()
     ENV(RECONFIG_STATE_STORAGE_WORKLOAD_BINARY="ydb/tests/stress/reconfig_state_storage_workload/reconfig_state_storage_workload")
     ENV(SHOW_CREATE_WORKLOAD_BINARY="ydb/tests/stress/show_create/view/show_create_view")
     ENV(CDC_WORKLOAD_BINARY="ydb/tests/stress/cdc/cdc")
+    ENV(STATISTICS_WORKLOAD_BINARY="ydb/tests/stress/statistics_workload/statistics_workload")
     ENV(NEMESIS_BINARY="ydb/tests/tools/nemesis/driver/nemesis")
 
     TEST_SRCS (
@@ -45,6 +46,7 @@ PY3TEST()
         test_workload_reconfig_state_storage.py
         test_workload_show_create.py
         test_workload_cdc.py
+        test_workload_statistics.py
     )
 
     PEERDIR (
@@ -69,6 +71,7 @@ PY3TEST()
             ydb/tests/stress/reconfig_state_storage_workload
             ydb/tests/stress/show_create/view
             ydb/tests/stress/cdc
+            ydb/tests/stress/statistics_workload
             ydb/tests/tools/nemesis/driver
         )
     ENDIF()