Skip to content

Commit 40b3e60

Browse files
committed
fix BoolQueryBuilderTests.testToQuery
Signed-off-by: guojialiang <[email protected]>
1 parent 1be3e46 commit 40b3e60

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

server/src/test/java/org/opensearch/index/query/BoolQueryBuilderTests.java

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,8 @@ protected BoolQueryBuilder doCreateTestQueryBuilder() {
9595
@Override
9696
protected void doAssertLuceneQuery(BoolQueryBuilder queryBuilder, Query query, QueryShardContext context) throws IOException {
9797
if (!queryBuilder.hasClauses()) {
98-
assertThat(query, instanceOf(MatchAllDocsQuery.class));
98+
assertThat(query, instanceOf(ApproximateScoreQuery.class));
99+
assertThat(((ApproximateScoreQuery) query).getOriginalQuery(), instanceOf(MatchAllDocsQuery.class));
99100
} else {
100101
List<BooleanClause> clauses = new ArrayList<>();
101102
clauses.addAll(getBooleanClauses(queryBuilder.must(), BooleanClause.Occur.MUST, context));
@@ -104,7 +105,8 @@ protected void doAssertLuceneQuery(BoolQueryBuilder queryBuilder, Query query, Q
104105
clauses.addAll(getBooleanClauses(queryBuilder.filter(), BooleanClause.Occur.FILTER, context));
105106

106107
if (clauses.isEmpty()) {
107-
assertThat(query, instanceOf(MatchAllDocsQuery.class));
108+
assertThat(query, instanceOf(ApproximateScoreQuery.class));
109+
assertThat(((ApproximateScoreQuery) query).getOriginalQuery(), instanceOf(MatchAllDocsQuery.class));
108110
} else if (query instanceof MatchNoDocsQuery == false) {
109111
assertThat(query, instanceOf(BooleanQuery.class));
110112
BooleanQuery booleanQuery = (BooleanQuery) query;

0 commit comments

Comments
 (0)