From 686cf76645d4e38d581852a46f21bc8308734efb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D0=9E=D0=BB=D0=B5=D0=B3?=
 <150132506+iddqdex@users.noreply.github.com>
Date: Fri, 31 Jan 2025 00:47:14 +0300
Subject: [PATCH] Make ydb/tests/stress/log/tests/test_workload parrallel
 (#14045)

---
 ydb/tests/stress/log/tests/test_workload.py | 39 ++++++++++++---------
 1 file changed, 23 insertions(+), 16 deletions(-)

diff --git a/ydb/tests/stress/log/tests/test_workload.py b/ydb/tests/stress/log/tests/test_workload.py
index 46a62327d644..41af853e2929 100644
--- a/ydb/tests/stress/log/tests/test_workload.py
+++ b/ydb/tests/stress/log/tests/test_workload.py
@@ -42,28 +42,35 @@ def teardown_class(cls):
 
     @pytest.mark.parametrize('store_type', ['row', 'column'])
     def test(self, store_type):
-        commands = [
-            # init
+        upload_commands = [
+            # bulk upsert workload
+            self.get_command_prefix(subcmds=['run', 'bulk_upsert'], path=store_type) + self.get_insert_command_params() + ['--seconds', '10', '--threads', '10'],
+
+            # upsert workload
+            self.get_command_prefix(subcmds=['run', 'upsert'], path=store_type) + self.get_insert_command_params() + ['--seconds', '10', '--threads', '10'],
+
+            # insert workload
+            self.get_command_prefix(subcmds=['run', 'insert'], path=store_type) + self.get_insert_command_params() + ['--seconds', '10', '--threads', '10'],
+        ]
+
+        # init
+        yatest.common.execute(
             self.get_command_prefix(subcmds=['init'], path=store_type) + self.get_insert_command_params() + [
                 '--store', store_type,
                 '--min-partitions', '100',
                 '--partition-size', '10',
                 '--auto-partition', '0',
             ],
+        )
 
-            # bulk upsert workload
-            self.get_command_prefix(subcmds=['run', 'bulk_upsert'], path=store_type) + self.get_insert_command_params(),
-
-            # upsert workload
-            self.get_command_prefix(subcmds=['run', 'upsert'], path=store_type) + self.get_insert_command_params(),
-
-            # insert workload
-            self.get_command_prefix(subcmds=['run', 'insert'], path=store_type) + self.get_insert_command_params(),
-
-            # select workload
+        select = yatest.common.execute(
             self.get_command_prefix(subcmds=['run', 'select'], path=store_type) + [
-                '--client-timeout', '10000'
-            ]
-        ]
-        for command in commands:
+                '--client-timeout', '10000',
+                '--threads', '10',
+                '--seconds', str(10 * len(upload_commands)),
+            ], wait=False)
+
+        for command in upload_commands:
             yatest.common.execute(command, wait=True)
+
+        select.wait()