ingestionEngine creation failure tes

Signed-off-by: Yupeng Fu <yupeng@uber.com>
yupeng9 · Jan 26, 2025 · f35df02 · f35df02
1 parent c38c302
commit f35df02
Showing 1 changed file with 32 additions and 0 deletions.
diff --git a/server/src/test/java/org/opensearch/index/engine/IngestionEngineTests.java b/server/src/test/java/org/opensearch/index/engine/IngestionEngineTests.java
@@ -36,6 +36,8 @@
 import java.util.concurrent.atomic.AtomicLong;
 
 import static org.awaitility.Awaitility.await;
+import static org.mockito.Mockito.doThrow;
+import static org.mockito.Mockito.mock;
 
 public class IngestionEngineTests extends EngineTestCase {
 
@@ -122,6 +124,36 @@ public void testRecovery() throws IOException {
         waitForResults(ingestionEngine, 4);
     }
 
+    public void testCreationFailure() throws IOException {
+        // Simulate an error scenario
+        Store mockStore = mock(Store.class);
+        doThrow(new IOException("Simulated IOException")).when(mockStore).readLastCommittedSegmentsInfo();
+
+        final AtomicLong globalCheckpoint = new AtomicLong(SequenceNumbers.NO_OPS_PERFORMED);
+
+        FakeIngestionSource.FakeIngestionConsumerFactory consumerFactory = new FakeIngestionSource.FakeIngestionConsumerFactory(messages);
+        EngineConfig engineConfig = config(
+            indexSettings,
+            store,
+            createTempDir(),
+            NoMergePolicy.INSTANCE,
+            null,
+            null,
+            globalCheckpoint::get
+        );
+        // overwrite the config with ingestion engine settings
+        String mapping = "{\"properties\":{\"name\":{\"type\": \"text\"},\"age\":{\"type\": \"integer\"}}}}";
+        MapperService mapperService = createMapperService(mapping);
+        engineConfig = config(engineConfig, () -> new DocumentMapperForType(mapperService.documentMapper(), null));
+        try {
+            new IngestionEngine(engineConfig, consumerFactory);
+            fail("Expected EngineException to be thrown");
+        } catch (EngineException e) {
+            assertEquals("failed to create engine", e.getMessage());
+            assertTrue(e.getCause() instanceof IOException);
+        }
+    }
+
     private IngestionEngine buildIngestionEngine(AtomicLong globalCheckpoint, Store store, IndexSettings settings) throws IOException {
         FakeIngestionSource.FakeIngestionConsumerFactory consumerFactory = new FakeIngestionSource.FakeIngestionConsumerFactory(messages);
         EngineConfig engineConfig = config(settings, store, createTempDir(), NoMergePolicy.INSTANCE, null, null, globalCheckpoint::get);