From a3530e09caad36ae3bbd717684b2c6ff8776dc55 Mon Sep 17 00:00:00 2001 From: "m.malygina" Date: Wed, 30 Aug 2023 10:27:10 +0300 Subject: [PATCH] support stop service by type Signed-off-by: m.malygina --- .../storage/controllers/cluster_state_controller.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/src/frostfs_testlib/storage/controllers/cluster_state_controller.py b/src/frostfs_testlib/storage/controllers/cluster_state_controller.py index 2d439d9..0c22d7c 100644 --- a/src/frostfs_testlib/storage/controllers/cluster_state_controller.py +++ b/src/frostfs_testlib/storage/controllers/cluster_state_controller.py @@ -17,6 +17,7 @@ from frostfs_testlib.utils.failover_utils import ( wait_for_host_online, wait_for_node_online, ) +from frostfs_testlib.storage.dataclasses.node_base import NodeBase reporter = get_reporter() @@ -170,6 +171,18 @@ class ClusterStateController: parallel(self.start_s3_gate, copy.copy(self.stopped_s3_gates)) self.stopped_s3_gates = [] + @run_optionally(optionals.OPTIONAL_FAILOVER_ENABLED) + @reporter.step_deco("Stop {service_type} services on {cluster_node} node") + def stop_services_of_type(self, cluster_node: ClusterNode, service_type: NodeBase): + service = cluster_node.service(service_type) + service.stop_service() + + @run_optionally(optionals.OPTIONAL_FAILOVER_ENABLED) + @reporter.step_deco("Start {service_type} services on {cluster_node} node") + def start_services_of_type(self, cluster_node: ClusterNode, service_type: NodeBase): + service = cluster_node.service(service_type) + service.start_service() + @run_optionally(optionals.OPTIONAL_FAILOVER_ENABLED) @reporter.step_deco("Suspend {process_name} service in {node}") def suspend_service(self, process_name: str, node: ClusterNode): -- 2.45.2