/*
* GeoTools - The Open Source Java GIS Toolkit
* http://geotools.org
*
* (C) 2002-2008, 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.data.wfs.internal.parsers;
import static org.geotools.data.wfs.WFSTestData.*;
import static org.geotools.data.wfs.WFSTestData.CUBEWERX_ROADSEG;
import static org.geotools.data.wfs.WFSTestData.GEOS_ARCHSITES_11;
import static org.geotools.data.wfs.WFSTestData.GEOS_ROADS_11;
import static org.geotools.data.wfs.WFSTestData.GEOS_STATES_10;
import static org.geotools.data.wfs.WFSTestData.GEOS_STATES_11;
import static org.geotools.data.wfs.WFSTestData.GEOS_TASMANIA_CITIES_11;
import static org.geotools.data.wfs.WFSTestData.IONIC_STATISTICAL_UNIT;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertTrue;
import java.io.IOException;
import java.net.URL;
import javax.xml.namespace.QName;
import org.geotools.data.DataUtilities;
import org.geotools.data.wfs.internal.GetFeatureParser;
import org.geotools.referencing.CRS;
import org.geotools.wfs.v1_1.WFSConfiguration;
import org.geotools.xml.Configuration;
import org.junit.Test;
import org.opengis.feature.Feature;
import org.opengis.feature.FeatureVisitor;
import org.opengis.feature.GeometryAttribute;
import org.opengis.feature.Property;
import org.opengis.feature.simple.SimpleFeature;
import org.opengis.feature.simple.SimpleFeatureType;
import org.opengis.feature.type.AttributeDescriptor;
import org.opengis.referencing.crs.CoordinateReferenceSystem;
import com.vividsolutions.jts.geom.MultiPolygon;
import com.vividsolutions.jts.geom.Polygon;
/**
* This abstract class comprises a sort of compliance tests for {@link GetFeatureParser}
* implementations.
* <p>
* Subclasses shall just provide an implementation for
* {@link #getParser(QName, SimpleFeatureType, String)}
* </p>
* <p>
* Note this test depends on {@link EmfAppSchemaParser} to function correctly in order to obtain the
* test FeatureTypes from the DescribeFeatureType response samples under {@code test-data/}.
* </p>
*
* @author Gabriel Roldan
* @version $Id$
* @since 2.5.x
*
*
*
* @source $URL$ java/org/geotools/wfs/v_1_1_0/data/StreamingParserFeatureReaderTest .java $
* @see XmlSimpleFeatureParserTest
* @see StreamingParserFeatureReaderTest
*/
@SuppressWarnings("nls")
public abstract class AbstractGetFeatureParserTest {
private boolean supportsCount = true;
protected void setSupportsCount(boolean supportsCount) {
this.supportsCount = supportsCount;
}
/**
* Configuration object used to parse the sample schemas
*
* @see #getTypeView(QName, String, String, String[])
*/
private static final Configuration wfsConfiguration = new WFSConfiguration();
/**
* A feature visitor used to assert the parsed features
*
* @author Gabriel Roldan (TOPP)
* @version $Id$
* @since 2.5.x
* @source $URL: http://svn.geotools.org/geotools/trunk/gt/modules/plugin/wfs /src/test
* /java/org/geotools/wfs/v_1_1_0/data/AbstractGetFeatureParserTest .java $
*/
private static class FeatureAssertor implements FeatureVisitor {
private SimpleFeatureType featureType;
/**
* A featuretype which might be a subset of the actual FeatureType whose attributes will be
* used to assert the features.
*
* @param featureType
*/
public FeatureAssertor(SimpleFeatureType featureType) {
this.featureType = featureType;
}
public void visit(final Feature feature) {
assertNotNull(feature);
assertNotNull(feature.getIdentifier().getID());
for (AttributeDescriptor descriptor : featureType.getAttributeDescriptors()) {
final String name = descriptor.getLocalName();
Property property = feature.getProperty(name);
assertNotNull(name + " property was not parsed", property);
assertNotNull("got null value for property " + name, property.getValue());
}
}
}
/**
* Parses the featuretype from the test file referenced by {@code schemaName} and returns a new,
* subset FeatureType comprised of only the required {@code properties}
*
* @param featureName
* the name of the Features produced for the target FeatureType (i.e.
* {@code topp:states} instead of {@code topp:states_Type})
* @param schemaLocation2
* the location of the schema file under
* {@code org/geotools/data/wfs/impl/test-data/}
* @param epsgCrsId
* the EPSG identifier for the feature type CRS (eg. {@code "EPSG:4326"})
* @param properties
* the property names to include from the original schema in the one to be returned
* @return a subset of the original featuretype containing only the required {@code properties}
*/
private SimpleFeatureType getTypeView(final QName featureName, final URL schemaLocation,
final String epsgCrsId, final String[] properties) throws Exception {
CoordinateReferenceSystem crs = CRS.decode(epsgCrsId);
SimpleFeatureType originalType = EmfAppSchemaParser.parseSimpleFeatureType(
wfsConfiguration, featureName, schemaLocation, crs);
SimpleFeatureType subsetType = DataUtilities.createSubType(originalType, properties);
return subsetType;
}
/**
* Uses a {@link PullParserFeatureReader} to parse the features while traversing the
* feature collection in a test {@code wfs:FeatureCollection} document; {@code assertor} is a
* visitor provided by the actual unit test calling this method, every feature fetched is passed
* to the visitor who contains the specific assertions.
*
* @param featureName
* the name of the features (not the feature type) expected
* @param getFeatureResultTestFile
* the name of the test file name to load in order to simulate the response of a
* GetFeature request
* @param assertor
* a FeatureVisitor to assert the contents or structure of the features
* @param expectedFeatureCount
* the number of features there should be on the feature collection, an assertion is
* made at the end of the method.
* @param schemaName
* @throws Exception
*/
private void testParseGetFeatures(final QName featureName,
final SimpleFeatureType queryFeatureType, final GetFeatureParser parser,
final FeatureVisitor assertor, final int expectedFeatureCount) throws Exception {
int featureCount = 0;
SimpleFeature feature;
try {
for (int i = 0; i < expectedFeatureCount; i++) {
feature = parser.parse();
assertNotNull(feature);
featureCount++;
assertor.visit(feature);
}
feature = parser.parse();
assertNull(feature);
} finally {
parser.close();
}
assertEquals(expectedFeatureCount, featureCount);
}
/**
* Subclasses need to implement in order to provide a specific {@link GetFeatureParser}
* implementation settled up for the given featureName and dataFile containing the test
* GetFeature request response.
*
* @param featureName
* @param schemaLocation
* @param featureType
* @param getFeaturesRequest
* the URL representing the GetFeature request. Opening its input stream shall
* suffice to get the GetFeature response.
* @return
* @throws IOException
*/
protected abstract GetFeatureParser getParser(QName featureName, URL schemaLocation,
SimpleFeatureType featureType, URL getFeaturesRequest) throws IOException;
/**
* Verifies correctness on parsing a normal geoserver WFS 1.1.0 GetFeature response.
*
* Test method for {@link PullParserFeatureReader#parse()}.
*
* @throws Exception
*/
@Test
public void testParseGeoServer_ArchSites_Point() throws Exception {
final QName featureName = GEOS_ARCHSITES_11.TYPENAME;
final int expectedCount = 3;
final URL schemaLocation = GEOS_ARCHSITES_11.SCHEMA;
final URL data = GEOS_ARCHSITES_11.DATA;
final String[] properties = { "cat", "str1", "the_geom" };
final SimpleFeatureType featureType;
featureType = getTypeView(featureName, schemaLocation, GEOS_ARCHSITES_11.CRS, properties);
final FeatureVisitor assertor = new FeatureAssertor(featureType);
GetFeatureParser parser = getParser(featureName, schemaLocation, featureType, data);
if (supportsCount) {
int nof = parser.getNumberOfFeatures();
assertEquals(expectedCount, nof);
}
testParseGetFeatures(featureName, featureType, parser, assertor, expectedCount);
}
@Test
public void testParseGeoServer_Poi_100() throws Exception {
final QName featureName = GEOS_POI_10.TYPENAME;
final int expectedCount = 3;
final URL schemaLocation = GEOS_POI_10.SCHEMA;
final URL data = GEOS_POI_10.DATA;
final String[] properties = { "the_geom", "NAME", "THUMBNAIL", "MAINPAGE" };
final SimpleFeatureType featureType;
featureType = getTypeView(featureName, schemaLocation, GEOS_POI_10.CRS, properties);
final FeatureVisitor assertor = new FeatureAssertor(featureType);
GetFeatureParser parser = getParser(featureName, schemaLocation, featureType, data);
int nof = parser.getNumberOfFeatures();
assertEquals(-1, nof);
testParseGetFeatures(featureName, featureType, parser, assertor, expectedCount);
}
/**
* Verifies correctness on parsing a normal geoserver WFS 1.1.0 GetFeature response for the
* usual topp:states feature type (multipolygon).
*
* Test method for {@link PullParserFeatureReader#parse()}.
*
* @throws Exception
*/
@Test
public void testParseGeoServer_States_polygon_with_hole() throws Exception {
final QName featureName = GEOS_STATES_11.TYPENAME;
final int expectedCount = 2;
final URL schemaLocation = GEOS_STATES_11.SCHEMA;
final String[] properties = { "the_geom", "STATE_NAME", "STATE_FIPS", "SUB_REGION",
"SAMP_POP" };
final SimpleFeatureType featureType;
featureType = getTypeView(featureName, schemaLocation, GEOS_STATES_11.CRS, properties);
final FeatureVisitor assertor = new FeatureAssertor(featureType) {
@Override
public void visit(final Feature feature) {
super.visit(feature);
final String fid = feature.getIdentifier().getID();
final int numPolygons;
final int expectedHoles;
if ("states.1".equals(fid)) {
numPolygons = 2;
expectedHoles = 1;
} else if ("states.2".equals(fid)) {
numPolygons = 1;
expectedHoles = 2;
} else {
throw new IllegalArgumentException("Expected states.1 or states.2, got " + fid);
}
GeometryAttribute defaultGeometryProperty = feature.getDefaultGeometryProperty();
assertNotNull(defaultGeometryProperty);
final Object value = defaultGeometryProperty.getValue();
assertNotNull(value);
assertTrue("value: " + value, value instanceof MultiPolygon);
MultiPolygon mp = (MultiPolygon) value;
assertEquals(numPolygons, mp.getNumGeometries());
for (int i = 0; i < numPolygons; i++) {
Polygon p = (Polygon) mp.getGeometryN(i);
assertEquals(expectedHoles, p.getNumInteriorRing());
}
}
};
GetFeatureParser parser = getParser(featureName, schemaLocation, featureType,
GEOS_STATES_11.DATA);
if (supportsCount) {
int nof = parser.getNumberOfFeatures();
assertEquals(expectedCount, nof);
}
testParseGetFeatures(featureName, featureType, parser, assertor, expectedCount);
}
@Test
public void testParseGeoServer_States_100() throws Exception {
final QName featureName = GEOS_STATES_10.TYPENAME;
final int expectedCount = 3;
final URL schemaLocation = GEOS_STATES_10.SCHEMA;
final String[] properties = { "the_geom", "STATE_NAME", "STATE_FIPS", "SUB_REGION",
"SAMP_POP" };
final SimpleFeatureType featureType;
featureType = getTypeView(featureName, schemaLocation, GEOS_STATES_11.CRS, properties);
final FeatureVisitor assertor = new FeatureAssertor(featureType) {
@Override
public void visit(final Feature feature) {
super.visit(feature);
final String fid = feature.getIdentifier().getID();
final int numPolygons;
final int expectedHoles;
if ("states.1".equals(fid)) {
numPolygons = 2;
expectedHoles = 1;
} else if ("states.2".equals(fid)) {
numPolygons = 1;
expectedHoles = 2;
} else if ("states.3".equals(fid)) {
numPolygons = 1;
expectedHoles = 0;
} else {
throw new IllegalArgumentException("Expected states.1 or states.2, got " + fid);
}
GeometryAttribute defaultGeometryProperty = feature.getDefaultGeometryProperty();
assertNotNull(defaultGeometryProperty);
final Object value = defaultGeometryProperty.getValue();
assertNotNull(value);
assertTrue("value: " + value, value instanceof MultiPolygon);
MultiPolygon mp = (MultiPolygon) value;
assertEquals(numPolygons, mp.getNumGeometries());
for (int i = 0; i < numPolygons; i++) {
Polygon p = (Polygon) mp.getGeometryN(i);
assertEquals(expectedHoles, p.getNumInteriorRing());
}
}
};
GetFeatureParser parser = getParser(featureName, schemaLocation, featureType,
GEOS_STATES_10.DATA);
int nof = parser.getNumberOfFeatures();
assertEquals(-1, nof);
testParseGetFeatures(featureName, featureType, parser, assertor, expectedCount);
}
@Test
public void testParseGeoServer_roads_MultiLineString_110() throws Exception {
final QName featureName = GEOS_ROADS_11.TYPENAME;
final int expectedCount = 3;
final URL schemaLocation = GEOS_ROADS_11.SCHEMA;
final String[] properties = { "the_geom", "label" };
final SimpleFeatureType featureType;
featureType = getTypeView(featureName, schemaLocation, GEOS_ROADS_11.CRS, properties);
final FeatureVisitor assertor = new FeatureAssertor(featureType);
GetFeatureParser parser = getParser(featureName, schemaLocation, featureType,
GEOS_ROADS_11.DATA);
if (supportsCount) {
int nof = parser.getNumberOfFeatures();
assertEquals(expectedCount, nof);
}
testParseGetFeatures(featureName, featureType, parser, assertor, expectedCount);
}
@Test
public void testParseGeoServer_roads_MultiLineString_100() throws Exception {
final QName featureName = GEOS_ROADS_10.TYPENAME;
final int expectedCount = 2;
final URL schemaLocation = GEOS_ROADS_10.SCHEMA;
final String[] properties = { "the_geom", "label" };
final SimpleFeatureType featureType;
featureType = getTypeView(featureName, schemaLocation, GEOS_ROADS_10.CRS, properties);
final FeatureVisitor assertor = new FeatureAssertor(featureType);
GetFeatureParser parser = getParser(featureName, schemaLocation, featureType,
GEOS_ROADS_10.DATA);
int nof = parser.getNumberOfFeatures();
assertEquals(-1, nof);
testParseGetFeatures(featureName, featureType, parser, assertor, expectedCount);
}
@Test
public void testParseGeoServer_tasmania_cities_MultiPoint() throws Exception {
final QName featureName = GEOS_TASMANIA_CITIES_11.TYPENAME;
final int expectedCount = 1;
final URL schemaLocation = GEOS_TASMANIA_CITIES_11.SCHEMA;
final String[] properties = { "the_geom", "CNTRY_NAME", "POP_CLASS" };
final SimpleFeatureType featureType;
featureType = getTypeView(featureName, schemaLocation, GEOS_TASMANIA_CITIES_11.CRS,
properties);
final FeatureVisitor assertor = new FeatureAssertor(featureType);
GetFeatureParser parser = getParser(featureName, schemaLocation, featureType,
GEOS_TASMANIA_CITIES_11.DATA);
if (supportsCount) {
int nof = parser.getNumberOfFeatures();
assertEquals(expectedCount, nof);
}
testParseGetFeatures(featureName, featureType, parser, assertor, expectedCount);
}
/**
* Verifies correctness on parsing a sample CubeWerx WFS 1.1.0 GetFeature response.
*
* @throws Exception
*/
@Test
public void testParseCubeWerx_GovernmentalUnitCE() throws Exception {
QName featureName = CUBEWERX_GOVUNITCE.TYPENAME;
URL schemaLocation = CUBEWERX_GOVUNITCE.SCHEMA;
String srsName = CUBEWERX_GOVUNITCE.CRS;
final int expectedCount = 3;
final String[] properties = { "geometry", "instanceName", "instanceCode" };
final SimpleFeatureType featureType = getTypeView(featureName, schemaLocation, srsName,
properties);
final FeatureVisitor assertor = new FeatureAssertor(featureType);
GetFeatureParser parser = getParser(featureName, schemaLocation, featureType,
CUBEWERX_GOVUNITCE.DATA);
int nof = parser.getNumberOfFeatures();
assertEquals(-1, nof);
testParseGetFeatures(featureName, featureType, parser, assertor, expectedCount);
}
@Test
public void testParseCubeWerx_RoadSeg() throws Exception {
final String[] properties = { "lastUpdateDate", "geometry", "status", "isAnchorSection" };
final QName featureName = CUBEWERX_ROADSEG.TYPENAME;
final URL schemaLocation = CUBEWERX_ROADSEG.SCHEMA;
final SimpleFeatureType featureType = getTypeView(featureName, schemaLocation,
CUBEWERX_ROADSEG.CRS, properties);
final GetFeatureParser parser = getParser(featureName, schemaLocation, featureType,
CUBEWERX_ROADSEG.DATA);
int nof = parser.getNumberOfFeatures();
assertEquals(-1, nof);
FeatureVisitor assertor = new FeatureAssertor(featureType);
testParseGetFeatures(featureName, featureType, parser, assertor, 3);
}
@Test
public void testParseIonic_StatisticalUnit() throws Exception {
final String[] properties = { "unitId", "typeAbbreviation", "instanceName", "geometry" };
final QName featureName = IONIC_STATISTICAL_UNIT.TYPENAME;
final URL schemaLocation = IONIC_STATISTICAL_UNIT.SCHEMA;
final SimpleFeatureType featureType = getTypeView(featureName, schemaLocation,
CUBEWERX_ROADSEG.CRS, properties);
final GetFeatureParser parser = getParser(featureName, schemaLocation, featureType,
IONIC_STATISTICAL_UNIT.DATA);
int nof = parser.getNumberOfFeatures();
assertEquals(-1, nof);
FeatureVisitor assertor = new FeatureAssertor(featureType);
testParseGetFeatures(featureName, featureType, parser, assertor, 2);
}
}