Fix unit test (opensearch-project#9099)

Signed-off-by: Suraj Singh <surajrider@gmail.com>
neetikasinghal · Aug 3, 2023 · ae9373a · ae9373a
1 parent b9c5a0d
commit ae9373a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/server/src/test/java/org/opensearch/index/store/RemoteSegmentStoreDirectoryTests.java b/server/src/test/java/org/opensearch/index/store/RemoteSegmentStoreDirectoryTests.java
@@ -766,7 +766,8 @@ public void testUploadMetadataMissingSegment() throws IOException {
         IndexOutput indexOutput = mock(IndexOutput.class);
 
         String generation = RemoteStoreUtils.invertLong(segmentInfos.getGeneration());
-        String primaryTerm = RemoteStoreUtils.invertLong(12);
+        long primaryTermLong = indexShard.getLatestReplicationCheckpoint().getPrimaryTerm();
+        String primaryTerm = RemoteStoreUtils.invertLong(primaryTermLong);
         when(storeDirectory.createOutput(startsWith("metadata__" + primaryTerm + "__" + generation), eq(IOContext.DEFAULT))).thenReturn(
             indexOutput
         );