Fixing assert statements

Signed-off-by: Harsha Vamsi Kalluri <[email protected]>
harshavamsi · Jan 3, 2024 · 0065a1e · 0065a1e
1 parent aa3ac06
commit 0065a1e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/server/src/test/java/org/opensearch/index/query/SimpleQueryStringBuilderTests.java b/server/src/test/java/org/opensearch/index/query/SimpleQueryStringBuilderTests.java
@@ -316,8 +316,10 @@ protected void doAssertLuceneQuery(SimpleQueryStringBuilder queryBuilder, Query
                 if (disjunct instanceof BoostQuery) {
                     termQuery = ((BoostQuery) disjunct).getQuery();
                 }
-                if (termQuery instanceof TermQuery) {
-                    TermQuery inner = (TermQuery) termQuery;
+                if (termQuery instanceof TermQuery || termQuery instanceof IndexOrDocValuesQuery) {
+                    TermQuery inner;
+                    if (termQuery instanceof IndexOrDocValuesQuery) inner = (TermQuery) ((IndexOrDocValuesQuery) termQuery).getIndexQuery();
+                    else inner = (TermQuery) termQuery;
                     assertThat(inner.getTerm().bytes().toString(), is(inner.getTerm().bytes().toString().toLowerCase(Locale.ROOT)));
                 } else {
                     assertThat(termQuery, instanceOf(MatchNoDocsQuery.class));

diff --git a/server/src/test/java/org/opensearch/index/search/MultiMatchQueryTests.java b/server/src/test/java/org/opensearch/index/search/MultiMatchQueryTests.java
@@ -364,6 +364,6 @@ public void testKeywordSplitQueriesOnWhitespace() throws IOException {
             ),
             0.0f
         );
-        assertThat(query, equalTo(expected));
+        assertThat(expected, equalTo(query));
     }
 }
-Original file line number
+Diff line change
@@ Expand Up @@
                 ),
 .0f
             );
-            assertThat(query, equalTo(expected));
+            assertThat(expected, equalTo(query));
         }
     }