/* (c) 2014 Open Source Geospatial Foundation - all rights reserved
* (c) 2001 - 2013 OpenPlans
* This code is licensed under the GPL 2.0 license, available at the root
* application directory.
*/
package org.geoserver.test;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.InputStream;
import java.io.Serializable;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import org.geoserver.data.CatalogWriter;
import org.geoserver.data.test.MockData;
import org.geoserver.data.test.SystemTestData;
import org.geoserver.data.test.TestData;
import org.geoserver.data.util.IOUtils;
import org.geotools.data.SampleDataAccess;
import org.geotools.data.SampleDataAccessData;
import org.geotools.data.SampleDataAccessFactory;
import com.vividsolutions.jts.geom.Envelope;
/**
* Mock data for testing integration of {@link SampleDataAccess} with GeoServer.
*
* Inspired by {@link MockData}.
*/
public class SampleDataAccessMockData extends SystemTestData {
public static final String DATASTORE_NAME = "datastore";
/**
* Use FeatureTypeInfo constants for srs handling as values
*/
public static final String KEY_SRS_HANDLINGS = "srsHandling";
/**
* The feature type alias, a string
*/
public static final String KEY_ALIAS = "alias";
/**
* The style name
*/
public static final String KEY_STYLE = "style";
/**
* The srs code (a number) for this layer
*/
public static final String KEY_SRS_NUMBER = "srs";
/**
* The lon/lat envelope as a JTS Envelope
*/
public static final String KEY_LL_ENVELOPE = "ll_envelope";
/**
* The native envelope as a JTS Envelope
*/
public static final String KEY_NATIVE_ENVELOPE = "native_envelope";
static final Envelope DEFAULT_ENVELOPE = new Envelope(-180, 180, -90, 90);
/** the 'featureTypes' directory, under 'data' */
File featureTypes;
/**
* Constructor. Creates empty mock data directory.
*
* @throws IOException
*/
public SampleDataAccessMockData() throws IOException {
data = IOUtils.createRandomDirectory("./target", "sample-data-access-mock", "data");
data.delete();
data.mkdir();
// create a featureTypes directory
featureTypes = new File(data, "featureTypes");
featureTypes.mkdir();
info(DATASTORE_NAME, SampleDataAccessData.NAMESPACE_PREFIX,
SampleDataAccessData.MAPPEDFEATURE_TYPE_NAME.getLocalPart());
// need to add nested type at top level so type definition is loaded into global schema and
// can be found during encoding
info(DATASTORE_NAME, SampleDataAccessData.NAMESPACE_PREFIX,
SampleDataAccessData.GEOLOGICUNIT_TYPE_NAME.getLocalPart());
}
/**
* Returns the root of the mock data directory,
*
* @see org.geoserver.data.test.TestData#getDataDirectoryRoot()
*/
public File getDataDirectoryRoot() {
return data;
}
/**
* Returns true.
*
* @see org.geoserver.data.test.TestData#isTestDataAvailable()
*/
public boolean isTestDataAvailable() {
return true;
}
/**
* Configures mock data directory.
*
* @see org.geoserver.data.test.TestData#setUp()
*/
public void setUp() throws Exception {
setUpCatalog();
copyTo(MockData.class.getResourceAsStream("services.xml"), "services.xml");
}
@Override
public void setUpDefault() throws Exception {
//do nothing
}
/**
* Removes the mock data directory.
*
* @see org.geoserver.data.test.TestData#tearDown()
*/
public void tearDown() throws Exception {
IOUtils.delete(data);
data = null;
}
/**
* Writes catalog.xml to the data directory.
*
* @throws IOException
*/
@SuppressWarnings("serial")
protected void setUpCatalog() throws IOException {
CatalogWriter writer = new CatalogWriter();
writer.dataStores(new HashMap<String, Map<String, Serializable>>() {
{
put(DATASTORE_NAME, SampleDataAccessFactory.PARAMS);
}
}, new HashMap<String, String>() {
{
put(DATASTORE_NAME, SampleDataAccessData.NAMESPACE_PREFIX);
}
}, Collections.<String> emptySet());
writer.coverageStores(new HashMap<String, Map<String, String>>(),
new HashMap<String, String>(), Collections.<String> emptySet());
writer.namespaces(new HashMap<String, String>() {
{
put(SampleDataAccessData.NAMESPACE_PREFIX, SampleDataAccessData.NAMESPACE_URI);
}
});
writer.styles(Collections.<String, String> emptyMap());
writer.write(new File(data, "catalog.xml"));
}
/**
* Copies from an {@link InputStream} to path under the mock data directory.
*
* @param input
* source from which file content is copied
* @param location
* path relative to mock data directory
*/
public void copyTo(InputStream input, String location) throws IOException {
IOUtils.copy(input, new File(getDataDirectoryRoot(), location));
}
/**
* Stolen from {@link MockData}.
*
* @param name
* @param extraParams
* @throws IOException
*/
public void info(String datastore, String prefix, String type) throws IOException {
// prepare extra params default
Map<String, Object> params = new HashMap<String, Object>();
params.put(KEY_STYLE, "Default");
params.put(KEY_SRS_HANDLINGS, 2);
params.put(KEY_ALIAS, null);
Integer srs = 4326;
params.put(KEY_SRS_NUMBER, srs);
File featureTypeDir = new File(featureTypes, prefix + "_" + type);
featureTypeDir.mkdir();
File info = new File(featureTypeDir, "info.xml");
info.delete();
info.createNewFile();
FileWriter writer = new FileWriter(info);
writer.write("<featureType datastore=\"" + datastore + "\">");
writer.write("<name>" + type + "</name>");
if (params.get(KEY_ALIAS) != null)
writer.write("<alias>" + params.get(KEY_ALIAS) + "</alias>");
writer.write("<SRS>" + params.get(KEY_SRS_NUMBER) + "</SRS>");
// this mock type may have wrong SRS compared to the actual one in the property files...
// let's configure SRS handling not to alter the original one, and have 4326 used only
// for capabilities
writer.write("<SRSHandling>" + params.get(KEY_SRS_HANDLINGS) + "</SRSHandling>");
writer.write("<title>" + type + "</title>");
writer.write("<abstract>abstract about " + type + "</abstract>");
writer.write("<numDecimals value=\"8\"/>");
writer.write("<keywords>" + type + "</keywords>");
Envelope llEnvelope = (Envelope) params.get(KEY_LL_ENVELOPE);
if (llEnvelope == null)
llEnvelope = DEFAULT_ENVELOPE;
writer.write("<latLonBoundingBox dynamic=\"false\" minx=\"" + llEnvelope.getMinX()
+ "\" miny=\"" + llEnvelope.getMinY() + "\" maxx=\"" + llEnvelope.getMaxX()
+ "\" maxy=\"" + llEnvelope.getMaxY() + "\"/>");
Envelope nativeEnvelope = (Envelope) params.get(KEY_NATIVE_ENVELOPE);
if (nativeEnvelope != null)
writer.write("<nativeBBox dynamic=\"false\" minx=\"" + nativeEnvelope.getMinX()
+ "\" miny=\"" + nativeEnvelope.getMinY() + "\" maxx=\""
+ nativeEnvelope.getMaxX() + "\" maxy=\"" + nativeEnvelope.getMaxY() + "\"/>");
String style = (String) params.get(KEY_STYLE);
if (style == null)
style = "Default";
writer.write("<styles default=\"" + style + "\"/>");
writer.write("</featureType>");
writer.flush();
writer.close();
}
}