Fix tests

Signed-off-by: Harsha Vamsi Kalluri <[email protected]>
opensearch-project · Nov 16, 2023 · bbd4e4e · bbd4e4e
1 parent d97865e
commit bbd4e4e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/server/src/test/java/org/opensearch/index/query/TermQueryBuilderTests.java b/server/src/test/java/org/opensearch/index/query/TermQueryBuilderTests.java
@@ -36,6 +36,7 @@
 
 import org.apache.lucene.index.Term;
 import org.apache.lucene.search.AutomatonQuery;
+import org.apache.lucene.search.IndexOrDocValuesQuery;
 import org.apache.lucene.search.MatchNoDocsQuery;
 import org.apache.lucene.search.PointRangeQuery;
 import org.apache.lucene.search.Query;
@@ -111,7 +112,7 @@ protected void doAssertLuceneQuery(TermQueryBuilder queryBuilder, Query query, Q
             query,
             either(instanceOf(TermQuery.class)).or(instanceOf(PointRangeQuery.class))
                 .or(instanceOf(MatchNoDocsQuery.class))
-                .or(instanceOf(AutomatonQuery.class))
+                .or(instanceOf(AutomatonQuery.class)).or(instanceOf(IndexOrDocValuesQuery.class))
         );
         MappedFieldType mapper = context.fieldMapper(queryBuilder.fieldName());
         if (query instanceof TermQuery) {