Get path to CLI binary from env in load tests (#13600)

ydb-platform · Jan 20, 2025 · 2b12185 · 2b12185
1 parent fdde655
commit 2b12185
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 1 deletion.
diff --git a/ydb/tests/functional/tpc/large/ya.make b/ydb/tests/functional/tpc/large/ya.make
@@ -13,6 +13,7 @@ REQUIREMENTS(ram:16)
 
 ENV(YDB_ENABLE_COLUMN_TABLES="true")
 ENV(YDB_DRIVER_BINARY="ydb/apps/ydbd/ydbd")
+ENV(YDB_CLI_BINARY="ydb/apps/ydb/ydb")
 ENV(NO_KUBER_LOGS="yes")
 
 PEERDIR(

diff --git a/ydb/tests/functional/tpc/medium/ya.make b/ydb/tests/functional/tpc/medium/ya.make
@@ -12,6 +12,7 @@ REQUIREMENTS(ram:16)
 
 ENV(YDB_ENABLE_COLUMN_TABLES="true")
 ENV(YDB_DRIVER_BINARY="ydb/apps/ydbd/ydbd")
+ENV(YDB_CLI_BINARY="ydb/apps/ydb/ydb")
 ENV(NO_KUBER_LOGS="yes")
 
 PEERDIR(

diff --git a/ydb/tests/olap/lib/ydb_cli.py b/ydb/tests/olap/lib/ydb_cli.py
@@ -34,7 +34,7 @@ def get_cli_command() -> list[str]:
         if cli == 'git':
             return [yatest.common.work_path('ydb')] + args
         elif cli == 'main':
-            return [yatest.common.binary_path('ydb/apps/ydb/ydb')] + args
+            return [yatest.common.binary_path(os.getenv('YDB_CLI_BINARY'))] + args
         else:
             return [cli] + args
 

diff --git a/ydb/tests/olap/load/ya.make b/ydb/tests/olap/load/ya.make
@@ -4,6 +4,8 @@ PY3TEST()
 
     SIZE(LARGE)
 
+    ENV(YDB_CLI_BINARY="ydb/apps/ydb/ydb")
+
     TEST_SRCS (
         test_clickbench.py
         test_tpcds.py