MapProxy 1.16.1__py3-none-any.whl

mapproxy/test/system/test_coverage.py

@@ -0,0 +1,140 @@
+# This file is part of the MapProxy project.
+# Copyright (C) 2010 Omniscale <http://omniscale.de>
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from __future__ import division
+
+from io import BytesIO
+
+import pytest
+
+from mapproxy.request.wms import WMS111MapRequest
+from mapproxy.compat.image import Image
+from mapproxy.test.image import is_png, tmp_image
+from mapproxy.test.http import mock_httpd
+from mapproxy.test.system import SysTest
+
+
+@pytest.fixture(scope="module")
+def config_file():
+    return "coverage.yaml"
+
+
+class TestCoverageWMS(SysTest):
+
+    def setup(self):
+        self.common_map_req = WMS111MapRequest(
+            url="/service?",
+            param=dict(
+                service="WMS",
+                version="1.1.1",
+                bbox="-180,0,0,80",
+                width="200",
+                height="200",
+                layers="wms_cache",
+                srs="EPSG:4326",
+                format="image/png",
+                styles="",
+                request="GetMap",
+            ),
+        )
+
+    def test_capababilities(self, app):
+        resp = app.get("/service?request=GetCapabilities&service=WMS&version=1.1.1")
+        xml = resp.lxml
+        # First: combined root, second: wms_cache, third: tms_cache, last: seed_only
+        assert xml.xpath("//LatLonBoundingBox/@minx") == ["10", "10", "12", "14"]
+        assert xml.xpath("//LatLonBoundingBox/@miny") == ["10", "15", "10", "13"]
+        assert xml.xpath("//LatLonBoundingBox/@maxx") == ["35", "30", "35", "24"]
+        assert xml.xpath("//LatLonBoundingBox/@maxy") == ["31", "31", "30", "23"]
+
+    def test_get_map_outside(self, app):
+        self.common_map_req.params.bbox = -90, 0, 0, 90
+        self.common_map_req.params["bgcolor"] = "0xff0005"
+        resp = app.get(self.common_map_req)
+        assert resp.content_type == "image/png"
+        data = BytesIO(resp.body)
+        assert is_png(data)
+        img = Image.open(data)
+        assert img.mode == "RGB"
+        assert img.getcolors() == [(200 * 200, (255, 0, 5))]
+
+    def test_get_map_outside_transparent(self, app):
+        self.common_map_req.params.bbox = -90, 0, 0, 90
+        self.common_map_req.params.transparent = True
+        resp = app.get(self.common_map_req)
+        assert resp.content_type == "image/png"
+        data = BytesIO(resp.body)
+        assert is_png(data)
+        img = Image.open(data)
+        assert img.mode == "RGBA"
+        assert img.getcolors()[0][0] == 200 * 200
+        assert img.getcolors()[0][1][3] == 0  # transparent
+
+    def test_get_map_intersection(self, app, cache_dir):
+        with tmp_image((256, 256), format="jpeg") as img:
+            expected_req = (
+                {
+                    "path": r"/service?LAYERs=foo,bar&SERVICE=WMS&FORMAT=image%2Fjpeg"
+                    "&REQUEST=GetMap&HEIGHT=91&SRS=EPSG%3A4326&styles="
+                    "&VERSION=1.1.1&BBOX=10,15,30,31"
+                    "&WIDTH=114"
+                },
+                {"body": img.read(), "headers": {"content-type": "image/jpeg"}},
+            )
+            with mock_httpd(("localhost", 42423), [expected_req]):
+                self.common_map_req.params.bbox = 0, 0, 40, 40
+                self.common_map_req.params.transparent = True
+                resp = app.get(self.common_map_req)
+                assert resp.content_type == "image/png"
+                data = BytesIO(resp.body)
+                assert is_png(data)
+                assert Image.open(data).mode == "RGBA"
+        assert cache_dir.join(
+            "wms_cache_EPSG4326/03/000/000/004/000/000/002.jpeg"
+        ).check()
+
+
+class TestCoverageTMS(SysTest):
+
+    def test_get_tile_intersections(self, app, cache_dir):
+        with tmp_image((256, 256), format="jpeg") as img:
+            expected_req = (
+                {
+                    "path": r"/service?LAYERs=foo,bar&SERVICE=WMS&FORMAT=image%2Fjpeg"
+                    "&REQUEST=GetMap&HEIGHT=25&SRS=EPSG%3A900913&styles="
+                    "&VERSION=1.1.1&BBOX=1113194.90793,1689200.13961,3339584.7238,3632749.14338"
+                    "&WIDTH=28"
+                },
+                {"body": img.read(), "headers": {"content-type": "image/jpeg"}},
+            )
+            with mock_httpd(
+                ("localhost", 42423), [expected_req], bbox_aware_query_comparator=True
+            ):
+                resp = app.get("/tms/1.0.0/wms_cache/0/1/1.jpeg")
+                assert resp.content_type == "image/jpeg"
+        cache_dir.join("wms_cache_EPSG900913/01/000/000/001/000/000/001.jpeg").check()
+
+    def test_get_tile_intersection_tms(self, app, cache_dir):
+        with tmp_image((256, 256), format="jpeg") as img:
+            expected_req = (
+                {"path": r"/tms/1.0.0/foo/1/1/1.jpeg"},
+                {"body": img.read(), "headers": {"content-type": "image/jpeg"}},
+            )
+            with mock_httpd(
+                ("localhost", 42423), [expected_req], bbox_aware_query_comparator=True
+            ):
+                resp = app.get("/tms/1.0.0/tms_cache/0/1/1.jpeg")
+                assert resp.content_type == "image/jpeg"
+        cache_dir.join("tms_cache_EPSG900913/01/000/000/001/000/000/001.jpeg").check()

mapproxy/test/system/test_decorate_img.py

@@ -0,0 +1,214 @@
+# This file is part of the MapProxy project.
+# Copyright (C) 2011 Omniscale <http://omniscale.de>
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from io import BytesIO
+
+import pytest
+
+from mapproxy.compat.image import Image
+from mapproxy.image import ImageSource
+from mapproxy.request.wms import WMS111MapRequest
+from mapproxy.request.wmts import WMTS100TileRequest
+from mapproxy.test.image import is_png, is_jpeg
+from mapproxy.test.system import SysTest
+
+
+@pytest.fixture(scope="module")
+def config_file():
+    return "layer.yaml"
+
+
+def to_greyscale(image, service, layers, **kw):
+    img = image.as_image()
+    if hasattr(image.image_opts, "transparent") and image.image_opts.transparent:
+        img = img.convert("LA").convert("RGBA")
+    else:
+        img = img.convert("L").convert("RGB")
+    return ImageSource(img, image.image_opts)
+
+
+@pytest.mark.usefixtures("fixture_cache_data")
+class TestDecorateImg(SysTest):
+
+    def setup(self):
+        self.common_tile_req = WMTS100TileRequest(
+            url="/service?",
+            param=dict(
+                service="WMTS",
+                version="1.0.0",
+                tilerow="0",
+                tilecol="0",
+                tilematrix="01",
+                tilematrixset="GLOBAL_MERCATOR",
+                layer="wms_cache",
+                format="image/jpeg",
+                style="",
+                request="GetTile",
+            ),
+        )
+
+    def test_wms(self, app):
+        req = WMS111MapRequest(
+            url="/service?",
+            param=dict(
+                service="WMS",
+                version="1.1.1",
+                bbox="-180,0,0,80",
+                width="200",
+                height="200",
+                layers="wms_cache",
+                srs="EPSG:4326",
+                format="image/png",
+                styles="",
+                request="GetMap",
+            ),
+        )
+        resp = app.get(req, extra_environ={"mapproxy.decorate_img": to_greyscale})
+        data = BytesIO(resp.body)
+        assert is_png(data)
+        img = Image.open(data)
+        assert img.mode == "RGB"
+
+    def test_wms_transparent(self, app):
+        req = WMS111MapRequest(
+            url="/service?",
+            param=dict(
+                service="WMS",
+                version="1.1.1",
+                bbox="-180,0,0,80",
+                width="200",
+                height="200",
+                layers="wms_cache_transparent",
+                srs="EPSG:4326",
+                format="image/png",
+                styles="",
+                request="GetMap",
+                transparent="True",
+            ),
+        )
+        resp = app.get(req, extra_environ={"mapproxy.decorate_img": to_greyscale})
+        data = BytesIO(resp.body)
+        assert is_png(data)
+        img = Image.open(data)
+        assert img.mode == "RGBA"
+
+    def test_wms_bgcolor(self, app):
+        req = WMS111MapRequest(
+            url="/service?",
+            param=dict(
+                service="WMS",
+                version="1.1.1",
+                bbox="-180,0,0,80",
+                width="200",
+                height="200",
+                layers="wms_cache_transparent",
+                srs="EPSG:4326",
+                format="image/png",
+                styles="",
+                request="GetMap",
+                bgcolor="0xff00a0",
+            ),
+        )
+        resp = app.get(req, extra_environ={"mapproxy.decorate_img": to_greyscale})
+        data = BytesIO(resp.body)
+        assert is_png(data)
+        img = Image.open(data)
+        assert img.mode == "RGB"
+        assert sorted(img.getcolors())[-1][1] == (94, 94, 94)
+
+    def test_wms_args(self, app):
+        req = WMS111MapRequest(
+            url="/service?",
+            param=dict(
+                service="WMS",
+                version="1.1.1",
+                bbox="-180,0,0,80",
+                width="200",
+                height="200",
+                layers="wms_cache,wms_cache_transparent",
+                srs="EPSG:4326",
+                format="image/png",
+                styles="",
+                request="GetMap",
+                transparent="True",
+            ),
+        )
+
+        def callback(img_src, service, layers, environ, query_extent):
+            assert isinstance(img_src, ImageSource)
+            assert "wms.map" == service
+            assert len(layers) == 2
+            assert "wms_cache_transparent" in layers
+            assert "wms_cache" in layers
+            assert isinstance(environ, dict)
+            assert len(query_extent) == 2
+            assert len(query_extent[1]) == 4
+            assert query_extent[0] == "EPSG:4326"
+            return img_src
+
+        app.get(req, extra_environ={"mapproxy.decorate_img": callback})
+
+    def test_tms(self, app):
+        resp = app.get(
+            "/tms/1.0.0/wms_cache/0/0/1.jpeg",
+            extra_environ={"mapproxy.decorate_img": to_greyscale},
+        )
+        assert resp.content_type == "image/jpeg"
+        assert resp.content_length == len(resp.body)
+        data = BytesIO(resp.body)
+        assert is_jpeg(data)
+
+    def test_tms_args(self, app):
+
+        def callback(img_src, service, layers, environ, query_extent):
+            assert isinstance(img_src, ImageSource)
+            assert "tms" == service
+            assert "wms_cache" == layers[0]
+            assert isinstance(environ, dict)
+            assert len(query_extent) == 2
+            assert len(query_extent[1]) == 4
+            assert query_extent[0] == "EPSG:900913"
+            return img_src
+
+        app.get(
+            "/tms/1.0.0/wms_cache/0/0/1.jpeg",
+            extra_environ={"mapproxy.decorate_img": callback},
+        )
+
+    def test_wmts(self, app):
+        resp = app.get(
+            str(self.common_tile_req),
+            extra_environ={"mapproxy.decorate_img": to_greyscale},
+        )
+        assert resp.content_type == "image/jpeg"
+        assert resp.content_length == len(resp.body)
+        data = BytesIO(resp.body)
+        assert is_jpeg(data)
+
+    def test_wmts_args(self, app):
+
+        def callback(img_src, service, layers, environ, query_extent):
+            assert isinstance(img_src, ImageSource)
+            assert "wmts" == service
+            assert "wms_cache" == layers[0]
+            assert isinstance(environ, dict)
+            assert len(query_extent) == 2
+            assert len(query_extent[1]) == 4
+            assert query_extent[0] == "EPSG:900913"
+            return img_src
+
+        app.get(
+            str(self.common_tile_req), extra_environ={"mapproxy.decorate_img": callback}
+        )

mapproxy/test/system/test_demo.py

@@ -0,0 +1,106 @@
+# This file is part of the MapProxy project.
+# Copyright (C) 2022 Even Rouault
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import pytest
+
+from mapproxy.test.http import mock_httpd
+from mapproxy.test.system import SysTest
+from webtest import AppError
+
+
+@pytest.fixture(scope="module")
+def config_file():
+    return "demo.yaml"
+
+
+class TestDemo(SysTest):
+
+    def test_basic(self, app):
+        resp = app.get("/demo/", status=200)
+        assert resp.content_type == "text/html"
+        assert 'href="../service?REQUEST=GetCapabilities"' in resp
+        assert 'href="../service?REQUEST=GetCapabilities&tiled=true"' in resp
+        assert 'href="../demo/?wmts_layer=wms_cache&format=jpeg&srs=EPSG%3A900913"' in resp
+        assert 'href="../demo/?tms_layer=wms_cache&format=jpeg&srs=EPSG%3A900913"' in resp
+
+    def test_previewmap(self, app):
+        resp = app.get("/demo/?srs=EPSG%3A3857&format=image%2Fpng&wms_layer=wms_cache", status=200)
+        assert resp.content_type == "text/html"
+        assert '<h2>Layer Preview - wms_cache</h2>' in resp
+
+    def test_external(self, app):
+        expected_req = (
+            {
+                "path": r"/path/service?REQUEST=GetCapabilities&SERVICE=WMS"
+            },
+            {"body": b"test-string", "headers": {"content-type": "text/xml"}}
+        )
+        with mock_httpd(
+                ("localhost", 42423), [expected_req]
+        ):
+            resp = app.get('/demo/?wms_capabilities&type=external', extra_environ={
+                'HTTP_X_FORWARDED_HOST': 'localhost:42423/path'
+            })
+            content = resp.text
+            assert 'test-string' in content
+            assert 'http://localhost:42423/path/service?REQUEST=GetCapabilities&SERVICE=WMS' in content
+
+    def test_external_xss_injection(self, app):
+        expected_req = (
+            {
+                "path": r"/path/&gt;&lt;script&gt;alert(XSS)&lt;/script&gt;/service?REQUEST=GetCapabilities&SERVICE=WMS"
+            },
+            {"body": b"test-string", "headers": {"content-type": "text/xml"}}
+        )
+
+        with mock_httpd(
+                ("localhost", 42423), [expected_req]
+        ):
+            resp = app.get('/demo/?wms_capabilities&type=external', extra_environ={
+                'HTTP_X_FORWARDED_HOST': 'localhost:42423/path/"><script>alert(\'XSS\')</script>'
+            })
+        content = resp.text
+        assert 'test-string' in content
+        assert '"><script>alert(\'XSS\')' not in content
+        assert '&gt;&lt;script&gt;alert(XSS)&lt;/script&gt;' in content
+
+    def test_external_file_protocol(self, app):
+        try:
+            app.get('/demo/?wms_capabilities&type=external', extra_environ={
+                'HTTP_X_FORWARDED_PROTO': 'file'
+            })
+        except AppError as e:
+            assert '400 Bad Request' in e.args[0]
+
+    def test_tms_layer_xss(self, app):
+        expected_req = (
+            {
+                "path": r"/tms/1.0.0/osm&gt;&lt;script&gt;alert(XSS)&lt;/script&gt;/1.0.0"
+            },
+            {"body": b"test-string", "headers": {"content-type": "text/xml"}}
+        )
+
+        with mock_httpd(
+                ("localhost", 42423), [expected_req]
+        ):
+            resp = app.get(
+                '/demo/?tms_capabilities&layer=osm"><script>alert(\'XSS\')</script>&type=external&srs=1.0.0',
+                extra_environ={
+                    'HTTP_X_FORWARDED_HOST': 'localhost:42423'
+                }
+            )
+        content = resp.text
+        assert '"><script>alert(\'XSS\')' not in content
+        assert '&gt;&lt;script&gt;alert(XSS)&lt;/script&gt;' in content

mapproxy/test/system/test_demo_with_extra_service.py

@@ -0,0 +1,53 @@
+# This file is part of the MapProxy project.
+# Copyright (C) 2022 Even Rouault
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import pytest
+
+from mapproxy.service.demo import extra_demo_server_handlers, extra_substitution_handlers, register_extra_demo_server_handler, register_extra_demo_substitution_handler
+from mapproxy.test.system import SysTest
+
+def demo_server_handler(demo_server, req):
+    if 'my_service' in req.args:
+        return 'my_return'
+    return None
+
+
+def demo_substitution_handler(demo_server, req, substitutions):
+    substitutions['extra_services_html_beginning'] += '<h2>My extra service</h2>'
+
+@pytest.fixture(scope="module")
+def config_file():
+    register_extra_demo_server_handler(demo_server_handler)
+    register_extra_demo_substitution_handler(demo_substitution_handler)
+    yield "demo.yaml"
+    extra_demo_server_handlers.clear()
+    extra_substitution_handlers.clear()
+
+
+class TestDemoWithExtraService(SysTest):
+
+    def test_basic(self, app):
+        resp = app.get("/demo/", status=200)
+        assert resp.content_type == "text/html"
+        assert 'href="../service?REQUEST=GetCapabilities"' in resp
+        assert 'href="../service?REQUEST=GetCapabilities&tiled=true"' in resp
+        assert 'href="../demo/?wmts_layer=wms_cache&format=jpeg&srs=EPSG%3A900913"' in resp
+        assert 'href="../demo/?tms_layer=wms_cache&format=jpeg&srs=EPSG%3A900913"' in resp
+        assert '<h2>My extra service</h2>' in resp
+
+    def test_my_service(self, app):
+        resp = app.get("/demo/?my_service=bar", status=200)
+        assert resp.content_type == "text/html"
+        assert 'my_return' in resp