0.8.1-rc1/xmlps/src/test/java/org/apache/oodt/xmlps/queryparser/TestHandlerQueryParser.java

/**
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.apache.oodt.xmlps.queryparser;

//APACHE imports
import org.apache.oodt.xmlps.mapping.Mapping;
import org.apache.oodt.xmlps.mapping.MappingField;
import org.apache.oodt.xmlps.queryparser.Expression;
import org.apache.oodt.xmlps.queryparser.HandlerQueryParser;
import org.apache.oodt.xmlps.util.XMLQueryHelper;
import org.apache.oodt.xmlquery.QueryElement;
import org.apache.oodt.xmlquery.XMLQuery;

//JDK imports
import java.util.List;
import java.util.Stack;

//Junit imports
import junit.framework.TestCase;

/**
 * Tests the XMLPS query handler parser.
 */
public class TestHandlerQueryParser extends TestCase {

  public TestHandlerQueryParser() {
  }

  public void testParseQuery() {
    String queryStr = "A = B AND C = D";
    String expected = "(C = D AND A = B)";

    XMLQuery query = XMLQueryHelper.getDefaultQueryFromQueryString(queryStr);
    assertNotNull(query);
    Stack<QueryElement> queryStack = HandlerQueryParser.createQueryStack(query
        .getWhereElementSet());
    assertNotNull(queryStack);
    Expression parsedQuery = HandlerQueryParser.parse(queryStack);
    assertNotNull(parsedQuery);
    assertEquals(expected, parsedQuery.evaluate());
  }
  
  public void testParseWildcardWithMapping(){
    String queryStr = "bar LIKE 'FOO'";
    String expected = "bar LIKE '%FOO%'";
    Mapping mapping = new Mapping();
    MappingField mf = new MappingField();
    mf.setDbName("bar");
    mf.setName("foo");
    mf.setString(true);
    mapping.addField("foo", mf);
    
    XMLQuery query = XMLQueryHelper.getDefaultQueryFromQueryString(queryStr);
    assertNotNull(query);
    Stack<QueryElement> queryStack = HandlerQueryParser.createQueryStack(query
        .getWhereElementSet());
    assertNotNull(queryStack);
    Expression parsedQuery = HandlerQueryParser.parse(queryStack, mapping);
    assertNotNull(parsedQuery);
    assertEquals(expected, parsedQuery.evaluate());    
  }
  
  private Mapping createFooBarMapping() {
    Mapping m = new Mapping();
    MappingField foof = new MappingField();
    foof.setName("foo");
    foof.setDbName("foo");
    foof.setString(true);
    MappingField barf = new MappingField();
    barf.setName("bar");
    barf.setDbName("bar");
    barf.setString(true);
    m.addField("foo", foof);
    m.addField("bar", barf);
    return m;
  }

  public void testParseLiteralQuotesWithParen() {
    Mapping m = createFooBarMapping();

    String q = "(foo = 'yes' OR bar = 'no') AND RETURN = foo AND RETURN = bar";
    String expected = "(bar = 'no' OR foo = 'yes')";

    XMLQuery query = XMLQueryHelper.getDefaultQueryFromQueryString(q);
    assertNotNull(query);
    Stack<QueryElement> where = HandlerQueryParser.createQueryStack(query.getWhereElementSet());
    assertNotNull(where);
    List<QueryElement> select = query.getSelectElementSet();
    assertNotNull(select);

    Expression parsed = HandlerQueryParser.parse(where, m);
    assertNotNull(parsed);
    assertEquals(expected, parsed.evaluate());
  }

}