Fix flaky test `SearchRestCancellationIT.testAutomaticCancellationMul…

…tiSearchDuringFetchPhase` (opensearch-project#7985) * add latch.await Signed-off-by: Stephen Crawford <[email protected]> * remove space Signed-off-by: Stephen Crawford <[email protected]> * Fix await timing Signed-off-by: Stephen Crawford <[email protected]> * fix line Signed-off-by: Stephen Crawford <[email protected]> * shorten await Signed-off-by: Stephen Crawford <[email protected]> --------- Signed-off-by: Stephen Crawford <[email protected]> Signed-off-by: Rishab Nahata <[email protected]>
imRishN · Jun 27, 2023 · 2a4cc8e · 2a4cc8e
1 parent bc8eadd
commit 2a4cc8e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/qa/smoke-test-http/src/test/java/org/opensearch/http/SearchRestCancellationIT.java b/qa/smoke-test-http/src/test/java/org/opensearch/http/SearchRestCancellationIT.java
@@ -83,6 +83,7 @@
 import static org.opensearch.test.hamcrest.OpenSearchAssertions.assertNoFailures;
 import static org.hamcrest.Matchers.greaterThan;
 import static org.hamcrest.Matchers.instanceOf;
+import static org.opensearch.test.hamcrest.OpenSearchAssertions.awaitLatch;
 
 public class SearchRestCancellationIT extends HttpSmokeTestCase {
 
@@ -183,6 +184,7 @@ public void onFailure(Exception exception) {
             }
         });
 
+        latch.await(2, TimeUnit.SECONDS);
         awaitForBlock(plugins);
         cancellable.cancel();
         ensureSearchTaskIsCancelled(searchAction, nodeIdToName::get);