DynamicDimensionsTest.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.wcs;

import static org.junit.Assert.*;

import java.awt.image.BufferedImage;

import javax.imageio.ImageIO;
import javax.xml.namespace.QName;

import org.geoserver.catalog.CoverageInfo;
import org.geoserver.catalog.DimensionInfo;
import org.geoserver.catalog.DimensionPresentation;
import org.geoserver.catalog.ResourceInfo;
import org.geoserver.catalog.impl.DimensionInfoImpl;
import org.geoserver.data.test.MockData;
import org.geoserver.data.test.SystemTestData;
import org.geoserver.wcs.test.CoverageTestSupport;
import org.junit.Test;

import org.springframework.mock.web.MockHttpServletResponse;

/**
 * Tests for custom dimensions in WCS requests.
 * 
 * @author Andrea Aime
 */
public class DynamicDimensionsTest extends CoverageTestSupport {

    private static final String DIMENSION_NAME = "wavelength";
    private static final QName WATTEMP = new QName(MockData.DEFAULT_URI, "watertemp", MockData.DEFAULT_PREFIX);


    @Override
    protected void onSetUp(SystemTestData testData) throws Exception {
        super.onSetUp(testData);
        
        testData.addRasterLayer(WATTEMP, "watertempDynamicDims.zip", null, null, SystemTestData.class, getCatalog());
        setupRasterDimension(DIMENSION_NAME, DimensionPresentation.LIST);
    }
    
    @Test
    public void testGetCoverageBadValue() throws Exception {
        // check that we get no data when requesting an incorrect value for custom dimension
        String request = getWaterTempRequest("bad_dimension_value");
        MockHttpServletResponse response = postAsServletResponse("wcs", request);
        BufferedImage image = ImageIO.read(getBinaryInputStream(response));
        assertNull(image);
    }
    
    @Test
    public void testGetCoverageGoodValue() throws Exception {
        // check that we get data when requesting a correct value for custom dimension
        String request = getWaterTempRequest("100");
        MockHttpServletResponse response = postAsServletResponse("wcs", request);
        BufferedImage image = ImageIO.read(getBinaryInputStream(response));
        assertNotNull(image);
        assertEquals("image/tiff", response.getContentType());
    }

    private String getWaterTempRequest(String dimensionValue) {
        String request =
            "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n" + 
                "<GetCoverage version=\"1.0.0\" service=\"WCS\"\n" + 
                "  xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\" xmlns=\"http://www.opengis.net/wcs\"\n" + 
                "  xmlns:ows=\"http://www.opengis.net/ows/1.1\" xmlns:gml=\"http://www.opengis.net/gml\"\n" + 
                "  xmlns:ogc=\"http://www.opengis.net/ogc\"\n" + 
                "  xsi:schemaLocation=\"http://www.opengis.net/wcs http://schemas.opengis.net/wcs/1.0.0/getCoverage.xsd\">\n" + 
                "  <sourceCoverage>" + getLayerId(WATTEMP) + "</sourceCoverage>\n" + 
                "  <domainSubset>\n" + 
                "    <spatialSubset>\n" + 
                "      <gml:Envelope srsName=\"EPSG:4326\">\n" + 
                "        <gml:pos>0.237 40.562</gml:pos>\n" + 
                "        <gml:pos>14.593 44.558</gml:pos>\n" + 
                "      </gml:Envelope>\n" + 
                "      <gml:Grid dimension=\"2\">\n" + 
                "        <gml:limits>\n" + 
                "          <gml:GridEnvelope>\n" + 
                "            <gml:low>0 0</gml:low>\n" + 
                "            <gml:high>25 24</gml:high>\n" + 
                "          </gml:GridEnvelope>\n" + 
                "        </gml:limits>\n" + 
                "        <gml:axisName>x</gml:axisName>\n" + 
                "        <gml:axisName>y</gml:axisName>\n" + 
                "      </gml:Grid>\n" + 
                "    </spatialSubset>\n" + 
                "  </domainSubset>\n" + 
                "  <rangeSubset>\n" + 
                "    <axisSubset name=\"" + DIMENSION_NAME + "\">\n" + 
                "      <singleValue>" + dimensionValue + "</singleValue>\n" + 
                "    </axisSubset>\n" + 
                "  </rangeSubset>\n" + 
                "  <output>\n" + 
                "    <crs>EPSG:4326</crs>\n" + 
                "    <format>GEOTIFF</format>\n" + 
                "  </output>\n" + 
                "</GetCoverage>";
        return request;
    }

    private void setupRasterDimension(String metadata, DimensionPresentation presentation) {
        CoverageInfo info = getCatalog().getCoverageByName(WATTEMP.getLocalPart());
        DimensionInfo di = new DimensionInfoImpl();
        di.setEnabled(true);
        di.setPresentation(presentation);
        info.getMetadata().put(ResourceInfo.CUSTOM_DIMENSION_PREFIX + metadata, di);
        getCatalog().save(info);
    }
}