From 5e7cd006ac9932d38f9814054daed391b809ceac Mon Sep 17 00:00:00 2001
From: Martin Gaievski <gaievski@amazon.com>
Date: Wed, 21 Dec 2022 12:53:52 -0800
Subject: [PATCH] Allow mapping service to be null for scenarious of shard
 recovery from translog (#685) (#687) (#689)

Signed-off-by: Martin Gaievski <gaievski@amazon.com>
    (cherry picked from commit c412c8a403cf2c4b6008df7b36847f2ee329e887)

    Co-authored-by: Martin Gaievski <gaievski@amazon.com>

Signed-off-by: Martin Gaievski <gaievski@amazon.com>

Signed-off-by: Martin Gaievski <gaievski@amazon.com>
---
 .../knn/index/codec/KNNCodecFactory.java      |  2 +-
 .../knn/index/codec/KNNCodecServiceTests.java | 67 +++++++++++++++++++
 2 files changed, 68 insertions(+), 1 deletion(-)
 create mode 100644 src/test/java/org/opensearch/knn/index/codec/KNNCodecServiceTests.java

diff --git a/src/main/java/org/opensearch/knn/index/codec/KNNCodecFactory.java b/src/main/java/org/opensearch/knn/index/codec/KNNCodecFactory.java
index 217bce3e8..41d357fcb 100644
--- a/src/main/java/org/opensearch/knn/index/codec/KNNCodecFactory.java
+++ b/src/main/java/org/opensearch/knn/index/codec/KNNCodecFactory.java
@@ -25,7 +25,7 @@ public class KNNCodecFactory {
     public Codec createKNNCodec(final Codec userCodec) {
         var codec = KNN920Codec.builder()
             .delegate(userCodec)
-            .knnVectorsFormat(new KNN920PerFieldKnnVectorsFormat(Optional.of(mapperService)))
+            .knnVectorsFormat(new KNN920PerFieldKnnVectorsFormat(Optional.ofNullable(mapperService)))
             .build();
         return codec;
     }
diff --git a/src/test/java/org/opensearch/knn/index/codec/KNNCodecServiceTests.java b/src/test/java/org/opensearch/knn/index/codec/KNNCodecServiceTests.java
new file mode 100644
index 000000000..9a90990b1
--- /dev/null
+++ b/src/test/java/org/opensearch/knn/index/codec/KNNCodecServiceTests.java
@@ -0,0 +1,67 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package org.opensearch.knn.index.codec;
+
+import org.apache.lucene.codecs.Codec;
+import org.opensearch.cluster.metadata.IndexMetadata;
+import org.opensearch.common.settings.Settings;
+import org.opensearch.index.Index;
+import org.opensearch.index.IndexSettings;
+import org.opensearch.index.codec.CodecServiceConfig;
+import org.opensearch.index.mapper.MapperService;
+import org.opensearch.knn.KNNTestCase;
+
+import org.apache.logging.log4j.Logger;
+
+import java.util.UUID;
+
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
+/**
+ * Test for KNNCodecService class with focus on codec by name lookup
+ */
+public class KNNCodecServiceTests extends KNNTestCase {
+    private static final String TEST_INDEX = "test-index";
+    private static final int NUM_OF_SHARDS = 1;
+    private static final UUID INDEX_UUID = UUID.randomUUID();
+
+    private IndexSettings indexSettings;
+
+    @Override
+    public void setUp() throws Exception {
+        super.setUp();
+        IndexMetadata indexMetadata = mock(IndexMetadata.class);
+        when(indexMetadata.getIndex()).thenReturn(new Index(TEST_INDEX, INDEX_UUID.toString()));
+        when(indexMetadata.getSettings()).thenReturn(Settings.EMPTY);
+        Settings settings = Settings.builder().put(IndexMetadata.SETTING_NUMBER_OF_SHARDS, Integer.toString(NUM_OF_SHARDS)).build();
+        indexSettings = new IndexSettings(indexMetadata, settings);
+    }
+
+    public void testGetCodecByName() {
+        MapperService mapperService = mock(MapperService.class);
+        Logger loggerMock = mock(Logger.class);
+        CodecServiceConfig codecServiceConfig = new CodecServiceConfig(indexSettings, mapperService, loggerMock);
+        KNNCodecService knnCodecService = new KNNCodecService(codecServiceConfig);
+        Codec codec = knnCodecService.codec("KNN920Codec");
+        assertNotNull(codec);
+    }
+
+    /**
+     * This test case covers scenarios when MapperService is null, for instance this may happen during index.close operation.
+     * In such scenario codec is not really required but is created as part of engine initialization, please check code references below:
+     * @see <a href="https://github.com/opensearch-project/OpenSearch/blob/main/server/src/main/java/org/opensearch/index/engine/EngineConfig.java#L378">EngineConfig.java</a>
+     * @see <a href="https://github.com/opensearch-project/OpenSearch/blob/main/server/src/main/java/org/opensearch/index/shard/IndexShard.java#L3315">IndexShard.java</a>
+     * @see <a href="https://github.com/opensearch-project/OpenSearch/blob/main/server/src/main/java/org/opensearch/index/engine/Engine.java#L906">Engine.java</a>
+     */
+    public void testGetCodecByNameWithNoMapperService() {
+        Logger loggerMock = mock(Logger.class);
+        CodecServiceConfig codecServiceConfig = new CodecServiceConfig(indexSettings, null, loggerMock);
+        KNNCodecService knnCodecService = new KNNCodecService(codecServiceConfig);
+        Codec codec = knnCodecService.codec("KNN920Codec");
+        assertNotNull(codec);
+    }
+}