QueryConverterTest.java

/*
 * JBoss, Home of Professional Open Source.
 * Copyright 2012, Red Hat, Inc., and individual contributors
 * as indicated by the @author tags. See the copyright.txt file in the
 * distribution for a full listing of individual contributors.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation; either version 2.1 of
 * the License, or (at your option) any later version.
 *
 * This software is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this software; if not, write to the Free
 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */

package org.jboss.test.capedwarf.search.test;

import java.util.Collections;

import org.apache.lucene.analysis.standard.StandardAnalyzer;
import org.apache.lucene.queryParser.ParseException;
import org.apache.lucene.queryParser.QueryParser;
import org.apache.lucene.search.Query;
import org.apache.lucene.util.Version;
import org.jboss.capedwarf.search.QueryConverter;
import org.jboss.test.capedwarf.common.support.All;
import org.junit.Assert;
import org.junit.Test;
import org.junit.experimental.categories.Category;


/**
 * @author <a href="mailto:mluksa@redhat.com">Marko Luksa</a>
 */
@Category(All.class)
public class QueryConverterTest {

    public static final Version LUCENE_VERSION = Version.LUCENE_35;

    @Test
    public void testConversion() throws Exception {
        assertQueryEquals("field:value", "field:value");
        assertQueryEquals("+__ALL_DOCS__:all -field:value", "NOT field:value");
        assertQueryEquals("field:aaa AND field:bbb", "field:aaa field:bbb");
        assertQueryEquals("field:aaa AND field:bbb AND field:ccc", "field:aaa field:bbb field:ccc");
        assertQueryEquals("field:aaa OR field:bbb OR field:ccc", "field:aaa OR field:bbb OR field:ccc");
        assertQueryEquals("field:aaa AND field:bbb AND field:ccc", "field:aaa AND field:bbb AND field:ccc");
        assertQueryEquals("author:rose OR (__ALL_DOCS__:all AND NOT body:filigree)", "author:rose OR NOT body:filigree");
        assertQueryEquals("+author:rose +(__ALL_DOCS__:all AND -body:filigree)", "author:rose AND NOT body:filigree");
        assertQueryEquals("+author:rose +(+__ALL_DOCS__:all -(body:filigree author:jones))", "author:rose NOT (body:filigree OR author:jones)");
        assertQueryEquals("+author:rose +(+__ALL_DOCS__:all -(+body:filigree +author:jones))", "author:rose NOT (body:filigree AND author:jones)");

        assertQueryEquals("field:[value TO value]", "field=value");
        assertQueryEquals("all:rose", "rose");
        assertQueryEquals("bob:hope OR bob:dope", "bob:(hope OR dope)");
        assertQueryEquals("bob:hope AND bob:dope", "bob:(hope AND dope)");
        assertQueryEquals("bob:hope AND bob:dope", "bob:(hope dope)");
        assertQueryEquals("bob:hope OR bob:dope", "bob=(hope OR dope)");

        assertQueryEquals("field:[12.0 TO 12.0]", "field=12");
        assertQueryEquals("field:{12.0 TO *}", "field>12");
        assertQueryEquals("field:[12.0 TO *]", "field>=12");
        assertQueryEquals("field:{* TO 12.0}", "field<12");
        assertQueryEquals("field:[* TO 12.0]", "field<=12");

        assertQueryEquals("field:{aaa TO *}", "field>aaa");
        assertQueryEquals("field:[aaa TO *]", "field>=aaa");
        assertQueryEquals("field:[* TO ddd]", "field<=ddd");
        assertQueryEquals("field:{* TO ddd}", "field<ddd");

        assertQueryEquals("field:\"any other name\"", "field:\"any other name\"");
        assertQueryEquals("all:\"any other name\"", "\"any other name\"");
    }

    @Test
    public void testCaseIsHandledCorrectly() throws Exception {
        assertQueryEquals("author:\"Rose Bloom\"", "author:\"Rose Bloom\"");
        assertQueryEquals("author:Rose", "author:Rose");
    }

    @Test
    public void testExamplesFromGAEDocumentation() throws Exception {
        assertQueryEquals("all:rose", "rose");
        assertQueryEquals("all:\"any other name\"", "\"any other name\"");
        assertQueryEquals("field:value", "field:value");
        assertQueryEquals("field:\"value as a string\"", "field:\"value as a string\"");
        assertQueryEquals("author:rose", "author:rose");
        assertQueryEquals("body:\"any other name\"", "body:\"any other name\"");
        assertQueryEquals("author:\"Rose Jones\" AND body:rose", "author:\"Rose Jones\" body:rose");
        assertQueryEquals("price:{* TO 100.0}", "price<100");
        assertQueryEquals("sent:[2011-02-28 TO *]", "sent>=2011-02-28");
        assertQueryEquals("product_code:[xyz1000 TO xyz1000]", "product_code = xyz1000");

        assertQueryEquals("author:bob OR ((author:rose OR author:tom) AND author:jones)", "author:(bob OR ((rose OR tom) AND jones))");
        assertQueryEquals("author:rose AND (__ALL_DOCS__:all AND NOT body:filigree)", "author:rose NOT body:filigree");
        assertQueryEquals("(author:Thomas OR author:Jones) AND (__ALL_DOCS__:all AND NOT body:rose)", "(author:Thomas OR author:Jones) AND (NOT body:rose)");
    }

    private static void assertQueryEquals(String expectedLuceneQueryString, String gaeQueryString) throws ParseException {
        Query query = new QueryConverter("all").convert(gaeQueryString);
        Query expectedQuery = new QueryParser(LUCENE_VERSION, null, new StandardAnalyzer(LUCENE_VERSION, Collections.emptySet())).parse(expectedLuceneQueryString);
        System.out.println("expectedQuery = " + expectedQuery + "    (" + expectedQuery.getClass() + ")");
        System.out.println("query = " + query + "    (" + query.getClass() + ")");
        Assert.assertEquals(expectedQuery.toString(), query.toString());
    }


}