assertBusy for verify

albertzaharovits · Jun 7, 2024 · b6f623b · b6f623b
1 parent 4f78f71
commit b6f623b
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/server/src/test/java/org/elasticsearch/transport/RemoteClusterAwareClientTests.java b/server/src/test/java/org/elasticsearch/transport/RemoteClusterAwareClientTests.java
@@ -150,7 +150,13 @@ public void testRemoteTaskCancellationOnFailedResponse() throws Exception {
                 // assert remote task is cancelled
                 assertBusy(cancelChildSent::get);
                 assertBusy(cancelChildReceived::get);
-                verify(remoteTransport.getTaskManager()).cancelChildLocal(eq(parentTaskId), eq(searchShardsRequestId.get()), anyString());
+                assertBusy(
+                    () -> verify(remoteTransport.getTaskManager()).cancelChildLocal(
+                        eq(parentTaskId),
+                        eq(searchShardsRequestId.get()),
+                        anyString()
+                    )
+                );
             }
         }
     }