changes to test

Signed-off-by: Poojita Raj <poojiraj@amazon.com>
opensearch-project · Nov 4, 2022 · eeeae29 · eeeae29
1 parent 80ef959
commit eeeae29
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 9 deletions.
diff --git a/...src/internalClusterTest/java/org/opensearch/indices/replication/SegmentReplicationIT.java b/...src/internalClusterTest/java/org/opensearch/indices/replication/SegmentReplicationIT.java
@@ -245,7 +245,7 @@ public void testIndexReopenClose() throws Exception {
         client().prepareIndex(INDEX_NAME).setId("1").setSource("foo", "bar").setRefreshPolicy(WriteRequest.RefreshPolicy.IMMEDIATE).get();
         refresh(INDEX_NAME);
 
-        final int initialDocCount = scaledRandomIntBetween(10000, 200000);
+        final int initialDocCount = scaledRandomIntBetween(100, 200);
         try (
             BackgroundIndexer indexer = new BackgroundIndexer(
                 INDEX_NAME,
@@ -259,20 +259,13 @@ public void testIndexReopenClose() throws Exception {
         ) {
             indexer.start(initialDocCount);
             waitForDocs(initialDocCount, indexer);
-            refresh(INDEX_NAME);
+            flush(INDEX_NAME);
             waitForReplicaUpdate();
         }
 
-        flushAndRefresh(INDEX_NAME);
-        waitForReplicaUpdate();
-
         logger.info("--> Closing the index ");
         client().admin().indices().prepareClose(INDEX_NAME).get();
 
-        // Add another node to kick off TransportNodesListGatewayStartedShards which fetches latestReplicationCheckpoint for SegRep enabled
-        // indices
-        final String replica2 = internalCluster().startNode();
-
         logger.info("--> Opening the index");
         client().admin().indices().prepareOpen(INDEX_NAME).get();
     }

diff --git a/server/src/main/java/org/opensearch/index/engine/NRTReplicationEngine.java b/server/src/main/java/org/opensearch/index/engine/NRTReplicationEngine.java
@@ -132,6 +132,7 @@ public synchronized void updateSegments(final SegmentInfos infos, long seqNo) th
             // lower/higher gens are possible from a new primary that was just elected.
             if (incomingGeneration != lastReceivedGen) {
                 commitSegmentInfos();
+                translogManager.getDeletionPolicy().setLocalCheckpointOfSafeCommit(seqNo);
                 translogManager.rollTranslogGeneration();
             }
             lastReceivedGen = incomingGeneration;