From 4aa121dd194920c069a64db1f7fdd24c59d1a99a Mon Sep 17 00:00:00 2001
From: Ashish Singh <ssashish@amazon.com>
Date: Mon, 8 Jan 2024 22:50:08 +0530
Subject: [PATCH] Fix flaky test #11256

Signed-off-by: Ashish Singh <ssashish@amazon.com>
---
 .../opensearch/index/shard/RemoteStoreRefreshListenerTests.java | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/server/src/test/java/org/opensearch/index/shard/RemoteStoreRefreshListenerTests.java b/server/src/test/java/org/opensearch/index/shard/RemoteStoreRefreshListenerTests.java
index 811d6a722d0f6..577520823d275 100644
--- a/server/src/test/java/org/opensearch/index/shard/RemoteStoreRefreshListenerTests.java
+++ b/server/src/test/java/org/opensearch/index/shard/RemoteStoreRefreshListenerTests.java
@@ -418,7 +418,7 @@ public void testTrackerData() throws Exception {
         RemoteStoreRefreshListener listener = tuple.v1();
         RemoteStoreStatsTrackerFactory trackerFactory = tuple.v2();
         RemoteSegmentTransferTracker tracker = trackerFactory.getRemoteSegmentTransferTracker(indexShard.shardId());
-        assertNoLag(tracker);
+        assertBusy(() -> assertNoLag(tracker));
         indexDocs(100, randomIntBetween(100, 200));
         indexShard.refresh("test");
         listener.afterRefresh(true);