Java tutorial
/* (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.wms.wms_1_1_1; import static org.hamcrest.CoreMatchers.containsString; import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertThat; import static org.junit.Assert.assertTrue; import java.awt.Color; import java.awt.image.BufferedImage; import java.io.File; import java.net.URLEncoder; import java.util.Collections; import javax.imageio.ImageIO; import javax.xml.namespace.QName; import org.apache.commons.io.IOUtils; import org.geoserver.catalog.Catalog; import org.geoserver.catalog.LegendInfo; import org.geoserver.catalog.impl.LegendInfoImpl; import org.geoserver.data.test.MockData; import org.geoserver.data.test.SystemTestData; import org.geoserver.data.test.TestData; import org.geoserver.platform.resource.Resource; import org.geoserver.wms.WMSTestSupport; import org.geotools.util.Converters; import org.junit.Test; import org.springframework.mock.web.MockHttpServletResponse; public class GetLegendGraphicTest extends WMSTestSupport { @Override protected void onSetUp(SystemTestData testData) throws Exception { super.onSetUp(testData); Catalog catalog = getCatalog(); testData.addStyle("paramFill", "paramFill.sld", getClass(), catalog); testData.addStyle("paramStroke", "paramStroke.sld", getClass(), catalog); testData.addStyle("raster", "raster.sld", getClass(), catalog); testData.addStyle("rasterScales", "rasterScales.sld", getClass(), catalog); testData.addStyle("Population", "Population.sld", getClass(), catalog); testData.addStyle("uom", "uomStroke.sld", getClass(), catalog); testData.addVectorLayer(new QName(MockData.SF_URI, "states", MockData.SF_PREFIX), Collections.EMPTY_MAP, "states.properties", getClass(), catalog); LegendInfo legend = new LegendInfoImpl(); legend.setWidth(22); legend.setHeight(22); legend.setFormat("image/png"); legend.setOnlineResource("legend.png"); File file = getResourceLoader().createFile("styles", "legend.png"); getResourceLoader().copyFromClassPath("../legend.png", file, getClass()); testData.addStyle(null, "custom", "point_test.sld", getClass(), catalog, legend); } /** * Tests GML output does not break when asking for an area that has no data with * GML feature bounding enabled * */ @Test public void testPlain() throws Exception { BufferedImage image = getAsImage("wms?service=WMS&version=1.1.1&request=GetLegendGraphic" + "&layer=" + getLayerId(MockData.LAKES) + "&style=Lakes" + "&format=image/png&width=20&height=20", "image/png"); assertPixel(image, 10, 10, Converters.convert("#4040C0", Color.class)); } /** * Tests GML output does not break when asking for an area that has no data with * GML feature bounding enabled * */ @Test public void testEnv() throws Exception { // no params, use fallback String base = "wms?service=WMS&version=1.1.1&request=GetLegendGraphic" + "&layer=" + getLayerId(MockData.LAKES) + "&style=paramFill" + "&format=image/png&width=20&height=20"; BufferedImage image = getAsImage(base, "image/png"); assertPixel(image, 10, 10, Converters.convert("#FFFFFF", Color.class)); // specify color explicitly image = getAsImage(base + "&env=color:#FF0000", "image/png"); assertPixel(image, 10, 10, Converters.convert("#FF0000", Color.class)); } /** * Tests an custom legend graphic */ @Test public void testCustomLegend() throws Exception { String base = "wms?service=WMS&version=1.1.1&request=GetLegendGraphic" + "&layer=sf:states&style=custom" + "&format=image/png&width=22&height=22"; BufferedImage image = getAsImage(base, "image/png"); Resource resource = getResourceLoader().get("styles/legend.png"); BufferedImage expected = ImageIO.read(resource.file()); assertEquals(getPixelColor(expected, 10, 2).getRGB(), getPixelColor(image, 10, 2).getRGB()); // test rescale base = "wms?service=WMS&version=1.1.1&request=GetLegendGraphic" + "&layer=sf:states&style=custom" + "&format=image/png&width=16&height=16"; image = getAsImage(base, "image/png"); Color expectedColor = getPixelColor(expected, 11, 11); Color actualColor = getPixelColor(image, 8, 8); assertEquals("red", expectedColor.getRed(), actualColor.getRed()); assertEquals("green", expectedColor.getGreen(), actualColor.getGreen()); assertEquals("blue", expectedColor.getBlue(), actualColor.getBlue()); assertEquals("alpha", expectedColor.getAlpha(), actualColor.getAlpha()); } /** * Tests an unscaled states legend */ @Test public void testStatesLegend() throws Exception { String base = "wms?service=WMS&version=1.1.1&request=GetLegendGraphic" + "&layer=sf:states&style=Population" + "&format=image/png&width=20&height=20"; BufferedImage image = getAsImage(base, "image/png"); // check RGB is in the expected positions assertPixel(image, 10, 10, Color.RED); assertPixel(image, 10, 30, Color.GREEN); assertPixel(image, 10, 50, Color.BLUE); } /** * Tests a dpi rescaled legend */ @Test public void testStatesLegendDpiRescaled() throws Exception { String base = "wms?service=WMS&version=1.1.1&request=GetLegendGraphic" + "&layer=sf:states&style=Population" + "&format=image/png&width=20&height=20&legend_options=dpi:180"; BufferedImage image = getAsImage(base, "image/png"); assertPixel(image, 20, 20, Color.RED); assertPixel(image, 20, 60, Color.GREEN); assertPixel(image, 20, 100, Color.BLUE); Color linePixel = getPixelColor(image, 20, 140); assertTrue(linePixel.getRed() < 10); assertTrue(linePixel.getGreen() < 10); assertTrue(linePixel.getBlue() < 10); } /** * Tests a dpi rescaled legend with specific rule name */ @Test public void testStatesLegendDpiRescaledSingleRule() throws Exception { String base = "wms?service=WMS&version=1.1.1&request=GetLegendGraphic" + "&layer=sf:states&style=Population" + "&format=image/png&width=20&height=20&legend_options=dpi:180&rule=2-4M"; BufferedImage image = getAsImage(base, "image/png"); // ImageIO.write(image, "PNG", new java.io.File("/tmp/rule.png")); // just one rule assertEquals(40, image.getWidth()); assertEquals(40, image.getHeight()); // the red one, big assertPixel(image, 20, 20, Color.RED); } /** * Tests a uom rescaled legend */ @Test public void testStatesLegendUomRescaled() throws Exception { String base = "wms?service=WMS&version=1.1.1&request=GetLegendGraphic" + "&layer=sf:states&style=uom" + "&format=image/png&width=20&height=20&scale=1000000"; BufferedImage image = getAsImage(base, "image/png"); assertPixel(image, 10, 10, Color.BLUE); assertPixel(image, 5, 10, Color.WHITE); assertPixel(image, 1, 10, Color.WHITE); // halve the scale denominator, we're zooming in, the thickness should double base = "wms?service=WMS&version=1.1.1&request=GetLegendGraphic" + "&layer=sf:states&style=uom" + "&format=image/png&width=20&height=20&scale=500000"; image = getAsImage(base, "image/png"); assertPixel(image, 10, 10, Color.BLUE); assertPixel(image, 5, 10, Color.BLUE); assertPixel(image, 1, 10, Color.WHITE); } /** * Tests a dpi _and_ uom rescaled image */ @Test public void testStatesLegendDpiUomRescaled() throws Exception { // halve the scale denominator, we're zooming in, the thickness should double String base = "wms?service=WMS&version=1.1.1&request=GetLegendGraphic" + "&layer=sf:states&style=uom" + "&format=image/png&width=20&height=20&scale=1000000&&legend_options=dpi:180"; BufferedImage image = getAsImage(base, "image/png"); assertPixel(image, 30, 10, Color.BLUE); assertPixel(image, 20, 20, Color.BLUE); assertPixel(image, 10, 30, Color.BLUE); // this assertion check that the thickness is not greater than twice the previous value assertPixel(image, 24, 6, Color.WHITE); assertPixel(image, 1, 20, Color.WHITE); } @Test public void testEntityExpansionSldBody() throws Exception { String base = "wms?LEGEND_OPTIONS=forceLabels:on&REQUEST=GetLegendGraphic&VERSION=1.0.0&FORMAT=image/png&WIDTH=200&HEIGHT=20&LAYER=" + getLayerId(MockData.POLYGONS) + "&SLD_BODY="; String sld = IOUtils.toString(TestData.class.getResource("externalEntities.sld")); MockHttpServletResponse response = getAsServletResponse(base + URLEncoder.encode(sld, "UTF-8")); // should fail with an error message poiting at entity resolution assertEquals("application/vnd.ogc.se_xml", response.getContentType()); final String content = response.getContentAsString(); assertThat(content, containsString("Entity resolution disallowed")); assertThat(content, containsString("/this/file/does/not/exist")); } }