Update node failure tests to use kill instead segv

alsugiliazova · alsugiliazova · commit f1827080b08c · 2026-03-10T14:22:11.000+01:00
diff --git a/swarms/tests/node_failure.py b/swarms/tests/node_failure.py
@@ -143,7 +143,7 @@ def check_restart_clickhouse_on_swarm_node(
     self,
     minio_root_user,
     minio_root_password,
-    row_count=100,
+    row_count=1000,
     batch_size=100,
     cluster_name="static_swarm_cluster",
     node=None,
@@ -170,14 +170,14 @@ def check_restart_clickhouse_on_swarm_node(
                 exitcode=32,
                 message="DB::Exception: Attempt to read after eof",
                 cluster_name=cluster_name,
-                delay_before_execution=1,
+                delay_before_execution=0,
             )
             Step(
                 "restart clickhouse on random swarm node",
                 test=actions.restart_clickhouse_on_random_swarm_node,
                 parallel=True,
                 executor=pool,
-            )(delay=0.0)
+            )(delay=30, signal="KILL", delay_before_execution=5)
             join()
 
 
@@ -457,8 +457,6 @@ def feature(self, minio_root_user, minio_root_password, node=None):
     Scenario(test=check_restart_clickhouse_on_swarm_node)(
         minio_root_user=minio_root_user,
         minio_root_password=minio_root_password,
-        row_count=row_count,
-        batch_size=batch_size,
     )
     Scenario(test=network_failure)(
         minio_root_user=minio_root_user,
diff --git a/swarms/tests/steps/swarm_node_actions.py b/swarms/tests/steps/swarm_node_actions.py
@@ -39,7 +39,9 @@ def restart_random_swarm_node(self, delay=None):
 
 
 @TestStep(Given)
-def restart_clickhouse_on_random_swarm_node(self, signal="SEGV", delay=None):
+def restart_clickhouse_on_random_swarm_node(
+    self, signal="SEGV", delay=None, delay_before_execution=0
+):
     """
     Send a kill signal to a random clickhouse swarm instance, wait, and restart.
     """
@@ -48,8 +50,12 @@ def restart_clickhouse_on_random_swarm_node(self, signal="SEGV", delay=None):
     if delay is None:
         delay = random.random() * 10 + 1
 
+    if delay_before_execution > 0:
+        with When(f"I wait {delay_before_execution}s"):
+            time.sleep(delay_before_execution)
+
     with interrupt_clickhouse(clickhouse_node, safe=False, signal=signal):
-        with When(f"I wait {delay:.2}s"):
+        with When(f"I wait {delay}s"):
             time.sleep(delay)
 
 
