chnage nodes count in sql tests (#13598)

ydb-platform · Jan 20, 2025 · 294b46f · 294b46f
1 parent decee74
commit 294b46f
Showing 1 changed file with 15 additions and 3 deletions.
diff --git a/ydb/tests/sql/lib/test_base.py b/ydb/tests/sql/lib/test_base.py
@@ -27,7 +27,7 @@ def setup_class(cls):
 
         cls.ydb_cli_path = yatest.common.build_path("ydb/apps/ydb/ydb")
 
-        cls.database = "/Root"
+        cls.database = "/Root/TestDb"
         cls.cluster = KiKiMR(KikimrConfigGenerator(erasure=cls.get_cluster_configuration(),
                                                    extra_feature_flags=["enable_resource_pools",
                                                                         "enable_external_data_sources",
@@ -42,6 +42,14 @@ def setup_class(cls):
                                                    additional_log_configs={
                                                        'TX_TIERING': LogLevels.DEBUG}))
         cls.cluster.start()
+        cls.cluster.create_database(
+            cls.database,
+            storage_pool_units_count={
+                'hdd': len(cls.cluster.nodes)
+            }
+        )
+        cls.cluster.register_and_start_slots(cls.database, count=cls.get_dynnodes_count())
+        cls.cluster.wait_tenant_up(cls.database)
         cls.driver = ydb.Driver(
             ydb.DriverConfig(
                 database=cls.get_database(),
@@ -59,6 +67,10 @@ def create_connection(self, user=None, password=None) -> Query:
     def get_cluster_configuration(self):
         return Erasure.NONE
 
+    @classmethod
+    def get_dynnodes_count(self):
+        return 1
+
     @classmethod
     def get_database(self):
         return self.database
@@ -85,8 +97,8 @@ def setup_method(self):
 class TpchTestBaseH1(TestBase):
 
     @classmethod
-    def get_cluster_configuration(self):
-        return Erasure.MIRROR_3_DC
+    def get_dynnodes_count(self):
+        return 4
 
     @classmethod
     def run_cli(cls, argv: list[str]) -> yatest.common.process._Execution: