From de8e7e20422affb543a5d7d8493228c531199402 Mon Sep 17 00:00:00 2001
From: Rishikesh1159 <rishireddy1159@gmail.com>
Date: Thu, 8 Feb 2024 00:28:42 +0000
Subject: [PATCH] refactor with usage of waitForReplication().

Signed-off-by: Rishikesh1159 <rishireddy1159@gmail.com>
---
 .../java/org/opensearch/test/OpenSearchIntegTestCase.java  | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/test/framework/src/main/java/org/opensearch/test/OpenSearchIntegTestCase.java b/test/framework/src/main/java/org/opensearch/test/OpenSearchIntegTestCase.java
index e4891fa3707d1..33d5669d33297 100644
--- a/test/framework/src/main/java/org/opensearch/test/OpenSearchIntegTestCase.java
+++ b/test/framework/src/main/java/org/opensearch/test/OpenSearchIntegTestCase.java
@@ -1555,11 +1555,16 @@ public void indexRandom(boolean forceRefresh, boolean dummyDocuments, boolean ma
                 );
             }
         }
+        if (forceRefresh) {
+            assertNoFailures(
+                client().admin().indices().prepareRefresh(indicesArray).setIndicesOptions(IndicesOptions.lenientExpandOpen()).get()
+            );
+        }
         if (dummyDocuments) {
             indexRandomForMultipleSlices(indicesArray);
         }
         if (forceRefresh) {
-            refreshAndWaitForReplication();
+            waitForReplication();
         }
     }