From 9357b6192e3c107844b4cbe9bacd3502a472e58d Mon Sep 17 00:00:00 2001
From: Suraj Singh <surajrider@gmail.com>
Date: Thu, 3 Aug 2023 14:42:15 -0700
Subject: [PATCH] [Remote Store] Add missing unit test in main (#9101)

Signed-off-by: Suraj Singh <surajrider@gmail.com>
---
 .../RemoteSegmentStoreDirectoryTests.java     | 29 +++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/server/src/test/java/org/opensearch/index/store/RemoteSegmentStoreDirectoryTests.java b/server/src/test/java/org/opensearch/index/store/RemoteSegmentStoreDirectoryTests.java
index 3b2e33388925a..f2d39b2ac7bee 100644
--- a/server/src/test/java/org/opensearch/index/store/RemoteSegmentStoreDirectoryTests.java
+++ b/server/src/test/java/org/opensearch/index/store/RemoteSegmentStoreDirectoryTests.java
@@ -758,6 +758,35 @@ public void testUploadMetadataNonEmpty() throws IOException {
         }
     }
 
+    public void testUploadMetadataMissingSegment() throws IOException {
+        populateMetadata();
+        remoteSegmentStoreDirectory.init();
+
+        Directory storeDirectory = mock(Directory.class);
+        IndexOutput indexOutput = mock(IndexOutput.class);
+
+        String generation = RemoteStoreUtils.invertLong(segmentInfos.getGeneration());
+        long primaryTermLong = indexShard.getLatestReplicationCheckpoint().getPrimaryTerm();
+        String primaryTerm = RemoteStoreUtils.invertLong(primaryTermLong);
+        when(storeDirectory.createOutput(startsWith("metadata__" + primaryTerm + "__" + generation), eq(IOContext.DEFAULT))).thenReturn(
+            indexOutput
+        );
+
+        Collection<String> segmentFiles = List.of("_123.si");
+        assertThrows(
+            NoSuchFileException.class,
+            () -> remoteSegmentStoreDirectory.uploadMetadata(
+                segmentFiles,
+                segmentInfos,
+                storeDirectory,
+                12L,
+                indexShard.getLatestReplicationCheckpoint()
+            )
+        );
+        verify(indexOutput).close();
+        verify(storeDirectory).deleteFile(startsWith("metadata__" + primaryTerm + "__" + generation));
+    }
+
     public void testUploadMetadataNoSegmentCommitInfos() throws IOException {
         SegmentInfos segInfos = indexShard.store().readLastCommittedSegmentsInfo();
         int numSegCommitInfos = segInfos.size();