LayerControllerTest.java

/* (c) 2014 - 2016 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.rest.catalog;

import static org.custommonkey.xmlunit.XMLAssert.assertXpathEvaluatesTo;
import static org.custommonkey.xmlunit.XMLAssert.assertXpathExists;
import static org.custommonkey.xmlunit.XMLAssert.assertXpathNotExists;
import static org.junit.Assert.*;

import java.io.IOException;

import net.sf.json.JSONObject;

import org.geoserver.catalog.Catalog;
import org.geoserver.catalog.LayerInfo;
import org.geoserver.catalog.StyleInfo;
import org.geoserver.data.test.SystemTestData;
import org.junit.After;
import org.junit.Test;
import org.w3c.dom.Document;
import static org.geoserver.rest.RestBaseController.ROOT_PATH;

import org.springframework.mock.web.MockHttpServletResponse;

public class LayerControllerTest extends CatalogRESTTestSupport {

    @After
    public void revertChanges() throws IOException {
        revertLayer(SystemTestData.BUILDINGS);
        revertLayer(SystemTestData.BRIDGES);
        StyleInfo si = getCatalog().getStyleByName("cite", "foo");
        if(si != null) {
            getCatalog().remove(si);
        }
    }
    
    @Override
    protected void onTearDown(SystemTestData testData) throws Exception {
        super.onTearDown(testData);
    }

    @Test
    public void testGetAsXML() throws Exception {
        Document dom = getAsDOM( ROOT_PATH + "/layers/cite:Buildings.xml",200);
        assertEquals( "layer", dom.getDocumentElement().getNodeName() );
        assertXpathEvaluatesTo("Buildings", "/layer/name", dom );
        // check the layer name is actually the first child (GEOS-3336 risked modifying
        // the order)
        assertXpathEvaluatesTo("Buildings", "/layer/*[1]", dom );
        assertXpathEvaluatesTo("http://localhost:8080/geoserver"+ROOT_PATH+"/styles/Buildings.xml",
                "/layer/defaultStyle/atom:link/attribute::href", dom);
    }
    
    @Test
    public void testGetAsHTML() throws Exception {
        getAsDOM(ROOT_PATH + "/layers/cite:Buildings.html",200);
    }
    
    @Test
    public void testGetWrongLayer() throws Exception {
        // Parameters for the request
        String layer = "cite:Buildingssssss";
        // Request path
        String requestPath = ROOT_PATH + "/layers/" + layer + ".html";
        // Exception path
        String exception = "No such layer: " + layer;
        // First request should thrown an exception
        MockHttpServletResponse response = getAsServletResponse(requestPath);
        assertEquals(404, response.getStatus());
        assertTrue(response.getContentAsString().contains(
                exception));
        // Same request with ?quietOnNotFound should not throw an exception
        response = getAsServletResponse(requestPath + "?quietOnNotFound=true");
        assertEquals(404, response.getStatus());
        String message = response.getContentAsString();
        assertFalse(message, message.contains(exception));
        // No exception thrown
        assertTrue(response.getContentAsString().isEmpty());
    }
    
    @Test
    public void testGetAllAsXML() throws Exception {
        Document dom = getAsDOM( ROOT_PATH + "/layers.xml",200);
        assertXpathEvaluatesTo(catalog.getLayers().size()+"", "count(//layer)", dom );
    }
    
    @Test
    public void testGetAllAsHTML() throws Exception {
        getAsDOM( ROOT_PATH + "/layers.html",200);
    }
    
    @Test
    public void testPut() throws Exception {
        LayerInfo l = catalog.getLayerByName( "cite:Buildings" );
        assertEquals( "Buildings", l.getDefaultStyle().getName() );
        String xml = 
            "<layer>" +
              "<defaultStyle>Forests</defaultStyle>" + 
              "<styles>" + 
                "<style>Ponds</style>" +
              "</styles>" + 
            "</layer>";
        MockHttpServletResponse response = 
            putAsServletResponse(ROOT_PATH + "/layers/cite:Buildings", xml, "text/xml");
        assertEquals( 200, response.getStatus() );
        
        l = catalog.getLayerByName("cite:Buildings");
        assertEquals( "Forests", l.getDefaultStyle().getName() );
    }
    @Test
    public void testPutNonDestructive() throws Exception {
        LayerInfo l = catalog.getLayerByName( "cite:Buildings" );
        
        assertTrue(l.isEnabled());
        boolean isAdvertised = l.isAdvertised();
        boolean isOpaque = l.isOpaque();
        boolean isQueryable = l.isQueryable();
        
        String xml = 
            "<layer>" +
              "<defaultStyle>Forests</defaultStyle>" + 
              "<styles>" + 
                "<style>Ponds</style>" +
              "</styles>" + 
            "</layer>";
        MockHttpServletResponse response = 
            putAsServletResponse(ROOT_PATH + "/layers/cite:Buildings", xml, "text/xml");
        assertEquals( 200, response.getStatus() );
        
        l = catalog.getLayerByName("cite:Buildings");
        
        assertTrue(l.isEnabled());
        assertEquals(isAdvertised, l.isAdvertised());
        assertEquals(isOpaque, l.isOpaque());
        assertEquals(isQueryable, l.isQueryable());
    }
    
    @Test
    public void testUpdateStyleJSON() throws Exception {
        LayerInfo l = catalog.getLayerByName( "cite:Buildings" );
        assertEquals( "Buildings", l.getDefaultStyle().getName() );
        JSONObject json = (JSONObject) getAsJSON(ROOT_PATH + "/layers/cite:Buildings.json");
        // print(json);
        JSONObject layer = (JSONObject) json.get("layer");
        JSONObject style = (JSONObject) layer.get("defaultStyle");
        style.put("name", "polygon");
        style.put("href", "http://localhost:8080/geoserver/rest/styles/polygon.json");
        String updatedJson = json.toString();
        MockHttpServletResponse response = 
                putAsServletResponse(ROOT_PATH + "/layers/cite:Buildings", updatedJson, "application/json");
        assertEquals( 200, response.getStatus() );

        l = catalog.getLayerByName("cite:Buildings");
        assertEquals( "polygon", l.getDefaultStyle().getName() );
    }
    
    @Test
    public void testDelete() throws Exception {
        assertNotNull(catalog.getLayerByName( "cite:Buildings" ));
        assertEquals(200, deleteAsServletResponse(ROOT_PATH + "/layers/cite:Buildings").getStatus());
        assertNull(catalog.getLayerByName( "cite:Buildings" ));
    }
    
    @Test
    public void testDeleteRecursive() throws Exception {
        assertNotNull(catalog.getLayerByName( "cite:Buildings" ));
        assertNotNull(catalog.getFeatureTypeByName( "cite", "Buildings" ));
        
        assertEquals(200, deleteAsServletResponse(ROOT_PATH + "/layers/cite:Buildings").getStatus());
        
        assertNull(catalog.getLayerByName( "cite:Buildings" ));
        assertNotNull(catalog.getFeatureTypeByName( "cite", "Buildings" ));
        
        assertNotNull(catalog.getLayerByName( "cite:Bridges" ));
        assertNotNull(catalog.getFeatureTypeByName( "cite", "Bridges" ));
        
        assertEquals(200, deleteAsServletResponse(ROOT_PATH + "/layers/cite:Bridges?recurse=true").getStatus());
    
        assertNull(catalog.getLayerByName( "cite:Bridges" ));
        assertNull(catalog.getFeatureTypeByName( "cite", "Bridges" ));
    }

    @Test
    public void testPutWorkspaceStyle() throws Exception {
        Catalog cat = getCatalog();
        assertNull(cat.getStyleByName("foo"));
        assertNull(cat.getStyleByName("cite", "foo"));

        String xml = 
            "<style>" +
              "<name>foo</name>" +
              "<filename>foo.sld</filename>" + 
            "</style>";
        
        MockHttpServletResponse response =
            postAsServletResponse(ROOT_PATH + "/workspaces/cite/styles", xml);
        
        System.out.println( response.getContentAsString() );
        assertEquals(201, response.getStatus());
        assertNotNull(cat.getStyleByName("cite", "foo"));

        xml = 
            "<layer>" + 
                "<defaultStyle>" + 
                    "<name>foo</name>" +
                    "<workspace>cite</workspace>" +
                "</defaultStyle>" +
                "<enabled>true</enabled>" + 
            "</layer>";
        response =
            putAsServletResponse(ROOT_PATH + "/layers/cite:Buildings", xml, "application/xml");
        assertEquals(200, response.getStatus());

        LayerInfo l = cat.getLayerByName("cite:Buildings");
        assertNotNull(l.getDefaultStyle());
        assertEquals("foo", l.getDefaultStyle().getName());
        assertNotNull(l.getDefaultStyle().getWorkspace());

        Document dom = getAsDOM(ROOT_PATH + "/layers/cite:Buildings.xml",200);
        assertXpathExists("/layer/defaultStyle/name[text() = 'cite:foo']", dom);
        assertXpathExists("/layer/defaultStyle/workspace[text() = 'cite']", dom);
        assertXpathEvaluatesTo("http://localhost:8080/geoserver"+ROOT_PATH+"/workspaces/cite/styles/foo.xml", 
            "//defaultStyle/atom:link/@href", dom );
    }
    
    @Test
    public void testPutWorkspaceAlternateStyle() throws Exception {
        Catalog cat = getCatalog();
        assertNull(cat.getStyleByName("foo"));
        assertNull(cat.getStyleByName("cite", "foo"));

        String xml = 
            "<style>" +
              "<name>foo</name>" +
              "<filename>foo.sld</filename>" + 
            "</style>";

        MockHttpServletResponse response =
            postAsServletResponse(ROOT_PATH + "/workspaces/cite/styles", xml);
        assertEquals(201, response.getStatus());
        assertNotNull(cat.getStyleByName("cite", "foo"));

        xml = 
            "<layer>" + 
                "<styles>" +
                  "<style>" +
                    "<name>foo</name>" +
                    "<workspace>cite</workspace>" +
                  "</style>" +
                "</styles>" +
                "<enabled>true</enabled>" + 
            "</layer>";
        response =
            putAsServletResponse(ROOT_PATH + "/layers/cite:Buildings", xml, "application/xml");
        assertEquals(200, response.getStatus());

        LayerInfo l = cat.getLayerByName("cite:Buildings");
        assertNotNull(l.getDefaultStyle());
        StyleInfo style = l.getStyles().iterator().next();
        assertEquals("foo", style.getName());
        assertNotNull(style.getWorkspace());

        Document dom = getAsDOM(ROOT_PATH + "/layers/cite:Buildings.xml",200);
        assertXpathExists("/layer/styles/style/name[text() = 'cite:foo']", dom);
        assertXpathExists("/layer/styles/style/workspace[text() = 'cite']", dom);
        assertXpathEvaluatesTo("http://localhost:8080/geoserver"+ROOT_PATH+"/workspaces/cite/styles/foo.xml", 
            "//styles/style/atom:link/@href", dom );
    }

    @Test
    public void testPutDefaultWMSInterpolationMethod() throws Exception {
        Catalog cat = getCatalog();
        LayerInfo l = cat.getLayerByName("cite:Buildings");
        assertNotNull(l);
        assertNull(l.getDefaultWMSInterpolationMethod());

        Document dom = getAsDOM(ROOT_PATH + "/layers/cite:Buildings.xml",200);
        assertEquals("layer", dom.getDocumentElement().getNodeName());
        assertXpathNotExists("/layer/defaultWMSInterpolationMethod", dom);

        String xml =
            "<layer>" +
                "<defaultWMSInterpolationMethod>" +
                    "Nearest" +
                "</defaultWMSInterpolationMethod>" +
                "<enabled>true</enabled>" +
            "</layer>";
        MockHttpServletResponse response =
            putAsServletResponse(ROOT_PATH + "/layers/cite:Buildings", xml, "application/xml");
        assertEquals(200, response.getStatus());

        l = cat.getLayerByName("cite:Buildings");
        assertNotNull(l.getDefaultWMSInterpolationMethod());
        assertEquals(LayerInfo.WMSInterpolation.Nearest, l.getDefaultWMSInterpolationMethod());

        dom = getAsDOM(ROOT_PATH + "/layers/cite:Buildings.xml",200);
        assertXpathEvaluatesTo("1", "count(/layer/defaultWMSInterpolationMethod)", dom);
        assertXpathExists("/layer/defaultWMSInterpolationMethod[text() = 'Nearest']", dom);
    }
}