MapProxy 2.1.0__py3-none-any.whl

mapproxy/test/unit/test_client_cgi.py

@@ -0,0 +1,140 @@
+# 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.
+
+import sys
+import os
+import shutil
+import stat
+import tempfile
+
+import pytest
+
+from mapproxy.client.cgi import CGIClient, split_cgi_response
+from mapproxy.client.http import HTTPClientError
+from mapproxy.source import SourceError
+
+
+@pytest.mark.skipif(sys.version_info < (3, 0), reason="tests skipped for python 2")
+class TestSplitHTTPResponse(object):
+    def test_n(self):
+        assert split_cgi_response(b'header1: foo\nheader2: bar\n\ncontent\n\ncontent') == \
+            ({'Header1': 'foo', 'Header2': 'bar'}, b'content\n\ncontent')
+
+    def test_rn(self):
+        assert split_cgi_response(b'header1\r\nheader2\r\n\r\ncontent\r\n\r\ncontent') == \
+            ({'Header1': None, 'Header2': None}, b'content\r\n\r\ncontent')
+
+    def test_mixed(self):
+        assert split_cgi_response(b'header1: bar:foo\r\nheader2\n\r\ncontent\r\n\r\ncontent') == \
+            ({'Header1': 'bar:foo', 'Header2': None}, b'content\r\n\r\ncontent')
+        assert split_cgi_response(b'header1\r\nheader2\n\ncontent\r\n\r\ncontent') == \
+            ({'Header1': None, 'Header2': None}, b'content\r\n\r\ncontent')
+        assert split_cgi_response(b'header1\nheader2\r\n\r\ncontent\r\n\r\ncontent') == \
+            ({'Header1': None, 'Header2': None}, b'content\r\n\r\ncontent')
+
+    def test_no_header(self):
+        assert split_cgi_response(b'content\r\ncontent') == \
+            ({}, b'content\r\ncontent')
+
+
+TEST_CGI_SCRIPT = br"""#! /usr/bin/env python3
+import sys
+import os
+w = sys.stdout.write
+w("Content-type: text/plain\r\n")
+w("\r\n")
+w(os.environ['QUERY_STRING'])
+"""
+
+TEST_CGI_SCRIPT_FAIL = TEST_CGI_SCRIPT + b'\nexit(1)'
+
+TEST_CGI_SCRIPT_CWD = TEST_CGI_SCRIPT + br"""
+if not os.path.exists('testfile'):
+    exit(2)
+"""
+
+
+@pytest.mark.skipif(sys.platform == 'win32', reason="tests not ported to windows")
+@pytest.mark.skipif(sys.version_info < (3, 0), reason="tests skipped for python 2")
+class TestCGIClient(object):
+    def setup_method(self):
+        self.script_dir = tempfile.mkdtemp()
+
+    def teardown_method(self):
+        shutil.rmtree(self.script_dir)
+
+    def create_script(self, script=TEST_CGI_SCRIPT, executable=True):
+        script_file = os.path.join(self.script_dir, 'minimal_cgi.py')
+        with open(script_file, 'wb') as f:
+            f.write(script)
+        if executable:
+            os.chmod(script_file, stat.S_IXUSR | stat.S_IRUSR | stat.S_IWUSR)
+        return script_file
+
+    def test_missing_script(self):
+        client = CGIClient('/tmp/doesnotexist')
+        try:
+            client.open('http://example.org/service?hello=bar')
+        except SourceError:
+            pass
+        else:
+            assert False, 'expected SourceError'
+
+    def test_script_not_executable(self):
+        script = self.create_script(executable=False)
+        client = CGIClient(script)
+        try:
+            client.open('http://example.org/service?hello=bar')
+        except SourceError:
+            pass
+        else:
+            assert False, 'expected SourceError'
+
+    def test_call(self):
+        script = self.create_script()
+        client = CGIClient(script)
+        resp = client.open('http://example.org/service?hello=bar')
+        assert resp.headers['Content-type'] == 'text/plain'
+        assert resp.read() == b'hello=bar'
+
+    def test_failed_call(self):
+        script = self.create_script(TEST_CGI_SCRIPT_FAIL)
+        client = CGIClient(script)
+        try:
+            client.open('http://example.org/service?hello=bar')
+        except HTTPClientError:
+            pass
+        else:
+            assert False, 'expected HTTPClientError'
+
+    def test_working_directory(self):
+        tmp_work_dir = os.path.join(self.script_dir, 'tmp')
+        os.mkdir(tmp_work_dir)
+        tmp_file = os.path.join(tmp_work_dir, 'testfile')
+        open(tmp_file, 'wb')
+
+        # start script in default directory
+        script = self.create_script(TEST_CGI_SCRIPT_CWD)
+        client = CGIClient(script)
+        try:
+            client.open('http://example.org/service?hello=bar')
+        except HTTPClientError:
+            pass
+        else:
+            assert False, 'expected HTTPClientError'
+
+        # start in tmp_work_dir
+        client = CGIClient(script, working_directory=tmp_work_dir)
+        client.open('http://example.org/service?hello=bar')

mapproxy/test/unit/test_collections.py

@@ -0,0 +1,116 @@
+# 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.
+
+import pytest
+
+from mapproxy.util.collections import LRU, ImmutableDictList
+
+
+class TestLRU(object):
+    def test_missing_key(self):
+        lru = LRU(10)
+        with pytest.raises(KeyError):
+            lru['foo']
+
+    def test_contains(self):
+        lru = LRU(10)
+        lru['foo1'] = 1
+
+        assert 'foo1' in lru
+        assert 'foo2' not in lru
+
+    def test_repr(self):
+        lru = LRU(10)
+        lru['foo1'] = 1
+        assert 'size=10' in repr(lru)
+        assert 'foo1' in repr(lru)
+
+    def test_getitem(self):
+        lru = LRU(10)
+        lru['foo1'] = 1
+        lru['foo2'] = 2
+        assert lru['foo1'] == 1
+        assert lru['foo2'] == 2
+
+    def test_get(self):
+        lru = LRU(10)
+        lru['foo1'] = 1
+        assert lru.get('foo1') == 1
+        assert lru.get('foo1', 2) == 1
+
+    def test_get_default(self):
+        lru = LRU(10)
+        lru['foo1'] = 1
+        assert lru.get('foo2') is None
+        assert lru.get('foo2', 2) == 2
+
+    def test_delitem(self):
+        lru = LRU(10)
+        lru['foo1'] = 1
+        assert 'foo1' in lru
+        del lru['foo1']
+        assert 'foo1' not in lru
+
+    def test_empty(self):
+        lru = LRU(10)
+        assert bool(lru) is False
+        lru['foo1'] = '1'
+        assert bool(lru) is True
+
+    def test_setitem_overflow(self):
+        lru = LRU(2)
+        lru['foo1'] = 1
+        lru['foo2'] = 2
+        lru['foo3'] = 3
+
+        assert 'foo1' not in lru
+        assert 'foo2' in lru
+        assert 'foo3' in lru
+
+    def test_length(self):
+        lru = LRU(2)
+        assert len(lru) == 0
+        lru['foo1'] = 1
+        assert len(lru) == 1
+        lru['foo2'] = 2
+        assert len(lru) == 2
+        lru['foo3'] = 3
+        assert len(lru) == 2
+
+        del lru['foo3']
+        assert len(lru) == 1
+
+
+class TestImmutableDictList(object):
+    def test_named(self):
+        res = ImmutableDictList([('one', 10), ('two', 5), ('three', 3)])
+        assert res[0] == 10
+        assert res[2] == 3
+        assert res['one'] == 10
+        assert res['three'] == 3
+        assert len(res) == 3
+
+    def test_named_iteritems(self):
+        res = ImmutableDictList([('one', 10), ('two', 5), ('three', 3)])
+        itr = res.iteritems()
+        assert next(itr) == ('one', 10)
+        assert next(itr) == ('two', 5)
+        assert next(itr) == ('three', 3)
+        try:
+            next(itr)
+        except StopIteration:
+            pass
+        else:
+            assert False, 'StopIteration expected'

mapproxy/test/unit/test_concat_legends.py

@@ -0,0 +1,37 @@
+# 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 mapproxy.compat.image import Image
+from mapproxy.image import ImageSource
+from mapproxy.image.merge import concat_legends
+from mapproxy.test.image import is_png
+
+
+class Test_Concat_Legends(object):
+    def test_concatenation(self):
+        legends = []
+        img_1 = Image.new(mode='RGBA', size=(30, 10), color="red")
+        img_2 = Image.new(mode='RGBA', size=(10, 10), color="black")
+        img_3 = Image.new(mode='RGBA', size=(50, 80), color="blue")
+        legends.append(ImageSource(img_1))
+        legends.append(ImageSource(img_2))
+        legends.append(ImageSource(img_3))
+        source = concat_legends(legends)
+        src_img = source.as_image()
+        assert src_img.getpixel((0, 90)) == (255, 0, 0, 255)
+        assert src_img.getpixel((0, 80)) == (0, 0, 0, 255)
+        assert src_img.getpixel((0, 0)) == (0, 0, 255, 255)
+        assert src_img.getpixel((49, 99)) == (255, 255, 255, 0)
+        assert is_png(source.as_buffer())