Fix test case around indexordocvalues query

Signed-off-by: Harsha Vamsi Kalluri <[email protected]>
harshavamsi · Oct 13, 2023 · b2d0d7f · b2d0d7f
1 parent 198b443
commit b2d0d7f
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/server/src/test/java/org/opensearch/index/query/MultiMatchQueryBuilderTests.java b/server/src/test/java/org/opensearch/index/query/MultiMatchQueryBuilderTests.java
@@ -32,6 +32,7 @@
 
 package org.opensearch.index.query;
 
+import org.apache.logging.log4j.Level;
 import org.apache.lucene.index.Term;
 import org.apache.lucene.queries.ExtendedCommonTermsQuery;
 import org.apache.lucene.search.BooleanClause;
@@ -42,6 +43,7 @@
 import org.apache.lucene.search.IndexOrDocValuesQuery;
 import org.apache.lucene.search.MatchAllDocsQuery;
 import org.apache.lucene.search.MatchNoDocsQuery;
+import org.apache.lucene.search.MultiTermQuery;
 import org.apache.lucene.search.PhraseQuery;
 import org.apache.lucene.search.PointRangeQuery;
 import org.apache.lucene.search.PrefixQuery;
@@ -304,10 +306,17 @@ public void testToQueryBooleanPrefixMultipleFields() throws IOException {
                 } else if (disjunct instanceof PrefixQuery) {
                     final PrefixQuery secondDisjunct = (PrefixQuery) disjunct;
                     assertThat(secondDisjunct.getPrefix(), equalTo(new Term(KEYWORD_FIELD_NAME, "foo bar")));
-                } else {
+                }
+                else if (disjunct instanceof IndexOrDocValuesQuery){
+                    final IndexOrDocValuesQuery iodvqDisjunct =
+                        (IndexOrDocValuesQuery) disjunct;
+                    assertThat(iodvqDisjunct.getIndexQuery().toString(), equalTo("mapped_string_2:foo bar*"));
+                }
+                else {
                     throw new AssertionError();
                 }
-                assertThat(disjunct, either(instanceOf(BooleanQuery.class)).or(instanceOf(PrefixQuery.class)));
+                assertThat(disjunct,
+                    either(instanceOf(BooleanQuery.class)).or(instanceOf(PrefixQuery.class)).or(instanceOf(IndexOrDocValuesQuery.class)));
             }
         }
     }