teuthology/misc: Add timeout parameter to stop_daemons_of_type for be…

…tter flexibility Updated stop_daemons_of_type to accept a timeout parameter, allowing dynamic control over the timeout value passed to the stop function of each daemon. Signed-off-by: Nitzan Mordechai <[email protected]>
ceph · Dec 10, 2024 · cbef9c6 · cbef9c6
1 parent e4eeebd
commit cbef9c6
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/teuthology/misc.py b/teuthology/misc.py
@@ -1148,15 +1148,17 @@ def ssh_keyscan_wait(hostname):
             log.info("try ssh_keyscan again for " + str(hostname))
         return success
 
-def stop_daemons_of_type(ctx, type_, cluster='ceph'):
+def stop_daemons_of_type(ctx, type_, cluster='ceph', timeout=300):
     """
     :param type_: type of daemons to be stopped.
+    :param cluster: Cluster name, default is 'ceph'.
+    :param timeout: Timeout in seconds for stopping each daemon.
     """
     log.info('Shutting down %s daemons...' % type_)
     exc = None
     for daemon in ctx.daemons.iter_daemons_of_role(type_, cluster):
         try:
-            daemon.stop()
+            daemon.stop(timeout)
         except (CommandFailedError,
                 CommandCrashedError,
                 ConnectionLostError) as e: