/*
* GeoTools - The Open Source Java GIS Toolkit
* http://geotools.org
*
* (C) 2005-2012, Open Source Geospatial Foundation (OSGeo)
*
* This library 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;
* version 2.1 of the License.
*
* This library 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.
*/
package org.geotools.filter.visitor;
import static org.junit.Assert.assertEquals;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashSet;
import java.util.List;
import org.geotools.filter.Capabilities;
import org.junit.Before;
import org.junit.Test;
import org.opengis.filter.Filter;
import org.opengis.filter.Id;
import org.opengis.filter.Or;
import org.opengis.filter.PropertyIsBetween;
import org.opengis.filter.PropertyIsLike;
import org.opengis.filter.PropertyIsNull;
import org.opengis.filter.spatial.BBOX;
/**
*
* @author Jesse
* @author ported from PostPreProcessFilterSplittingVisitor at 2.5.2 by Gabriel Roldan
*
*
* @source $URL$
*/
@SuppressWarnings( { "nls", "unchecked" })
public class CapabilitiesFilterSplitterTest extends AbstractCapabilitiesFilterSplitterTests {
private Capabilities simpleLogicalCaps = new Capabilities();
private CapabilitiesFilterSplitter visitor;
@Before
public void setUp() throws Exception {
simpleLogicalCaps.addAll(Capabilities.SIMPLE_COMPARISONS_OPENGIS);
simpleLogicalCaps.addAll(Capabilities.LOGICAL_OPENGIS);
}
@Test
public void testVisitBetweenFilter() throws Exception {
PropertyIsBetween filter = ff.between(ff.literal(0), ff.property(numAtt), ff.literal(4));
runTest(filter, newCapabilities(PropertyIsBetween.class), numAtt);
}
@Test
public void testNullTransactionAccessor() throws Exception {
accessor = null;
Filter f1 = createPropertyIsEqualToFilter(nameAtt, "david");
Filter f2 = createPropertyIsEqualToFilter(nameAtt, "david");
runTest(ff.and(f1, f2), simpleLogicalCaps, nameAtt);
}
@Test
public void testVisitLogicalANDFilter() throws Exception {
Filter f1 = createPropertyIsEqualToFilter(nameAtt, "david");
Filter f2 = createPropertyIsEqualToFilter(nameAtt, "david");
runTest(ff.and(f1, f2), simpleLogicalCaps, nameAtt);
}
@Test
public void testVisitLogicalNOTFilter() throws Exception {
Filter f1 = createPropertyIsEqualToFilter(nameAtt, "david");
runTest(ff.not(f1), simpleLogicalCaps, nameAtt);
}
@Test
public void testVisitLogicalORFilter() throws Exception {
Filter f1 = createPropertyIsEqualToFilter(nameAtt, "david");
Filter f2 = createPropertyIsEqualToFilter("name", "jose");
Filter orFilter = ff.or(f1, f2);
runTest(orFilter, simpleLogicalCaps, nameAtt);
visitor = newVisitor(simpleLogicalCaps);
f2 = ff.bbox(geomAtt, 10.0, 20.0, 10.0, 20.0, "");
orFilter = ff.or(f1, f2);
orFilter.accept(visitor, null);
// f1 could be pre-processed but since f2 can't all the processing has to be done on the
// client side :-(
assertEquals(Filter.INCLUDE, visitor.getFilterPre());
assertEquals(orFilter, visitor.getFilterPost());
}
@Test
public void testMassOrFilter() throws Exception {
List filters = new ArrayList();
for (int i = 0; i < 10000; i++) {
filters.add(ff.equals(ff.property(nameAtt), ff.literal("" + i)));
}
Or orFilter = ff.or(filters);
visitor = newVisitor(simpleLogicalCaps);
// this would have throw the thing into a stack overflow error
orFilter.accept(visitor, null);
assertEquals(orFilter, visitor.getFilterPre());
assertEquals(Filter.INCLUDE, visitor.getFilterPost());
}
@Test
public void testVisitCompareFilter() throws Exception {
Filter f = createPropertyIsEqualToFilter(nameAtt, "david");
runTest(f, Capabilities.SIMPLE_COMPARISONS_OPENGIS, nameAtt);
}
/**
* an update is in transaction that modifies an attribute that NOT is referenced in the query
*/
@Test
public void testVisitCompareFilterWithUpdateDifferentAttribute() throws Exception {
Filter f = createPropertyIsEqualToFilter(nameAtt, "david");
Filter updateFilter = createPropertyIsEqualToFilter(nameAtt, "jose");
accessor = new TestAccessor();
accessor.setUpdate(geomAtt, updateFilter);
visitor = newVisitor(simpleLogicalCaps);
f.accept(visitor, null);
assertEquals(visitor.getFilterPost().toString(), Filter.INCLUDE, visitor.getFilterPost());
assertEquals(visitor.getFilterPre().toString(), f, visitor.getFilterPre());
}
@Test
public void testVisitLikeFilter() throws Exception {
Filter filter = ff.like(ff.property(nameAtt), "j*", "*", "?", "\\");
runTest(filter, newCapabilities(PropertyIsLike.class), nameAtt);
}
@Test
public void testVisitNullFilter() throws Exception {
Filter filter = ff.isNull(ff.property(nameAtt));
runTest(filter, newCapabilities(PropertyIsNull.class), nameAtt);
}
@Test
public void testVisitFidFilter() throws Exception {
HashSet ids = new HashSet();
ids.add(ff.featureId("david"));
Filter filter = ff.id(ids);
visitor = newVisitor(newCapabilities(Id.class));
filter.accept(visitor, null);
assertEquals(Filter.INCLUDE, visitor.getFilterPost());
assertEquals(filter, visitor.getFilterPre());
}
@Test
@SuppressWarnings("rawtypes")
public void testVisitIdFilterWithNoIdCapabilities() throws Exception {
// Id Filter
HashSet ids = new HashSet();
ids.add(ff.featureId("david"));
Filter idFilter = ff.id(ids);
// no Id Capabilities
visitor = newVisitor(Capabilities.SIMPLE_COMPARISONS_OPENGIS);
idFilter.accept(visitor, null);
assertEquals(Filter.INCLUDE, visitor.getFilterPre());
assertEquals(idFilter, visitor.getFilterPost());
}
@Test
public void testFunctionFilter() throws Exception {
simpleLogicalCaps.addType(BBOX.class);
visitor = newVisitor(simpleLogicalCaps);
Filter filter = createFunctionFilter();
filter.accept(visitor, null);
assertEquals(filter, visitor.getFilterPost());
assertEquals(Filter.INCLUDE, visitor.getFilterPre());
simpleLogicalCaps.addName(testFunction.getName(), testFunction.getParameters().size());
visitor = newVisitor(simpleLogicalCaps);
filter.accept(visitor, null);
assertEquals(Filter.INCLUDE, visitor.getFilterPost());
assertEquals(filter, visitor.getFilterPre());
}
@Test
public void testFunctionANDGeometryFilter() throws Exception {
simpleLogicalCaps.addType(BBOX.class);
visitor = newVisitor(simpleLogicalCaps);
Filter funtionFilter = createFunctionFilter();
Filter geomFilter = ff.bbox(geomAtt, 10, 20, 10, 20, "");
Filter andFilter = ff.and(funtionFilter, geomFilter);
andFilter.accept(visitor, null);
assertEquals(funtionFilter.toString(), visitor.getFilterPost().toString());
assertEquals(geomFilter.toString(), visitor.getFilterPre().toString());
simpleLogicalCaps.addName(testFunction.getName(), testFunction.getParameters().size());
visitor = newVisitor(simpleLogicalCaps);
andFilter.accept(visitor, null);
assertEquals(Filter.INCLUDE, visitor.getFilterPost());
assertEquals(andFilter, visitor.getFilterPre());
}
@Test
public void testFunctionORGeometryFilter() throws Exception {
simpleLogicalCaps.addType(BBOX.class);
visitor = newVisitor(simpleLogicalCaps);
Filter funtionFilter = createFunctionFilter();
Filter geomFilter = ff.bbox(geomAtt, 10, 20, 10, 20, "");
Filter orFilter = ff.or(funtionFilter, geomFilter);
orFilter.accept(visitor, null);
assertEquals(Filter.INCLUDE, visitor.getFilterPre());
assertEquals(orFilter, visitor.getFilterPost());
simpleLogicalCaps.addName(testFunction.getName(), testFunction.getParameters().size());
visitor = newVisitor(simpleLogicalCaps);
orFilter.accept(visitor, null);
assertEquals(Filter.INCLUDE, visitor.getFilterPost());
assertEquals(orFilter, visitor.getFilterPre());
}
@Test
public void testFunctionNOTFilter() throws Exception {
simpleLogicalCaps.addType(BBOX.class);
visitor = newVisitor(simpleLogicalCaps);
Filter funtionFilter = createFunctionFilter();
Filter not = ff.not(funtionFilter);
not.accept(visitor, null);
assertEquals(not, visitor.getFilterPost());
assertEquals(Filter.INCLUDE, visitor.getFilterPre());
simpleLogicalCaps.addName(testFunction.getName(), testFunction.getParameters().size());
visitor = newVisitor(simpleLogicalCaps);
not.accept(visitor, null);
assertEquals(Filter.INCLUDE, visitor.getFilterPost());
assertEquals(not, visitor.getFilterPre());
}
@Test
public void testNullParentNullAccessor() throws Exception {
simpleLogicalCaps.addType(BBOX.class);
simpleLogicalCaps.addName(testFunction.getName(), testFunction.getParameters().size());
visitor = newVisitor(simpleLogicalCaps);
Filter funtionFilter = createFunctionFilter();
Filter geomFilter = ff.bbox(geomAtt, 10.0, 20.0, 10.0, 20.0, "");
Filter orFilter = ff.or(funtionFilter, geomFilter);
visitor = new CapabilitiesFilterSplitter(new Capabilities(), null, null);
orFilter.accept(visitor, null);
assertEquals(Filter.INCLUDE, visitor.getFilterPre());
assertEquals(orFilter, visitor.getFilterPost());
visitor = new CapabilitiesFilterSplitter(simpleLogicalCaps, null, null);
orFilter.accept(visitor, null);
assertEquals(Filter.INCLUDE, visitor.getFilterPost());
assertEquals(orFilter, visitor.getFilterPre());
}
@Test
public void testComplicatedOrFilter() throws Exception {
Filter c1 = ff.equals(ff.property("eventstatus"), ff.literal("deleted"));
Filter c2 = ff.equals(ff.property("eventtype"), ff.literal("road hazard"));
Filter c3 = ff.equals(ff.property("eventtype"), ff.literal("area warning"));
Filter g = ff.bbox("geom", 0, 180, 0, 90, "");
Filter f = ff.or(c2, c3);
f = ff.and(ff.not(c1), f);
f = ff.and(f, g);
simpleLogicalCaps.addType(BBOX.class);
simpleLogicalCaps.addName(testFunction.getName(), testFunction.getParameters().size());
visitor = new CapabilitiesFilterSplitter(simpleLogicalCaps, null, null);
f.accept(visitor, null);
assertEquals(f, visitor.getFilterPre());
assertEquals(Filter.INCLUDE, visitor.getFilterPost());
visitor = new CapabilitiesFilterSplitter(simpleLogicalCaps, null,
new ClientTransactionAccessor() {
public Filter getDeleteFilter() {
return null;
}
public Filter getUpdateFilter(String attributePath) {
if (attributePath.equals("eventtype")) {
HashSet ids = new HashSet();
ids.add(ff.featureId("fid"));
return ff.id(ids);
}
return null;
}
});
f.accept(visitor, null);
HashSet ids = new HashSet();
ids.add(ff.featureId("fid"));
assertEquals(f, visitor.getFilterPost());
assertEquals(ff.or(f, ff.id(ids)), visitor.getFilterPre());
}
@Test
public void testTemporalFilter() {
visitor = new CapabilitiesFilterSplitter(simpleLogicalCaps, null, null);
Filter f1 = ff.after(ff.property("foo"), ff.literal("2011-06-20"));
f1.accept(visitor, null);
assertEquals(Filter.INCLUDE, visitor.getFilterPre());
assertEquals(f1, visitor.getFilterPost());
Filter f2 = ff.equal(ff.property("bar"), ff.literal("hello"), true);
Filter f3 = ff.and(f1, f2);
f3.accept(visitor, null);
assertEquals(f2, visitor.getFilterPre());
assertEquals(f1, visitor.getFilterPost());
}
@Test
public void testAndOptimization() throws Exception {
Capabilities caps = new Capabilities();
// no logical operator capabilities
caps.addAll(Capabilities.SIMPLE_COMPARISONS_OPENGIS);
visitor = newVisitor(caps);
// try with two filters
Filter f1 = ff.greater(ff.property("foo"), ff.literal(42));
Filter f2 = ff.less(ff.property("bar"), ff.literal(21));
Filter andFilter = ff.and(f1, f2);
// visit filter
andFilter.accept(visitor, null);
// test
assertEquals(f1, visitor.getFilterPre());
assertEquals(f2, visitor.getFilterPost());
// try with a third filter
Filter f3 = ff.greater(ff.property("height"), ff.literal(84));
andFilter = ff.and(Arrays.asList(new Filter[] { f1, f2, f3 }));
// visit filter
andFilter.accept(visitor, null);
// test
assertEquals(f1, visitor.getFilterPre());
assertEquals(ff.and(f2, f3), visitor.getFilterPost());
}
}