Example usage for org.springframework.mock.web MockHttpServletResponse getContentType

List of usage examples for org.springframework.mock.web MockHttpServletResponse getContentType

Introduction

In this page you can find the example usage for org.springframework.mock.web MockHttpServletResponse getContentType.

Prototype

@Override
    @Nullable
    public String getContentType() 

Source Link

Usage

From source file:org.geoserver.wps.gs.download.DownloadMapProcessTest.java

@Test
public void testExecuteSingleDecorated() throws Exception {
    String xml = IOUtils.toString(getClass().getResourceAsStream("mapSimpleDecorated.xml"));
    MockHttpServletResponse response = postAsServletResponse("wps", xml);
    assertEquals("image/png", response.getContentType());
    BufferedImage image = ImageIO.read(new ByteArrayInputStream(response.getContentAsByteArray()));
    ImageAssert.assertEquals(new File(SAMPLES + "watermarked.png"), image, 100);
}

From source file:org.geoserver.wps.gs.download.DownloadMapProcessTest.java

@Test
public void testExecuteMultiName() throws Exception {
    String xml = IOUtils.toString(getClass().getResourceAsStream("mapMultiName.xml"));
    MockHttpServletResponse response = postAsServletResponse("wps", xml);
    assertEquals("image/png", response.getContentType());
    assertEquals("inline; filename=result.png", response.getHeader("Content-disposition"));
    BufferedImage image = ImageIO.read(new ByteArrayInputStream(response.getContentAsByteArray()));
    ImageAssert.assertEquals(new File(SAMPLES + "mapMultiName.png"), image, 100);
}

From source file:org.geoserver.wps.gs.download.DownloadMapProcessTest.java

@Test
public void testExecuteMultiLayer() throws Exception {
    String xml = IOUtils.toString(getClass().getResourceAsStream("mapMultiLayer.xml"));
    MockHttpServletResponse response = postAsServletResponse("wps", xml);
    assertEquals("image/png", response.getContentType());
    BufferedImage image = ImageIO.read(new ByteArrayInputStream(response.getContentAsByteArray()));
    // not a typo, the output should indeed be the same as testExecuteMultiName
    ImageAssert.assertEquals(new File(SAMPLES + "mapMultiName.png"), image, 100);
}

From source file:org.geoserver.wps.gs.download.DownloadMapProcessTest.java

@Test
public void testExecuteGeotiff() throws Exception {
    String request = IOUtils.toString(getClass().getResourceAsStream("mapMultiLayer.xml"));
    request = request.replaceAll("image/png", "image/geotiff");
    MockHttpServletResponse response = postAsServletResponse("wps", request);
    assertEquals("image/geotiff", response.getContentType());
    assertEquals("attachment; filename=result.tif", response.getHeader("Content-disposition"));
}

From source file:org.geoserver.wps.gs.download.DownloadMapProcessTest.java

public void testExecutMultiLayerKmz(String mime) throws Exception {
    String request = IOUtils.toString(getClass().getResourceAsStream("mapMultiLayer.xml"));
    request = request.replaceAll("image/png", mime);
    MockHttpServletResponse response = postAsServletResponse("wps", request);
    assertEquals(KMZMapOutputFormat.MIME_TYPE, response.getContentType());
    assertEquals("inline; filename=result.kmz", response.getHeader("Content-disposition"));

    ZipInputStream zis = new ZipInputStream(new ByteArrayInputStream(response.getContentAsByteArray()));
    try {// w ww. jav a 2s .c o  m
        // first entry, the kml document itself
        ZipEntry entry = zis.getNextEntry();
        assertEquals("wms.kml", entry.getName());
        byte[] data = IOUtils.toByteArray(zis);
        Document dom = dom(new ByteArrayInputStream(data));
        // print(dom);
        assertXpathEvaluatesTo("1", "count(//kml:Folder/kml:GroundOverlay)", dom);
        String href = XMLUnit.newXpathEngine().evaluate("//kml:Folder/kml:GroundOverlay/kml:Icon/kml:href",
                dom);
        assertEquals("image.png", href);
        zis.closeEntry();

        // the ground overlay for the raster layer
        entry = zis.getNextEntry();
        assertEquals("image.png", entry.getName());
        BufferedImage image = ImageIO.read(zis);
        zis.closeEntry();
        assertNull(zis.getNextEntry());

        // check the output, same as mapMultiName
        ImageAssert.assertEquals(new File(SAMPLES + "mapMultiName.png"), image, 100);
    } finally {
        zis.close();
    }
}

From source file:org.geoserver.wps.gs.download.DownloadMapProcessTest.java

@Test
public void testTimeFilter() throws Exception {
    String xml = IOUtils.toString(getClass().getResourceAsStream("mapTimeFilter.xml"));
    MockHttpServletResponse response = postAsServletResponse("wps", xml);
    assertEquals("image/png", response.getContentType());
    BufferedImage image = ImageIO.read(new ByteArrayInputStream(response.getContentAsByteArray()));

    // same test as DimensionRasterGetMapTest#testTime
    assertPixel(image, 36, 31, new Color(246, 246, 255));
    assertPixel(image, 68, 72, new Color(255, 181, 181));

    // making extra sure
    ImageAssert.assertEquals(new File(SAMPLES + "mapTimeFilter.png"), image, 100);
}

From source file:org.geoserver.wps.gs.download.DownloadMapProcessTest.java

@Test
public void testTimeFilterTimestamped() throws Exception {
    String xml = IOUtils.toString(getClass().getResourceAsStream("mapTimeFilterTimestamped.xml"));
    MockHttpServletResponse response = postAsServletResponse("wps", xml);
    assertEquals("image/png", response.getContentType());
    BufferedImage image = ImageIO.read(new ByteArrayInputStream(response.getContentAsByteArray()));
    ImageAssert.assertEquals(new File(SAMPLES + "mapTimeFilterTimestamped.png"), image, 200);
}

From source file:org.geoserver.wps.gs.download.DownloadMapProcessTest.java

@Test
public void testTimeFilterFormattedTimestamp() throws Exception {
    String xml = IOUtils.toString(getClass().getResourceAsStream("mapTimeFilterFormattedTimestamp.xml"));
    MockHttpServletResponse response = postAsServletResponse("wps", xml);
    assertEquals("image/png", response.getContentType());
    BufferedImage image = ImageIO.read(new ByteArrayInputStream(response.getContentAsByteArray()));
    ImageAssert.assertEquals(new File(SAMPLES + "mapTimeFilterFormattedTimestamp.png"), image, 200);
}

From source file:org.geoserver.wps.gs.download.DownloadMapProcessTest.java

@Test
public void downloadMapGif() throws Exception {
    String request = IOUtils.toString(getClass().getResourceAsStream("mapSimple.xml"));
    request = request.replaceAll("image/png", "image/gif");
    MockHttpServletResponse response = postAsServletResponse("wps", request);
    assertEquals("image/gif", response.getContentType());
    BufferedImage image = ImageIO.read(new ByteArrayInputStream(response.getContentAsByteArray()));
    ImageAssert.assertEquals(new File(SAMPLES + "mapSimple.png"), image, 200);
}

From source file:org.geoserver.wps.gs.download.DownloadMapProcessTest.java

@Test
public void downloadRemoteSimple11() throws Exception {
    String request = IOUtils.toString(getClass().getResourceAsStream("mapRemoteSimple11.xml"));
    String caps111 = IOUtils.toString(getClass().getResourceAsStream("caps111.xml"));
    byte[] getMapBytes = FileUtils.readFileToByteArray(new File(SAMPLES + "mapSimple.png"));
    DownloadMapProcess process = applicationContext.getBean(DownloadMapProcess.class);
    MockHttpClient client = new MockHttpClient();
    client.expectGet(/*www.j ava  2 s  .  c om*/
            new URL("http://geoserver"
                    + ".org/geoserver/wms?service=WMS&request=GetCapabilities&version=1.1.0"),
            new MockHttpResponse(caps111, "text/xml"));
    // check it follows the links in the caps document
    client.expectGet(new URL("http://mock.test.geoserver"
            + ".org/wms11?SERVICE=WMS&LAYERS=cite:BasicPolygons&FORMAT=image%2Fpng&HEIGHT=256&TRANSPARENT=false"
            + "&REQUEST=GetMap&WIDTH=256&BBOX=-2.4,1.4,0.4,4.2&SRS=EPSG:4326&VERSION=1.1.1"),
            new MockHttpResponse(getMapBytes, "image/png"));
    // switch from the standard supplier to one using the mock client prepared above
    Supplier<HTTPClient> oldSupplier = process.getHttpClientSupplier();
    try {
        process.setHttpClientSupplier(() -> client);

        MockHttpServletResponse response = postAsServletResponse("wps", request);
        assertEquals("image/png", response.getContentType());
        BufferedImage image = ImageIO.read(new ByteArrayInputStream(response.getContentAsByteArray()));
        ImageAssert.assertEquals(new File(SAMPLES + "mapSimple.png"), image, 100);
    } finally {
        process.setHttpClientSupplier(oldSupplier);
    }
}