New methods with nodes

Signed-off-by: Dmitriy Zayakin <d.zayakin@yadro.com>

src/frostfs_testlib/storage/cluster.py

@@ -2,9 +2,11 @@ import random
 import re
 
 import yaml
+from yarl import URL
 
 from frostfs_testlib.hosting import Host, Hosting
 from frostfs_testlib.hosting.config import ServiceConfig
+from frostfs_testlib.reporter import get_reporter
 from frostfs_testlib.storage import get_service_registry
 from frostfs_testlib.storage.constants import ConfigAttributes
 from frostfs_testlib.storage.dataclasses.frostfs_services import (
@@ -17,6 +19,8 @@ from frostfs_testlib.storage.dataclasses.frostfs_services import (
 from frostfs_testlib.storage.dataclasses.node_base import NodeBase, ServiceClass
 from frostfs_testlib.storage.service_registry import ServiceRegistry
 
+reporter = get_reporter()
+
 
 class ClusterNode:
     """
@@ -250,3 +254,10 @@ class Cluster:
     def get_morph_endpoints(self) -> list[str]:
         nodes: list[MorphChain] = self.services(MorphChain)
         return [node.get_endpoint() for node in nodes]
+
+    def get_nodes_by_ip(self, ips: list[str]) -> list[ClusterNode]:
+        cluster_nodes = [
+            node for node in self.cluster_nodes if URL(node.morph_chain.get_endpoint()).host in ips
+        ]
+        with reporter.step(f"Return cluster nodes - {cluster_nodes}"):
+            return cluster_nodes

src/frostfs_testlib/storage/controllers/cluster_state_controller.py

@@ -25,6 +25,7 @@ class ClusterStateController:
         self.stopped_storage_nodes: list[ClusterNode] = []
         self.cluster = cluster
         self.shell = shell
+        self.suspended_services: dict[str, list[ClusterNode]] = {}
 
     @run_optionally(optionals.OPTIONAL_FAILOVER_ENABLED)
     @reporter.step_deco("Stop host of node {node}")
@@ -130,6 +131,31 @@ class ClusterStateController:
         wait_all_storage_nodes_returned(self.shell, self.cluster)
         self.stopped_storage_nodes = []
 
+    @run_optionally(optionals.OPTIONAL_FAILOVER_ENABLED)
+    @reporter.step_deco("Suspend {process_name} service in {node}")
+    def suspend_service(self, process_name: str, node: ClusterNode):
+        node.host.wait_success_suspend_process(process_name)
+        if self.suspended_services.get(process_name):
+            self.suspended_services[process_name].append(node)
+        else:
+            self.suspended_services[process_name] = [node]
+
+    @run_optionally(optionals.OPTIONAL_FAILOVER_ENABLED)
+    @reporter.step_deco("Resume {process_name} service in {node}")
+    def resume_service(self, process_name: str, node: ClusterNode):
+        node.host.wait_success_resume_process(process_name)
+        if self.suspended_services.get(process_name):
+            self.suspended_services[process_name].append(node)
+        else:
+            self.suspended_services[process_name] = [node]
+
+    @run_optionally(optionals.OPTIONAL_FAILOVER_ENABLED)
+    @reporter.step_deco("Start suspend processes services")
+    def resume_suspended_services(self):
+        for process_name, list_nodes in self.suspended_services.items():
+            [node.host.wait_success_resume_process(process_name) for node in list_nodes]
+        self.suspended_services = {}
+
     @run_optionally(optionals.OPTIONAL_FAILOVER_ENABLED)
     @reporter.step_deco("Hard reboot host {node} via magic SysRq option")
     def panic_reboot_host(self, node: ClusterNode, wait_for_return: bool = True):

src/frostfs_testlib/storage/dataclasses/node_base.py

@@ -19,6 +19,7 @@ class NodeBase(ABC):
     id: str
     name: str
     host: Host
+    _process_name: str
 
     def __init__(self, id, name, host) -> None:
         self.id = id
@@ -48,6 +49,9 @@ class NodeBase(ABC):
     def get_service_systemctl_name(self) -> str:
         return self._get_attribute(ConfigAttributes.SERVICE_NAME)
 
+    def get_process_name(self) -> str:
+        return self._process_name
+
     def start_service(self):
         self.host.start_service(self.name)