From b6f623b9314de599fa9f070cb7f9b3dfa5cf087b Mon Sep 17 00:00:00 2001
From: Albert Zaharovits <albert.zaharovits@elastic.co>
Date: Fri, 7 Jun 2024 14:20:46 +0300
Subject: [PATCH] assertBusy for verify

---
 .../transport/RemoteClusterAwareClientTests.java          | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/server/src/test/java/org/elasticsearch/transport/RemoteClusterAwareClientTests.java b/server/src/test/java/org/elasticsearch/transport/RemoteClusterAwareClientTests.java
index 7f355f7cb90dc..059a154e6b7de 100644
--- 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()
+                    )
+                );
             }
         }
     }