262 lines
8.1 KiB
Java
262 lines
8.1 KiB
Java
/*
|
|
* GeoServer-Manager - Simple Manager Library for GeoServer
|
|
*
|
|
* Copyright (C) 2007,2011 GeoSolutions S.A.S.
|
|
* http://www.geo-solutions.it
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
* in the Software without restriction, including without limitation the rights
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
* furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in
|
|
* all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
* THE SOFTWARE.
|
|
*/
|
|
|
|
package it.geosolutions.geoserver.rest.publisher;
|
|
|
|
import it.geosolutions.geoserver.rest.GeoserverRESTTest;
|
|
import it.geosolutions.geoserver.rest.decoder.RESTLayer;
|
|
import it.geosolutions.geoserver.rest.decoder.utils.JDOMBuilder;
|
|
import it.geosolutions.geoserver.rest.encoder.GSLayerEncoder;
|
|
|
|
import java.io.File;
|
|
import java.io.FileInputStream;
|
|
import java.io.FileNotFoundException;
|
|
import java.io.IOException;
|
|
|
|
import org.apache.commons.io.IOUtils;
|
|
import org.jdom.Element;
|
|
import org.jdom.Namespace;
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
import org.springframework.core.io.ClassPathResource;
|
|
|
|
/**
|
|
* Testcase for publishing layers on geoserver. We need a running GeoServer to
|
|
* properly run the tests. If such geoserver instance cannot be contacted, tests
|
|
* will be skipped.
|
|
*
|
|
* @author etj
|
|
* @author Carlo Cancellieri - carlo.cancellieri@geo-solutions.it
|
|
*/
|
|
public class GeoserverRESTStyleTest extends GeoserverRESTTest {
|
|
|
|
private final static Logger LOGGER = LoggerFactory
|
|
.getLogger(GeoserverRESTStyleTest.class);
|
|
|
|
public GeoserverRESTStyleTest(String testName) {
|
|
super(testName);
|
|
}
|
|
|
|
public void testStyles() throws IOException {
|
|
if (!enabled())
|
|
return;
|
|
deleteAll();
|
|
|
|
assertEquals(0, reader.getStyles().size());
|
|
|
|
final String styleName = "restteststyle";
|
|
File sldFile = new ClassPathResource("testdata/restteststyle.sld")
|
|
.getFile();
|
|
|
|
// insert style
|
|
assertTrue(publisher.publishStyle(sldFile));
|
|
assertTrue(reader.existsStyle(styleName));
|
|
|
|
assertFalse(publisher.publishStyle(sldFile));
|
|
assertTrue(reader.existsStyle(styleName));
|
|
|
|
String sld = reader.getSLD(styleName);
|
|
assertNotNull(sld);
|
|
|
|
Element styleEl = JDOMBuilder.buildElement(sld);
|
|
assertNotNull(styleEl);
|
|
|
|
Namespace SLDNS = Namespace.getNamespace("sld",
|
|
"http://www.opengis.net/sld");
|
|
|
|
try {
|
|
|
|
assertEquals(styleName, styleEl.getChild("NamedLayer", SLDNS)
|
|
.getChild("Name", SLDNS).getText());
|
|
assertEquals(
|
|
"STYLE FOR TESTING PURPOSES",
|
|
styleEl.getChild("NamedLayer", SLDNS)
|
|
.getChild("UserStyle", SLDNS)
|
|
.getChild("Title", SLDNS).getText());
|
|
} catch (NullPointerException npe) {
|
|
fail("Error in SLD");
|
|
}
|
|
|
|
// assertEquals(1475, sld.length());
|
|
|
|
assertEquals(1, reader.getStyles().size());
|
|
}
|
|
|
|
protected void cleanupTestStyle(final String styleName) {
|
|
// dry run delete to work in a known state
|
|
if (reader.existsStyle(styleName)) {
|
|
LOGGER.info("Clearing stale test style " + styleName);
|
|
boolean ok = publisher.removeStyle(styleName);
|
|
if (!ok) {
|
|
fail("Could not unpublish style " + styleName);
|
|
}
|
|
}
|
|
assertFalse("Cleanup failed", reader.existsStyle(styleName));
|
|
}
|
|
|
|
public void testPublishDeleteStyleFile() throws FileNotFoundException,
|
|
IOException {
|
|
if (!enabled()) {
|
|
return;
|
|
}
|
|
// Assume.assumeTrue(enabled);
|
|
final String styleName = "restteststyle";
|
|
|
|
File sldFile = new ClassPathResource("testdata/restteststyle.sld")
|
|
.getFile();
|
|
|
|
// known state?
|
|
cleanupTestStyle(styleName);
|
|
|
|
// test insert
|
|
boolean published = publisher.publishStyle(sldFile); // Will take the
|
|
// name from sld
|
|
// contents
|
|
assertTrue("publish() failed", published);
|
|
assertTrue(reader.existsStyle(styleName));
|
|
|
|
sldFile = new ClassPathResource("testdata/restteststyle2.sld").getFile();
|
|
published = publisher.updateStyle(sldFile, styleName); // update
|
|
assertTrue("update() failed", published);
|
|
|
|
// test delete
|
|
boolean ok = publisher.removeStyle(styleName);
|
|
assertTrue("Unpublish() failed", ok);
|
|
assertFalse(reader.existsStyle(styleName));
|
|
}
|
|
|
|
public void testPublishDeleteStyleString() throws FileNotFoundException,
|
|
IOException {
|
|
if (!enabled()) {
|
|
return;
|
|
}
|
|
// Assume.assumeTrue(enabled);
|
|
String styleName = "restteststyle";
|
|
|
|
File sldFile = new ClassPathResource("testdata/restteststyle.sld")
|
|
.getFile();
|
|
|
|
// known state?
|
|
cleanupTestStyle(styleName);
|
|
|
|
// test insert
|
|
String sldContent = IOUtils.toString(new FileInputStream(sldFile));
|
|
|
|
boolean published = publisher.publishStyle(sldContent); // Will take the
|
|
// name from sld
|
|
// contents
|
|
assertTrue("publish() failed", published);
|
|
assertTrue(reader.existsStyle(styleName));
|
|
// test delete
|
|
boolean ok = publisher.removeStyle(styleName);
|
|
assertTrue("Unpublish() failed", ok);
|
|
assertFalse(reader.existsStyle(styleName));
|
|
|
|
styleName = "restteststyle_with_name";
|
|
// test insert with name
|
|
published = publisher.publishStyle(sldContent, styleName); // Will set
|
|
// the name
|
|
assertTrue("publish() failed", published);
|
|
assertTrue(reader.existsStyle(styleName));
|
|
String newSldContent = sldContent.replace(
|
|
"<sld:Title>STYLE FOR TESTING PURPOSES</sld:Title>",
|
|
"<sld:Title>MODIFIED STYLE FOR TESTING</sld:Title>");
|
|
published = publisher.updateStyle(newSldContent, styleName); // update
|
|
assertTrue("publish() failed", published);
|
|
|
|
// test delete
|
|
ok = publisher.removeStyle(styleName);
|
|
assertTrue("Unpublish() failed", ok);
|
|
assertFalse(reader.existsStyle(styleName));
|
|
|
|
}
|
|
|
|
public void testUpdateDefaultStyle() throws FileNotFoundException,
|
|
IOException {
|
|
if (!enabled()) {
|
|
return;
|
|
}
|
|
deleteAll();
|
|
|
|
String storeName = "resttestshp";
|
|
String layerName = "cities";
|
|
|
|
final String styleName = "restteststyle";
|
|
{
|
|
File sldFile = new ClassPathResource("testdata/restteststyle.sld")
|
|
.getFile();
|
|
cleanupTestStyle(styleName);
|
|
boolean sldpublished = publisher.publishStyle(sldFile); // Will take
|
|
// the name
|
|
// from sld
|
|
// contents
|
|
assertTrue("style publish() failed", sldpublished);
|
|
assertTrue(reader.existsStyle(styleName));
|
|
}
|
|
|
|
final String styleName2 = "restteststyle2";
|
|
{
|
|
File sldFile = new ClassPathResource("testdata/restteststyle2.sld")
|
|
.getFile();
|
|
cleanupTestStyle(styleName2);
|
|
boolean sldpublished = publisher.publishStyle(sldFile, styleName2);
|
|
assertTrue("style publish() failed", sldpublished);
|
|
assertTrue(reader.existsStyle(styleName2));
|
|
}
|
|
|
|
File zipFile = new ClassPathResource("testdata/resttestshp.zip")
|
|
.getFile();
|
|
|
|
assertTrue(publisher.createWorkspace(DEFAULT_WS));
|
|
|
|
// test insert
|
|
boolean published = publisher.publishShp(DEFAULT_WS, storeName,
|
|
layerName, zipFile, "EPSG:4326", styleName);
|
|
assertTrue("publish() failed", published);
|
|
assertTrue(existsLayer(layerName));
|
|
|
|
{
|
|
RESTLayer layer = reader.getLayer(layerName);
|
|
LOGGER.info("Layer style is " + layer.getDefaultStyle());
|
|
assertEquals(styleName, layer.getDefaultStyle());
|
|
}
|
|
|
|
GSLayerEncoder le = new GSLayerEncoder();
|
|
le.setDefaultStyle(styleName2);
|
|
publisher.configureLayer(DEFAULT_WS, layerName, le);
|
|
|
|
{
|
|
RESTLayer layer = reader.getLayer(layerName);
|
|
LOGGER.info("Layer style is " + layer.getDefaultStyle());
|
|
assertEquals(styleName2, layer.getDefaultStyle());
|
|
}
|
|
|
|
// remove layer and datastore
|
|
boolean dsRemoved = publisher.removeDatastore(DEFAULT_WS, storeName,
|
|
true);
|
|
assertTrue("removeDatastore() failed", dsRemoved);
|
|
}
|
|
}
|