Add unit test for deprecated constructor

Signed-off-by: Andrew Ross <[email protected]>
opensearch-project · Feb 21, 2024 · caea317 · caea317
1 parent 02af284
commit caea317
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 3 deletions.
diff --git a/server/src/test/java/org/opensearch/index/query/QueryShardContextTests.java b/server/src/test/java/org/opensearch/index/query/QueryShardContextTests.java
@@ -91,6 +91,8 @@
 
 public class QueryShardContextTests extends OpenSearchTestCase {
 
+    private static final int SHARD_ID = 0;
+
     public void testFailIfFieldMappingNotFound() {
         QueryShardContext context = createQueryShardContext(IndexMetadata.INDEX_UUID_NA_VALUE, null);
         context.setAllowUnmappedFields(false);
@@ -307,9 +309,9 @@ public void testFielddataLookupOneFieldManyReferences() throws IOException {
         assertEquals(Arrays.asList(expectedFirstDoc.toString(), expectedSecondDoc.toString()), collect("field", queryShardContext));
     }
 
-    public void testSearchLookupShardId() throws IOException {
+    public void testSearchLookupShardId() {
         SearchLookup searchLookup = createQueryShardContext("uuid", null, null).lookup();
-        assertEquals(0, searchLookup.shardId());
+        assertEquals(SHARD_ID, searchLookup.shardId());
     }
 
     public static QueryShardContext createQueryShardContext(String indexUuid, String clusterAlias) {
@@ -348,7 +350,7 @@ private static QueryShardContext createQueryShardContext(
         }
         final long nowInMillis = randomNonNegativeLong();
         return new QueryShardContext(
-            0,
+            SHARD_ID,
             indexSettings,
             BigArrays.NON_RECYCLING_INSTANCE,
             null,

diff --git a/server/src/test/java/org/opensearch/search/lookup/SearchLookupTest.java b/server/src/test/java/org/opensearch/search/lookup/SearchLookupTest.java
@@ -0,0 +1,22 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * The OpenSearch Contributors require contributions made to
+ * this file be licensed under the Apache-2.0 license or a
+ * compatible open source license.
+ */
+
+package org.opensearch.search.lookup;
+
+import org.opensearch.index.mapper.MapperService;
+import org.opensearch.test.OpenSearchTestCase;
+
+import static org.mockito.Mockito.mock;
+
+public class SearchLookupTest extends OpenSearchTestCase {
+
+    public void testDeprecatedConstructorShardId() {
+        final SearchLookup searchLookup = new SearchLookup(mock(MapperService.class), (a,b) -> null);
+        assertThrows(IllegalStateException.class, searchLookup::shardId);
+    }
+}