protected ExpandedQuery makeQuery(String input) { return new ExpandedQuery(new WhiteSpaceQuerqyParser().parse(input)); }
if (requestAdapter.isMatchAllQuery(queryString)) { parsedInput = new ExpandedQuery(new MatchAllQuery()); parserDebugInfo = parser.getClass().getName(); parsedInput = new ExpandedQuery(parser.parse(queryString));
@Test public void testThatShinglingDoesNotTriggerExceptionOnSingleTerm() throws Exception { Query query = new Query(); addTerm(query, "t1"); ExpandedQuery expandedQuery = new ExpandedQuery(query); ShingleRewriter rewriter = new ShingleRewriter(); rewriter.rewrite(expandedQuery); assertThat((Query) expandedQuery.getUserQuery(), bq( dmq( term("t1") ) ) ); }
@Test public void testShinglingForTwoTokensWithOnFieldNameNullDontShingle() { Query query = new Query(); addTerm(query, "f1", "cde"); addTerm(query, "ajk"); ExpandedQuery expandedQuery = new ExpandedQuery(query); ShingleRewriter rewriter = new ShingleRewriter(); rewriter.rewrite(expandedQuery); assertThat((Query) expandedQuery.getUserQuery(), bq(dmq(term("f1", "cde")), dmq(term("ajk")))); }
@Test public void testShinglingForTwoTokensWithDifferentFieldsDontShingle() { Query query = new Query(); addTerm(query, "f1", "cde"); addTerm(query, "f2", "ajk"); ExpandedQuery expandedQuery = new ExpandedQuery(query); ShingleRewriter rewriter = new ShingleRewriter(); rewriter.rewrite(expandedQuery); assertThat((Query) expandedQuery.getUserQuery(), bq(dmq(term("cde")), dmq(term("ajk")))); }
@Test public void testShinglingForTwoTokensWithSameFieldAndGeneratedFlag() { Query query = new Query(); addTerm(query, "f1", "cde", true); addTerm(query, "f1", "ajk", true); ExpandedQuery expandedQuery = new ExpandedQuery(query); ShingleRewriter rewriter = new ShingleRewriter(true); rewriter.rewrite(expandedQuery); assertThat((Query) expandedQuery.getUserQuery(), bq( dmq( term("f1", "cde"), term("f1", "cdeajk") ), dmq( term("f1", "ajk"), term("f1", "cdeajk") ) ) ); }
@Test public void testShinglingForTwoTokensWithSameField() { Query query = new Query(); addTerm(query, "f1", "cde"); addTerm(query, "f1", "ajk"); ExpandedQuery expandedQuery = new ExpandedQuery(query); ShingleRewriter rewriter = new ShingleRewriter(); rewriter.rewrite(expandedQuery); assertThat((Query) expandedQuery.getUserQuery(), bq( dmq( term("f1", "cde"), term("f1", "cdeajk") ), dmq( term("f1", "ajk"), term("f1", "cdeajk") ) ) ); }
@Test public void testShinglingForTwoTokens() { Query query = new Query(); addTerm(query, "cde"); addTerm(query, "ajk"); ExpandedQuery expandedQuery = new ExpandedQuery(query); ShingleRewriter rewriter = new ShingleRewriter(); rewriter.rewrite(expandedQuery); assertThat((Query) expandedQuery.getUserQuery(), bq( dmq( term("cde"), term("cdeajk") ), dmq( term("ajk"), term("cdeajk") ) ) ); }
@Test public void testShinglingForThreeTokensWithMixedFields() { Query query = new Query(); addTerm(query, "f1", "cde"); addTerm(query, "f1", "ajk"); addTerm(query, "f2", "xyz"); ExpandedQuery expandedQuery = new ExpandedQuery(query); ShingleRewriter rewriter = new ShingleRewriter(); rewriter.rewrite(expandedQuery); assertThat((Query) expandedQuery.getUserQuery(), bq( dmq( term("f1", "cde"), term("f1", "cdeajk") ), dmq( term("f1", "ajk"), term("f1", "cdeajk") ), dmq(term("f2", "xyz")) ) ); }
@Test public void testShinglingForThreeTokensWithOneTokenGeneratedIgnoringGenerated() { Query query = new Query(); addTerm(query, "cde", false); addTerm(query, "ajk", false); addTerm(query, "xyz", true); ExpandedQuery expandedQuery = new ExpandedQuery(query); ShingleRewriter rewriter = new ShingleRewriter(false); rewriter.rewrite(expandedQuery); assertThat((Query) expandedQuery.getUserQuery(), bq( dmq( term("cde"), term("cdeajk") ), dmq( term("ajk"), term("cdeajk") ), dmq( term("xyz") ) ) ); }
addTerm(query, "xyz", false); ExpandedQuery expandedQuery = new ExpandedQuery(query); ShingleRewriter rewriter = new ShingleRewriter(false); ExpandedQuery rewritten = rewriter.rewrite(expandedQuery);
@Test public void testShinglingForThreeTokens() { Query query = new Query(); addTerm(query, "cde"); addTerm(query, "ajk"); addTerm(query, "xyz"); ExpandedQuery expandedQuery = new ExpandedQuery(query); ShingleRewriter rewriter = new ShingleRewriter(); rewriter.rewrite(expandedQuery); assertThat((Query) expandedQuery.getUserQuery(), bq( dmq( term("cde"), term("cdeajk") ), dmq( term("ajk"), term("cdeajk"), term("ajkxyz") ), dmq( term("xyz"), term("ajkxyz") ) ) ); }
@Test public void testShinglingForThreeTokensWithThreeTokenGenerated() { Query query = new Query(); addTerm(query, "cde", true); addTerm(query, "ajk", true); addTerm(query, "xyz", true); ExpandedQuery expandedQuery = new ExpandedQuery(query); ShingleRewriter rewriter = new ShingleRewriter(true); rewriter.rewrite(expandedQuery); assertThat((Query) expandedQuery.getUserQuery(), bq( dmq( term("cde"), term("cdeajk") ), dmq( term("ajk"), term("cdeajk"), term("ajkxyz") ), dmq( term("xyz"), term("ajkxyz") ) ) ); }