| package org.apache.lucene.queryParser; |
| |
| /* ==================================================================== |
| * The Apache Software License, Version 1.1 |
| * |
| * Copyright (c) 2001 The Apache Software Foundation. All rights |
| * reserved. |
| * |
| * Redistribution and use in source and binary forms, with or without |
| * modification, are permitted provided that the following conditions |
| * are met: |
| * |
| * 1. Redistributions of source code must retain the above copyright |
| * notice, this list of conditions and the following disclaimer. |
| * |
| * 2. Redistributions in binary form must reproduce the above copyright |
| * notice, this list of conditions and the following disclaimer in |
| * the documentation and/or other materials provided with the |
| * distribution. |
| * |
| * 3. The end-user documentation included with the redistribution, |
| * if any, must include the following acknowledgment: |
| * "This product includes software developed by the |
| * Apache Software Foundation (http://www.apache.org/)." |
| * Alternately, this acknowledgment may appear in the software itself, |
| * if and wherever such third-party acknowledgments normally appear. |
| * |
| * 4. The names "Apache" and "Apache Software Foundation" and |
| * "Apache Lucene" must not be used to endorse or promote products |
| * derived from this software without prior written permission. For |
| * written permission, please contact apache@apache.org. |
| * |
| * 5. Products derived from this software may not be called "Apache", |
| * "Apache Lucene", nor may "Apache" appear in their name, without |
| * prior written permission of the Apache Software Foundation. |
| * |
| * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED |
| * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES |
| * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE |
| * DISCLAIMED. IN NO EVENT SHALL THE APACHE SOFTWARE FOUNDATION OR |
| * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF |
| * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND |
| * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, |
| * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT |
| * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF |
| * SUCH DAMAGE. |
| * ==================================================================== |
| * |
| * This software consists of voluntary contributions made by many |
| * individuals on behalf of the Apache Software Foundation. For more |
| * information on the Apache Software Foundation, please see |
| * <http://www.apache.org/>. |
| */ |
| |
| import java.io.*; |
| import junit.framework.*; |
| |
| import org.apache.lucene.*; |
| import org.apache.lucene.queryParser.*; |
| import org.apache.lucene.search.*; |
| import org.apache.lucene.analysis.*; |
| import org.apache.lucene.analysis.Token; |
| |
| public class TestQueryParser extends TestCase { |
| |
| public TestQueryParser(String name) { |
| super(name); |
| } |
| |
| public static Analyzer qpAnalyzer = new QPTestAnalyzer(); |
| |
| public static class QPTestFilter extends TokenFilter { |
| |
| /** |
| * Filter which discards the token 'stop' and which expands the |
| * token 'phrase' into 'phrase1 phrase2' |
| */ |
| public QPTestFilter(TokenStream in) { |
| input = in; |
| } |
| |
| boolean inPhrase = false; |
| int savedStart=0, savedEnd=0; |
| |
| public Token next() throws IOException { |
| if (inPhrase) { |
| inPhrase = false; |
| return new Token("phrase2", savedStart, savedEnd); |
| } |
| else |
| for (Token token = input.next(); token != null; token = input.next()) |
| if (token.termText().equals("phrase")) { |
| inPhrase = true; |
| savedStart = token.startOffset(); |
| savedEnd = token.endOffset(); |
| return new Token("phrase1", savedStart, savedEnd); |
| } |
| else if (!token.termText().equals("stop")) |
| return token; |
| return null; |
| } |
| } |
| |
| public static class QPTestAnalyzer extends Analyzer { |
| |
| public QPTestAnalyzer() { |
| } |
| |
| /** Filters LowerCaseTokenizer with StopFilter. */ |
| public final TokenStream tokenStream(String fieldName, Reader reader) { |
| return new QPTestFilter(new LowerCaseTokenizer(reader)); |
| } |
| } |
| |
| /** |
| * initialize this TemplateTester by creating a WebMacro instance |
| * and a default Context. |
| */ |
| public void init () throws Exception |
| { |
| } |
| |
| public void assertQueryEquals(String query, Analyzer a, String result) |
| throws Exception { |
| if (a == null) |
| a = new SimpleAnalyzer(); |
| QueryParser qp = new QueryParser("field", a); |
| Query q = qp.parse(query); |
| String s = q.toString("field"); |
| if (!s.equals(result)) { |
| System.err.println("Query /" + query + "/ yielded /" + s |
| + "/, expecting /" + result + "/"); |
| assert(false); |
| } |
| } |
| |
| public void testSimple() throws Exception { |
| assertQueryEquals("term term term", null, "term term term"); |
| assertQueryEquals("term term1 term2", null, "term term term"); |
| assertQueryEquals("term 1.0 1 2", null, "term"); |
| |
| assertQueryEquals("a AND b", null, "+a +b"); |
| assertQueryEquals("a AND NOT b", null, "+a -b"); |
| assertQueryEquals("a AND -b", null, "+a -b"); |
| assertQueryEquals("a AND !b", null, "+a -b"); |
| assertQueryEquals("a && b", null, "+a +b"); |
| assertQueryEquals("a&&b", null, "+a +b"); |
| assertQueryEquals("a && ! b", null, "+a -b"); |
| |
| assertQueryEquals("a OR b", null, "a b"); |
| assertQueryEquals("a || b", null, "a b"); |
| assertQueryEquals("a OR !b", null, "a -b"); |
| assertQueryEquals("a OR ! b", null, "a -b"); |
| assertQueryEquals("a OR -b", null, "a -b"); |
| |
| assertQueryEquals("+term -term term", null, "+term -term term"); |
| assertQueryEquals("foo:term AND field:anotherTerm", null, |
| "+foo:term +anotherterm"); |
| assertQueryEquals("term AND \"phrase phrase\"", null, |
| "+term +\"phrase phrase\""); |
| |
| assertQueryEquals("germ term^2.0", null, "germ term^2.0"); |
| assertQueryEquals("term^2.0", null, "term^2.0"); |
| |
| assertQueryEquals("(foo OR bar) AND (baz OR boo)", null, |
| "+(foo bar) +(baz boo)"); |
| assertQueryEquals("((a OR b) AND NOT c) OR d", null, |
| "(+(a b) -c) d"); |
| assertQueryEquals("+(apple \"steve jobs\") -(foo bar baz)", null, |
| "+(apple \"steve jobs\") -(foo bar baz)"); |
| assertQueryEquals("+title:(dog OR cat) -author:\"bob dole\"", null, |
| "+(title:dog title:cat) -author:\"bob dole\""); |
| } |
| |
| public void testQPA() throws Exception { |
| assertQueryEquals("term term term", qpAnalyzer, "term term term"); |
| assertQueryEquals("term +stop term", qpAnalyzer, "term term"); |
| assertQueryEquals("term -stop term", qpAnalyzer, "term term"); |
| assertQueryEquals("drop AND stop AND roll", qpAnalyzer, "+drop +roll"); |
| assertQueryEquals("term phrase term", qpAnalyzer, |
| "term \"phrase1 phrase2\" term"); |
| assertQueryEquals("term AND NOT phrase term", qpAnalyzer, |
| "+term -\"phrase1 phrase2\" term"); |
| assertQueryEquals("stop", qpAnalyzer, ""); |
| } |
| } |
| |