From 5dc618574b36235e5f5c9ad8630ad14cd4562ef2 Mon Sep 17 00:00:00 2001 From: Drew Hoskins Date: Thu, 27 Jun 2024 10:00:33 -0700 Subject: [PATCH] Fix flaky test --- .../updates_and_signals/safe_message_handlers/workflow_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/updates_and_signals/safe_message_handlers/workflow_test.py b/tests/updates_and_signals/safe_message_handlers/workflow_test.py index adcc7242..3c9ccf4f 100644 --- a/tests/updates_and_signals/safe_message_handlers/workflow_test.py +++ b/tests/updates_and_signals/safe_message_handlers/workflow_test.py @@ -34,7 +34,7 @@ async def test_safe_message_handlers(client: Client): ) await do_cluster_lifecycle(cluster_manager_handle, delay_seconds=1) result = await cluster_manager_handle.result() - assert result.max_assigned_nodes == 12 + assert result.max_assigned_nodes + result.num_bad_nodes == 12 assert result.num_currently_assigned_nodes == 0