check nodes after test and restart they before one (#16118)

Author: iddqdex

ydb/tests/functional/tpc/large/ya.make

@@ -15,6 +15,7 @@ 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")
+ENV(WAIT_CLUSTER_ALIVE_TIMEOUT="60")
 
 PEERDIR(
     ydb/tests/functional/tpc/lib

ydb/tests/functional/tpc/lib/conftest.py

@@ -40,6 +40,21 @@ def setup_cluster(cls, table_service_config: dict = {}, memory_controller_config
         cls.cluster.register_and_start_slots(db, count=YdbCluster.get_dyn_nodes_count())
         cls.cluster.wait_tenant_up(db)
 
+    def setup_method(self, method):
+        for node in self.cluster.nodes.values():
+            node.start()
+        for slot in self.cluster.slots.values():
+            slot.start()
+        self.cluster.wait_tenant_up(f'/{YdbCluster.ydb_database}')
+
+    def teardown_method(self, method):
+        for node in self.cluster.nodes.values():
+            if not node.is_alive():
+                node.stop()
+        for slot in self.cluster.slots.values():
+            if not slot.is_alive():
+                slot.stop()
+
     @classmethod
     def run_cli(cls, argv: list[str]) -> yatest.common.process._Execution:
         return yatest.common.execute(YdbCliHelper.get_cli_command() + argv)

ydb/tests/functional/tpc/medium/ya.make

@@ -14,6 +14,7 @@ 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")
+ENV(WAIT_CLUSTER_ALIVE_TIMEOUT="60")
 
 PEERDIR(
     ydb/tests/functional/tpc/lib

ydb/tests/olap/lib/ydb_cli.py

@@ -278,7 +278,7 @@ def _exec_cli(self) -> None:
 
         def process(self) -> YdbCliHelper.WorkloadRunResult:
             try:
-                wait_error = YdbCluster.wait_ydb_alive(20 * 60, self.db_path)
+                wait_error = YdbCluster.wait_ydb_alive(int(os.getenv('WAIT_CLUSTER_ALIVE_TIMEOUT', 20 * 60)), self.db_path)
                 if wait_error is not None:
                     self.result.error_message = wait_error
                 else:

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

@@ -249,7 +249,7 @@ def _attach_plans(plan: YdbCliHelper.QueryPlan, name: str) -> None:
     @classmethod
     def setup_class(cls) -> None:
         start_time = time()
-        error = YdbCluster.wait_ydb_alive(20 * 60)
+        error = YdbCluster.wait_ydb_alive(int(os.getenv('WAIT_CLUSTER_ALIVE_TIMEOUT', 20 * 60)))
         tb = None
         if not error and hasattr(cls, 'do_setup_class'):
             try: