From 7ae21dedd339888357180cc4648e8157c7d055d8 Mon Sep 17 00:00:00 2001
From: Oleg Geller <iddqd@ydb.tech>
Date: Fri, 17 Jan 2025 10:39:58 +0300
Subject: [PATCH] Add test for clickbench

---
 .../functional/tpc/medium/test_clickbench.py     | 16 ++++++++++++++++
 ydb/tests/functional/tpc/medium/ya.make          |  5 +++++
 2 files changed, 21 insertions(+)
 create mode 100644 ydb/tests/functional/tpc/medium/test_clickbench.py

diff --git a/ydb/tests/functional/tpc/medium/test_clickbench.py b/ydb/tests/functional/tpc/medium/test_clickbench.py
new file mode 100644
index 000000000000..ef40bb82b07e
--- /dev/null
+++ b/ydb/tests/functional/tpc/medium/test_clickbench.py
@@ -0,0 +1,16 @@
+import os
+import ydb.tests.olap.load.lib.clickbench as clickbench
+import yatest.common
+from ydb.tests.functional.tpc.lib.conftest import FunctionalTestBase
+
+
+class TestClickbench(clickbench.TestClickbench, FunctionalTestBase):
+    iterations: int = 1
+
+    @classmethod
+    def setup_class(cls) -> None:
+        cls.setup_cluster()
+        cls.run_cli(['workload', 'clickbench', '-p', 'olap_yatests/clickbench/hits', 'init', '--store=column'])
+        cls.run_cli(['workload', 'clickbench', '-p', 'olap_yatests/clickbench/hits', 'import', 'files', '--input', yatest.common.source_path("ydb/tests/functional/clickbench/data/hits.csv")])
+        os.environ['NO_VERIFY_DATA_CLICKBENCH'] = '1'
+        clickbench.TestClickbench.setup_class()
diff --git a/ydb/tests/functional/tpc/medium/ya.make b/ydb/tests/functional/tpc/medium/ya.make
index 92000f8718ab..59631d5cc03a 100644
--- a/ydb/tests/functional/tpc/medium/ya.make
+++ b/ydb/tests/functional/tpc/medium/ya.make
@@ -4,6 +4,7 @@ PY3TEST()
 ENV(YDB_HARD_MEMORY_LIMIT_BYTES="107374182400")
 
 TEST_SRCS(
+    test_clickbench.py
     test_tpch.py
 )
 
@@ -24,6 +25,10 @@ DEPENDS(
     ydb/apps/ydbd
 )
 
+DATA(
+    arcadia/ydb/tests/functional/clickbench/data/hits.csv
+)
+
 FORK_TEST_FILES()
 
 END()