wmts cookies test

Author: Asher Gross-Gellman

owslib/wmts.py

@@ -136,7 +136,7 @@ def __getitem__(self, name):
             raise KeyError("No content named %s" % name)
 
     def __init__(self, url, version='1.0.0', xml=None, username=None, password=None,
-                 parse_remote_metadata=False, vendor_kwargs=None, headers=None, cookies=None, auth=None,
+                 parse_remote_metadata=False, vendor_kwargs=None, headers=None, auth=None, cookies=None,
                  timeout=30):
         """Initialize.
 
@@ -180,7 +180,7 @@ def __init__(self, url, version='1.0.0', xml=None, username=None, password=None,
 
         # Authentication handled by Reader
         reader = WMTSCapabilitiesReader(
-            self.version, url=self.url, headers=self.headers, cookies=self.cookies, auth=self.auth)
+            self.version, url=self.url, headers=self.headers, auth=self.auth, cookies=self.cookies)
         if xml is not None:  # read from stored xml
             self._capabilities = reader.readString(xml)
         else:  # read from server
@@ -888,7 +888,7 @@ class WMTSCapabilitiesReader:
     """Read and parse capabilities document into a lxml.etree infoset
     """
 
-    def __init__(self, version='1.0.0', url=None, un=None, pw=None, headers=None, cookies=None, auth=None):
+    def __init__(self, version='1.0.0', url=None, un=None, pw=None, headers=None, auth=None, cookies=None):
         """Initialize"""
         self.version = version
         self._infoset = None

tests/test_wmts_cookies.py

@@ -0,0 +1,25 @@
+from tests.utils import scratch_file
+from tests.utils import service_ok
+
+from owslib.wmts import WebMapTileService
+
+import pytest
+
+SERVICE_URL_EXAMPLE = 'https://mapsneu.wien.gv.at/basemapneu/1.0.0/WMTSCapabilities.xml'
+
+
+@pytest.mark.online
+@pytest.mark.skipif(not service_ok(SERVICE_URL_EXAMPLE, timeout=20),
+                    reason="WMTS service is unreachable")
+def test_wmts_cookies():
+    from owslib.wmts import WebMapTileService
+    cookies = {
+        'cookie1': 'example1',
+        'cookie2': 'example2'
+    }
+    wmts = WebMapTileService(SERVICE_URL_EXAMPLE, cookies=cookies)
+    tile = wmts.gettile(layer="bmaporthofoto30cm", tilematrix="10", row=357, column=547)
+    tile_raw_cookies = tile._response.request.headers['Cookie']
+    tile_cookies = (dict(i.split('=',1) for i in tile_raw_cookies.split('; ')))
+    assert tile_cookies.get('cookie1') == 'example1'
+    assert tile_cookies.get('cookie2') == 'example2'