package org.geoserver.wcs2_0.xml;
import static junit.framework.TestCase.assertEquals;
import static junit.framework.TestCase.assertTrue;
import java.io.File;
import org.apache.commons.io.FileUtils;
import org.geoserver.wcs2_0.WCSTestSupport;
import org.geoserver.wcs2_0.exception.WCS20Exception.WCS20ExceptionCode;
import org.geotools.coverage.grid.GridCoverage2D;
import org.geotools.gce.geotiff.GeoTiffReader;
import org.geotools.referencing.CRS;
import org.junit.Test;
import com.mockrunner.mock.web.MockHttpServletResponse;
/**
* Testing range subsetting capabilities
*
* @author Simone Giannecchini, GeoSolutions
* TODO more tests with a landsat
*/
public class RangeSubsetExtentionTest extends WCSTestSupport {
@Test
public void testBasic() throws Exception {
final File xml= new File("./src/test/resources/rangesubset/requestGetCoverageRangeSubsetting.xml");
final String request= FileUtils.readFileToString(xml);
MockHttpServletResponse response = postAsServletResponse("wcs", request);
assertEquals("image/tiff", response.getContentType());
byte[] tiffContents = getBinary(response);
File file = File.createTempFile("bm_gtiff", "bm_gtiff.tiff", new File("./target"));
FileUtils.writeByteArrayToFile(file, tiffContents);
final GeoTiffReader reader = new GeoTiffReader(file);
assertTrue(CRS.equalsIgnoreMetadata(reader.getCoordinateReferenceSystem(), CRS.decode("EPSG:4326",true)));
assertEquals(360, reader.getOriginalGridRange().getSpan(0));
assertEquals(360, reader.getOriginalGridRange().getSpan(1));
final GridCoverage2D coverage = reader.read(null);
assertEquals(1, coverage.getSampleDimensions().length);
GridCoverage2D sourceCoverage = (GridCoverage2D) this.getCatalog().getCoverageByName("BlueMarble").getGridCoverageReader(null, null).read(null);
assertEnvelopeEquals(sourceCoverage, coverage);
reader.dispose();
scheduleForCleaning(coverage);
scheduleForCleaning(sourceCoverage);
}
@Test
public void mixed() throws Exception {
final File xml= new File("./src/test/resources/rangesubset/requestGetCoverageRangeSubsettingInterval2.xml");
final String request= FileUtils.readFileToString(xml);
MockHttpServletResponse response = postAsServletResponse("wcs", request);
assertEquals("image/tiff", response.getContentType());
byte[] tiffContents = getBinary(response);
File file = File.createTempFile("bm_gtiff", "bm_gtiff.tiff", new File("./target"));
FileUtils.writeByteArrayToFile(file, tiffContents);
final GeoTiffReader reader = new GeoTiffReader(file);
assertTrue(CRS.equalsIgnoreMetadata(reader.getCoordinateReferenceSystem(), CRS.decode("EPSG:4326",true)));
assertEquals(360, reader.getOriginalGridRange().getSpan(0));
assertEquals(360, reader.getOriginalGridRange().getSpan(1));
final GridCoverage2D coverage = reader.read(null);
assertEquals(5, coverage.getSampleDimensions().length);
GridCoverage2D sourceCoverage = (GridCoverage2D) this.getCatalog().getCoverageByName("BlueMarble").getGridCoverageReader(null, null).read(null);
assertEnvelopeEquals(sourceCoverage, coverage);
reader.dispose();
scheduleForCleaning(coverage);
scheduleForCleaning(sourceCoverage);
}
@Test
public void testWrong() throws Exception {
final File xml= new File("./src/test/resources/rangesubset/requestGetCoverageWrongRangeSubsetting.xml");
final String request= FileUtils.readFileToString(xml);
MockHttpServletResponse response = postAsServletResponse("wcs", request);
assertEquals("application/xml", response.getContentType());
checkOws20Exception(response, 404, WCS20ExceptionCode.NoSuchField.getExceptionCode(), "Band1");
}
@Test
public void test9to3() throws Exception {
final File xml= new File("./src/test/resources/rangesubset/requestGetCoverageRangeSubsetting9to3.xml");
final String request= FileUtils.readFileToString(xml);
MockHttpServletResponse response = postAsServletResponse("wcs", request);
assertEquals("image/tiff", response.getContentType());
byte[] tiffContents = getBinary(response);
File file = File.createTempFile("gtiff", "gtiff.tiff", new File("./target"));
FileUtils.writeByteArrayToFile(file, tiffContents);
final GeoTiffReader reader = new GeoTiffReader(file);
assertTrue(CRS.equalsIgnoreMetadata(reader.getCoordinateReferenceSystem(), CRS.decode("EPSG:32611",true)));
assertEquals(68, reader.getOriginalGridRange().getSpan(0));
assertEquals(56 , reader.getOriginalGridRange().getSpan(1));
final GridCoverage2D coverage = reader.read(null);
assertEquals(3, coverage.getSampleDimensions().length);
GridCoverage2D sourceCoverage = (GridCoverage2D) this.getCatalog().getCoverageByName("multiband").getGridCoverageReader(null, null).read(null);
assertEnvelopeEquals(sourceCoverage, coverage);
reader.dispose();
scheduleForCleaning(coverage);
scheduleForCleaning(sourceCoverage);
}
@Test
public void test9to4() throws Exception {
final File xml= new File("./src/test/resources/rangesubset/requestGetCoverageRangeSubsetting9to4.xml");
final String request= FileUtils.readFileToString(xml);
MockHttpServletResponse response = postAsServletResponse("wcs", request);
assertEquals("image/tiff", response.getContentType());
byte[] tiffContents = getBinary(response);
File file = File.createTempFile("gtiff", "gtiff.tiff", new File("./target"));
FileUtils.writeByteArrayToFile(file, tiffContents);
final GeoTiffReader reader = new GeoTiffReader(file);
assertTrue(CRS.equalsIgnoreMetadata(reader.getCoordinateReferenceSystem(), CRS.decode("EPSG:32611",true)));
assertEquals(68, reader.getOriginalGridRange().getSpan(0));
assertEquals(56 , reader.getOriginalGridRange().getSpan(1));
final GridCoverage2D coverage = reader.read(null);
assertEquals(4, coverage.getSampleDimensions().length);
GridCoverage2D sourceCoverage = (GridCoverage2D) this.getCatalog().getCoverageByName("multiband").getGridCoverageReader(null, null).read(null);
assertEnvelopeEquals(sourceCoverage, coverage);
reader.dispose();
scheduleForCleaning(coverage);
scheduleForCleaning(sourceCoverage);
}
@Test
public void test9to7() throws Exception {
final File xml= new File("./src/test/resources/rangesubset/requestGetCoverageRangeSubsetting9to7.xml");
final String request= FileUtils.readFileToString(xml);
MockHttpServletResponse response = postAsServletResponse("wcs", request);
assertEquals("image/tiff", response.getContentType());
byte[] tiffContents = getBinary(response);
File file = File.createTempFile("gtiff", "gtiff.tiff", new File("./target"));
FileUtils.writeByteArrayToFile(file, tiffContents);
final GeoTiffReader reader = new GeoTiffReader(file);
assertTrue(CRS.equalsIgnoreMetadata(reader.getCoordinateReferenceSystem(), CRS.decode("EPSG:32611",true)));
assertEquals(68, reader.getOriginalGridRange().getSpan(0));
assertEquals(56 , reader.getOriginalGridRange().getSpan(1));
final GridCoverage2D coverage = reader.read(null);
assertEquals(7, coverage.getSampleDimensions().length);
GridCoverage2D sourceCoverage = (GridCoverage2D) this.getCatalog().getCoverageByName("multiband").getGridCoverageReader(null, null).read(null);
assertEnvelopeEquals(sourceCoverage, coverage);
reader.dispose();
scheduleForCleaning(coverage);
scheduleForCleaning(sourceCoverage);
}
@Test
public void testRange() throws Exception {
final File xml= new File("./src/test/resources/rangesubset/requestGetCoverageRangeSubsettingInterval.xml");
final String request= FileUtils.readFileToString(xml);
MockHttpServletResponse response = postAsServletResponse("wcs", request);
assertEquals("image/tiff", response.getContentType());
byte[] tiffContents = getBinary(response);
File file = File.createTempFile("bm_gtiff", "bm_gtiff.tiff", new File("./target"));
FileUtils.writeByteArrayToFile(file, tiffContents);
final GeoTiffReader reader = new GeoTiffReader(file);
assertTrue(CRS.equalsIgnoreMetadata(reader.getCoordinateReferenceSystem(), CRS.decode("EPSG:4326",true)));
assertEquals(360, reader.getOriginalGridRange().getSpan(0));
assertEquals(360, reader.getOriginalGridRange().getSpan(1));
final GridCoverage2D coverage = reader.read(null);
assertEquals(3, coverage.getSampleDimensions().length);
GridCoverage2D sourceCoverage = (GridCoverage2D) this.getCatalog().getCoverageByName("BlueMarble").getGridCoverageReader(null, null).read(null);
assertEnvelopeEquals(sourceCoverage, coverage);
reader.dispose();
scheduleForCleaning(coverage);
scheduleForCleaning(sourceCoverage);
}
}