Examples with Result org.opensearch.percolator.QueryAnalyzer.Result used on opensource projects

Example 16 with Result

use of org.opensearch.percolator.QueryAnalyzer.Result in project OpenSearch by opensearch-project.

the class QueryAnalyzerTests method testExtractQueryMetadata_matchAllDocsQuery.

public void testExtractQueryMetadata_matchAllDocsQuery() {
    Result result = analyze(new MatchAllDocsQuery(), Version.CURRENT);
    assertThat(result.verified, is(true));
    assertThat(result.matchAllDocs, is(true));
    assertThat(result.minimumShouldMatch, equalTo(0));
    assertThat(result.extractions.size(), equalTo(0));
    BooleanQuery.Builder builder = new BooleanQuery.Builder();
    builder.add(new TermQuery(new Term("field", "value")), BooleanClause.Occur.MUST);
    builder.add(new MatchAllDocsQuery(), BooleanClause.Occur.MUST);
    result = analyze(builder.build(), Version.CURRENT);
    assertThat(result.verified, is(true));
    assertThat(result.matchAllDocs, is(false));
    assertThat(result.minimumShouldMatch, equalTo(1));
    assertTermsEqual(result.extractions, new Term("field", "value"));
    builder = new BooleanQuery.Builder();
    builder.add(new MatchAllDocsQuery(), BooleanClause.Occur.MUST);
    builder.add(new MatchAllDocsQuery(), BooleanClause.Occur.MUST);
    builder.add(new MatchAllDocsQuery(), BooleanClause.Occur.MUST);
    BooleanQuery bq1 = builder.build();
    result = analyze(bq1, Version.CURRENT);
    assertThat(result.verified, is(true));
    assertThat(result.matchAllDocs, is(true));
    assertThat(result.minimumShouldMatch, equalTo(0));
    assertThat(result.extractions.size(), equalTo(0));
    builder = new BooleanQuery.Builder();
    builder.add(new MatchAllDocsQuery(), BooleanClause.Occur.MUST_NOT);
    builder.add(new MatchAllDocsQuery(), BooleanClause.Occur.MUST);
    builder.add(new MatchAllDocsQuery(), BooleanClause.Occur.MUST);
    BooleanQuery bq2 = builder.build();
    result = analyze(bq2, Version.CURRENT);
    assertThat(result.verified, is(false));
    assertThat(result.matchAllDocs, is(true));
    assertThat(result.minimumShouldMatch, equalTo(0));
    assertThat(result.extractions.size(), equalTo(0));
    builder = new BooleanQuery.Builder();
    builder.add(new MatchAllDocsQuery(), BooleanClause.Occur.SHOULD);
    builder.add(new MatchAllDocsQuery(), BooleanClause.Occur.SHOULD);
    builder.add(new MatchAllDocsQuery(), BooleanClause.Occur.SHOULD);
    BooleanQuery bq3 = builder.build();
    result = analyze(bq3, Version.CURRENT);
    assertThat(result.verified, is(true));
    assertThat(result.matchAllDocs, is(true));
    assertThat(result.minimumShouldMatch, equalTo(0));
    assertThat(result.extractions.size(), equalTo(0));
    builder = new BooleanQuery.Builder();
    builder.add(new MatchAllDocsQuery(), BooleanClause.Occur.MUST_NOT);
    builder.add(new MatchAllDocsQuery(), BooleanClause.Occur.SHOULD);
    builder.add(new MatchAllDocsQuery(), BooleanClause.Occur.SHOULD);
    BooleanQuery bq4 = builder.build();
    result = analyze(bq4, Version.CURRENT);
    assertThat(result.verified, is(false));
    assertThat(result.matchAllDocs, is(true));
    assertThat(result.minimumShouldMatch, equalTo(0));
    assertThat(result.extractions.size(), equalTo(0));
    builder = new BooleanQuery.Builder();
    builder.add(new TermQuery(new Term("field", "value")), BooleanClause.Occur.SHOULD);
    builder.add(new MatchAllDocsQuery(), BooleanClause.Occur.SHOULD);
    BooleanQuery bq5 = builder.build();
    result = analyze(bq5, Version.CURRENT);
    assertThat(result.verified, is(true));
    assertThat(result.matchAllDocs, is(true));
    assertThat(result.minimumShouldMatch, equalTo(0));
    assertThat(result.extractions.size(), equalTo(0));
    builder = new BooleanQuery.Builder();
    builder.add(new MatchAllDocsQuery(), BooleanClause.Occur.SHOULD);
    builder.add(new TermQuery(new Term("field", "value")), BooleanClause.Occur.SHOULD);
    builder.setMinimumNumberShouldMatch(2);
    BooleanQuery bq6 = builder.build();
    result = analyze(bq6, Version.CURRENT);
    assertThat(result.verified, is(true));
    assertThat(result.matchAllDocs, is(false));
    assertThat(result.minimumShouldMatch, equalTo(1));
    assertThat(result.extractions.size(), equalTo(1));
    assertTermsEqual(result.extractions, new Term("field", "value"));
    builder = new BooleanQuery.Builder();
    builder.add(new MatchAllDocsQuery(), BooleanClause.Occur.SHOULD);
    builder.add(new MatchAllDocsQuery(), BooleanClause.Occur.SHOULD);
    builder.add(new TermQuery(new Term("field", "value")), BooleanClause.Occur.SHOULD);
    builder.setMinimumNumberShouldMatch(2);
    BooleanQuery bq7 = builder.build();
    result = analyze(bq7, Version.CURRENT);
    assertThat(result.verified, is(true));
    assertThat(result.matchAllDocs, is(true));
    assertThat(result.minimumShouldMatch, equalTo(0));
    assertThat(result.extractions.size(), equalTo(0));
}

Example 17 with Result

use of org.opensearch.percolator.QueryAnalyzer.Result in project OpenSearch by opensearch-project.

the class QueryAnalyzerTests method testExtractQueryMetadata_multiPhraseQuery.

public void testExtractQueryMetadata_multiPhraseQuery() {
    MultiPhraseQuery multiPhraseQuery = new MultiPhraseQuery.Builder().add(new Term("_field", "_term1")).add(new Term[] { new Term("_field", "_term2"), new Term("_field", "_term3") }).add(new Term[] { new Term("_field", "_term4"), new Term("_field", "_term5") }).add(new Term[] { new Term("_field", "_term6") }).build();
    Result result = analyze(multiPhraseQuery, Version.CURRENT);
    assertThat(result.verified, is(false));
    assertThat(result.minimumShouldMatch, equalTo(4));
    List<QueryExtraction> terms = new ArrayList<>(result.extractions);
    terms.sort(Comparator.comparing(qt -> qt.term));
    assertThat(terms.size(), equalTo(6));
    assertThat(terms.get(0).field(), equalTo("_field"));
    assertThat(terms.get(0).bytes().utf8ToString(), equalTo("_term1"));
    assertThat(terms.get(1).field(), equalTo("_field"));
    assertThat(terms.get(1).bytes().utf8ToString(), equalTo("_term2"));
    assertThat(terms.get(2).field(), equalTo("_field"));
    assertThat(terms.get(2).bytes().utf8ToString(), equalTo("_term3"));
    assertThat(terms.get(3).field(), equalTo("_field"));
    assertThat(terms.get(3).bytes().utf8ToString(), equalTo("_term4"));
    assertThat(terms.get(4).field(), equalTo("_field"));
    assertThat(terms.get(4).bytes().utf8ToString(), equalTo("_term5"));
    assertThat(terms.get(5).field(), equalTo("_field"));
    assertThat(terms.get(5).bytes().utf8ToString(), equalTo("_term6"));
}

Example 18 with Result

use of org.opensearch.percolator.QueryAnalyzer.Result in project OpenSearch by opensearch-project.

the class QueryAnalyzerTests method testExtractQueryMetadata_matchNoDocsQuery.

public void testExtractQueryMetadata_matchNoDocsQuery() {
    Result result = analyze(new MatchNoDocsQuery("sometimes there is no reason at all"), Version.CURRENT);
    assertThat(result.verified, is(true));
    assertEquals(0, result.extractions.size());
    assertThat(result.minimumShouldMatch, equalTo(0));
    BooleanQuery.Builder bq = new BooleanQuery.Builder();
    bq.add(new TermQuery(new Term("field", "value")), BooleanClause.Occur.MUST);
    bq.add(new MatchNoDocsQuery("sometimes there is no reason at all"), BooleanClause.Occur.MUST);
    result = analyze(bq.build(), Version.CURRENT);
    assertThat(result.verified, is(true));
    assertEquals(0, result.extractions.size());
    assertThat(result.minimumShouldMatch, equalTo(0));
    bq = new BooleanQuery.Builder();
    bq.add(new TermQuery(new Term("field", "value")), BooleanClause.Occur.SHOULD);
    bq.add(new MatchNoDocsQuery("sometimes there is no reason at all"), BooleanClause.Occur.SHOULD);
    result = analyze(bq.build(), Version.CURRENT);
    assertThat(result.verified, is(true));
    assertThat(result.minimumShouldMatch, equalTo(1));
    assertTermsEqual(result.extractions, new Term("field", "value"));
    DisjunctionMaxQuery disjunctionMaxQuery = new DisjunctionMaxQuery(Arrays.asList(new TermQuery(new Term("field", "value")), new MatchNoDocsQuery("sometimes there is no reason at all")), 1f);
    result = analyze(disjunctionMaxQuery, Version.CURRENT);
    assertThat(result.verified, is(true));
    assertThat(result.minimumShouldMatch, equalTo(1));
    assertTermsEqual(result.extractions, new Term("field", "value"));
}

Example 19 with Result

use of org.opensearch.percolator.QueryAnalyzer.Result in project OpenSearch by opensearch-project.

the class QueryAnalyzerTests method testFunctionScoreQuery_withMatchAll.

public void testFunctionScoreQuery_withMatchAll() {
    MatchAllDocsQuery innerQuery = new MatchAllDocsQuery();
    FunctionScoreQuery functionScoreQuery1 = new FunctionScoreQuery(innerQuery, new RandomScoreFunction(0, 0, null));
    Result result = analyze(functionScoreQuery1, Version.CURRENT);
    assertThat(result.verified, is(true));
    assertThat(result.minimumShouldMatch, equalTo(0));
    assertThat(result.matchAllDocs, is(true));
    assertThat(result.extractions.isEmpty(), is(true));
    FunctionScoreQuery functionScoreQuery2 = new FunctionScoreQuery(innerQuery, new RandomScoreFunction(0, 0, null), CombineFunction.MULTIPLY, 1f, 10f);
    result = analyze(functionScoreQuery2, Version.CURRENT);
    assertThat(result.verified, is(false));
    assertThat(result.minimumShouldMatch, equalTo(0));
    assertThat(result.matchAllDocs, is(true));
    assertThat(result.extractions.isEmpty(), is(true));
}

Example 20 with Result

use of org.opensearch.percolator.QueryAnalyzer.Result in project OpenSearch by opensearch-project.

the class QueryAnalyzerTests method testExtractQueryMetadata_phraseQuery.

public void testExtractQueryMetadata_phraseQuery() {
    PhraseQuery phraseQuery = new PhraseQuery("_field", "_term1", "term2");
    Result result = analyze(phraseQuery, Version.CURRENT);
    assertThat(result.verified, is(false));
    assertThat(result.minimumShouldMatch, equalTo(2));
    List<QueryExtraction> terms = new ArrayList<>(result.extractions);
    terms.sort(Comparator.comparing(qt -> qt.term));
    assertThat(terms.size(), equalTo(2));
    assertThat(terms.get(0).field(), equalTo(phraseQuery.getTerms()[0].field()));
    assertThat(terms.get(0).bytes(), equalTo(phraseQuery.getTerms()[0].bytes()));
    assertThat(terms.get(1).field(), equalTo(phraseQuery.getTerms()[1].field()));
    assertThat(terms.get(1).bytes(), equalTo(phraseQuery.getTerms()[1].bytes()));
}