Fix flaky TransportMultiSearchActionTests.testCancellation

Signed-off-by: kkewwei <kewei.11@bytedance.com>
Signed-off-by: kkewwei <kkewwei@163.com>

Author: kkewwei

server/src/test/java/org/opensearch/action/search/TransportMultiSearchActionTests.java

@@ -293,7 +293,7 @@ public void testDefaultMaxConcurrentSearches() {
         assertThat(result, equalTo(1));
     }
 
-    public void testCancellation() {
+    public void testCancellation() throws InterruptedException {
         // Initialize dependencies of TransportMultiSearchAction
         Settings settings = Settings.builder().put("node.name", TransportMultiSearchActionTests.class.getSimpleName()).build();
         ActionFilters actionFilters = mock(ActionFilters.class);
@@ -385,22 +385,34 @@ public String getLocalNodeId() {
             }
             MultiSearchResponse[] responses = new MultiSearchResponse[1];
             Exception[] exceptions = new Exception[1];
+            CountDownLatch executedLatch = new CountDownLatch(1);
             parentTask[0] = (CancellableTask) action.execute(multiSearchRequest, new TaskListener<>() {
                 @Override
                 public void onResponse(Task task, MultiSearchResponse items) {
                     responses[0] = items;
+                    executedLatch.countDown();
                 }
 
                 @Override
                 public void onFailure(Task task, Exception e) {
                     exceptions[0] = e;
+                    executedLatch.countDown();
                 }
             });
             parentTask[0].cancel("Giving up");
             canceledLatch.countDown();
 
-            assertNull(responses[0]);
-            assertNull(exceptions[0]);
+            if (!executedLatch.await(10, TimeUnit.SECONDS)) {
+                fail("Latch should have counted down");
+            }
+
+            boolean concalled = false;
+            for (MultiSearchResponse.Item item : responses[0].getResponses()) {
+                if (item.isFailure() && item.getFailure().getMessage().contains("Parent task was cancelled")) {
+                    concalled = true;
+                }
+            }
+            assertTrue(concalled);
         } finally {
             assertTrue(OpenSearchTestCase.terminate(threadPool));
         }