[Bug] Wait for outstanding requests to complete (#1925)

Signed-off-by: Suraj Singh <[email protected]>
opensearch-project · Jan 18, 2022 · 7691f07 · 7691f07
1 parent d3a4c47
commit 7691f07
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/...er/src/internalClusterTest/java/org/opensearch/index/ShardIndexingPressureSettingsIT.java b/...er/src/internalClusterTest/java/org/opensearch/index/ShardIndexingPressureSettingsIT.java
@@ -403,6 +403,15 @@ public void testShardIndexingPressureEnforcedEnabledDisabledSetting() throws Exc
         secondSuccessFuture = client(coordinatingOnlyNode).bulk(bulkRequest);
         Thread.sleep(25);
 
+        assertBusy(
+            () -> {
+                assertEquals(
+                    coordinatingShardTracker.getCoordinatingOperationTracker().getPerformanceTracker().getTotalOutstandingRequests(),
+                    2
+                );
+            }
+        );
+
         // This request breaches the threshold and hence will be rejected
         expectThrows(OpenSearchRejectedExecutionException.class, () -> client(coordinatingOnlyNode).bulk(bulkRequest).actionGet());