CustomDimensionTimeTest.java

/* (c) 2014 Open Source Geospatial Foundation - all rights reserved
 * (c) 2001 - 2014 OpenPlans
 * This code is licensed under the GPL 2.0 license, available at the root
 * application directory.
 */
package org.geoserver.wms.dimension;

import static org.junit.Assert.assertEquals;

import java.text.SimpleDateFormat;
import java.util.Collections;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.TimeZone;

import javax.xml.namespace.QName;

import org.geoserver.catalog.testreader.CustomFormat;
import org.geoserver.data.test.MockData;
import org.geoserver.data.test.SystemTestData;
import org.geoserver.wms.GetMapRequest;
import org.geoserver.wms.MapLayerInfo;
import org.geoserver.wms.WMS;
import org.geoserver.wms.WMSTestSupport;
import org.geotools.coverage.grid.io.GridCoverage2DReader;
import org.junit.Before;
import org.junit.Test;
import org.opengis.filter.Filter;
import org.opengis.parameter.GeneralParameterValue;
import org.opengis.parameter.ParameterValue;

/**
 * Tests the WMS default value support for a custom dimension 
 * that uses the java.util.Date class rather than Strings
 */
public class CustomDimensionTimeTest extends WMSTestSupport {    
  
    private static final QName WATTEMP_TIME = new QName(MockData.SF_URI, "watertemp_time",
            MockData.SF_PREFIX);

    private static SimpleDateFormat DF = new SimpleDateFormat("yyyy-MM-dd hh:mm:ss");
       
    static {
        DF.setTimeZone(TimeZone.getTimeZone("UTC"));
    }

    WMS wms;

    @Override
    protected void onSetUp(SystemTestData testData) throws Exception {
        super.onSetUp(testData);
    }

    @Before
    public void setup() throws Exception {
        wms = getWMS(); //with the initialized application context          
        ((SystemTestData)testData).addRasterLayer(WATTEMP_TIME, "custwatertemp_time.zip", null, 
                Collections.emptyMap(), 
                getClass(), getCatalog());
    }
        
    @Test
    public void testTimeDimension() throws Exception {        
        MapLayerInfo mapLayerInfo = new MapLayerInfo(getCatalog().getLayerByName(WATTEMP_TIME.getLocalPart()));
        final GridCoverage2DReader reader = (GridCoverage2DReader) mapLayerInfo.getCoverageReader();
        GetMapRequest req = new GetMapRequest();
        req.setRawKvp(new HashMap<String, String>());
        req.getRawKvp().put("DIM_" + CustomFormat.CUSTOM_DIMENSION_NAME, 
                "2001-05-01T00:00:00.000Z, 2001-05-02T00:00:00.000Z");

        GeneralParameterValue[] readParam = wms.getWMSReadParameters(req, mapLayerInfo, Filter.INCLUDE, null, null, reader, false);
        @SuppressWarnings("unchecked")
        ParameterValue<List<Date>> val = (ParameterValue<List<Date>>) readParam[readParam.length - 1];
        assertEquals(2, val.getValue().size());
        assertEquals(DF.parse("2001-05-01 00:00:00"), val.getValue().get(0));
        assertEquals(DF.parse("2001-05-02 00:00:00"), val.getValue().get(1));
    }

   
}