chore: added additional tests for cube:dimensions extension and WMTS parameter override

Author: jverrydt

tests/fixtures/catalog.json

@@ -153,6 +153,157 @@
         "https://stac-extensions.github.io/item-assets/v1.0.0/schema.json",
         "https://stac-extensions.github.io/render/v1.0.0/schema.json"
       ]
+    },
+    {
+      "id": "MAXAR_BayofBengal_Cyclone_Mocha_May_23_cube_dimensions",
+      "type": "Collection",
+      "links": [
+        {
+          "rel": "items",
+          "type": "application/geo+json",
+          "href": "https://stac.endpoint.io/collections/MAXAR_BayofBengal_Cyclone_Mocha_May_23/items"
+        },
+        {
+          "rel": "parent",
+          "type": "application/json",
+          "href": "https://stac.endpoint.io/"
+        },
+        {
+          "rel": "root",
+          "type": "application/json",
+          "href": "https://stac.endpoint.io/"
+        },
+        {
+          "rel": "self",
+          "type": "application/json",
+          "href": "https://stac.endpoint.io/collections/MAXAR_BayofBengal_Cyclone_Mocha_May_23"
+        }
+      ],
+      "title": "Bay of Bengal Cyclone Mocha 2023",
+      "cube:dimensions": {
+        "time": {
+          "type": "temporal",
+          "extent": [
+            "2023-01-03T04:30:17Z",
+            "2023-05-22T04:35:25Z"
+          ],
+          "values": [
+            "2023-01-03T04:30:17Z",
+            "2023-02-03T04:30:17Z",
+            "2023-03-03T04:30:17Z",
+            "2023-04-03T04:30:17Z",
+            "2023-05-03T04:30:17Z",
+            "2023-05-22T04:35:25Z"
+          ]
+        }
+      },
+      "extent": {
+        "spatial": {
+          "bbox": [
+            [
+              91.831615,
+              19.982078842323997,
+              92.97426268500965,
+              21.666101
+            ],
+            [
+              92.567815,
+              20.18811887678192,
+              92.74417544237298,
+              20.62968532404085
+            ],
+            [
+              92.72278776887262,
+              20.104801,
+              92.893524,
+              20.630214
+            ],
+            [
+              92.75855246040959,
+              19.982078842323997,
+              92.89682495377032,
+              20.514473160464657
+            ],
+            [
+              92.84253515935835,
+              19.984656587012033,
+              92.97426268500965,
+              20.514418665444474
+            ],
+            [
+              91.831615,
+              21.518411,
+              91.957078,
+              21.666101
+            ]
+          ]
+        },
+        "temporal": {
+          "interval": [
+            [
+              "2023-01-03T04:30:17Z",
+              "2023-05-22T04:35:25Z"
+            ]
+          ]
+        }
+      },
+      "license": "CC-BY-NC-4.0",
+      "renders": {
+        "visual": {
+          "title": "Visual Image",
+          "assets": [
+            "visual"
+          ],
+          "asset_bidx": "visual|1,2,3",
+          "minmax_zoom": [
+            8,
+            22
+          ],
+          "tilematrixsets": {
+            "WebMercatorQuad": [
+                8,
+                22
+            ]
+          }
+        }
+      },
+      "description": "Maxar OpenData | Cyclone Mocha, a category five cyclone with 130 mph winds and torrential rain, hit parts of Myanmar and Bangladesh, forcing mass evacuations ahead of the storm. The cyclone, one of the most powerful to hit the region in the last decade, made landfall on Sunday, May 14, 2023, near Sittwe in Myanmar's Rakhine state. Rain and a storm surge caused widespread flooding in low-lying areas. The United National Office Coordination of Humanitarian Affairs stated that there had been extensive damage among already vulnerable communities and that communications with the affected areas have been difficult.",
+      "item_assets": {
+        "visual": {
+          "type": "image/tiff; application=geotiff; profile=cloud-optimized",
+          "roles": [
+            "visual"
+          ],
+          "title": "Visual Image"
+        },
+        "data-mask": {
+          "type": "application/geopackage+sqlite3",
+          "roles": [
+            "data-mask"
+          ],
+          "title": "Data Mask"
+        },
+        "ms_analytic": {
+          "type": "image/tiff; application=geotiff; profile=cloud-optimized",
+          "roles": [
+            "data"
+          ],
+          "title": "Multispectral Image"
+        },
+        "pan_analytic": {
+          "type": "image/tiff; application=geotiff; profile=cloud-optimized",
+          "roles": [
+            "data"
+          ],
+          "title": "Panchromatic Image"
+        }
+      },
+      "stac_version": "1.0.0",
+      "stac_extensions": [
+        "https://stac-extensions.github.io/item-assets/v1.0.0/schema.json",
+        "https://stac-extensions.github.io/render/v1.0.0/schema.json",
+        "https://stac-extensions.github.io/datacube/v2.2.0/schema.json"
+      ]
     }
   ],
   "links": [

tests/test_render.py

@@ -25,7 +25,7 @@ def test_render(client):
     collections_render = get_layer_from_collections(
         "https://something.stac", None, None
     )
-    assert len(collections_render) == 3
+    assert len(collections_render) == 4
 
     visual = collections_render["MAXAR_BayofBengal_Cyclone_Mocha_May_23_visual"]
     assert visual["bbox"]

tests/test_wmts.py

@@ -88,10 +88,11 @@ def test_wmts_getcapabilities(client, app):
     assert response.status_code == 200
     wmts = WebMapTileService(url="/wmts", xml=response.text.encode())
     layers = list(wmts.contents)
-    assert len(layers) == 3
+    assert len(layers) == 4
     assert "MAXAR_BayofBengal_Cyclone_Mocha_May_23_visual" in layers
     assert "MAXAR_BayofBengal_Cyclone_Mocha_May_23_color" in layers
     assert "MAXAR_BayofBengal_Cyclone_Mocha_May_23_visualr" in layers
+    assert "MAXAR_BayofBengal_Cyclone_Mocha_May_23_cube_dimensions_visual" in layers
 
     layer = wmts["MAXAR_BayofBengal_Cyclone_Mocha_May_23_visual"]
     assert "WebMercatorQuad" in layer.tilematrixsetlinks
@@ -104,6 +105,11 @@ def test_wmts_getcapabilities(client, app):
     assert query["assets"] == ["visual"]
     assert query["asset_bidx"] == ["visual|1,2,3"]
 
+    layer = wmts["MAXAR_BayofBengal_Cyclone_Mocha_May_23_cube_dimensions_visual"]
+    assert "TIME" in layer.dimensions
+    times = layer.dimensions["TIME"]["values"]
+    assert len(times) == 6
+
 
 @patch("rio_tiler.io.rasterio.rasterio")
 @patch("titiler.stacapi.factory.STACAPIBackend.get_assets")
@@ -287,6 +293,63 @@ def test_wmts_gettile(client, get_assets, rio, app):
     assert response.status_code == 200
 
 
+@patch("rio_tiler.io.rasterio.rasterio")
+@patch("titiler.stacapi.factory.STACAPIBackend.get_assets")
+@patch("titiler.stacapi.factory.Client")
+def test_wmts_gettile_param_override(client, get_assets, rio, app):
+    """test STAC items endpoints."""
+    rio.open = mock_rasterio_open
+
+    with open(catalog_json, "r") as f:
+        collections = [
+            pystac.Collection.from_dict(c) for c in json.loads(f.read())["collections"]
+        ]
+        client.open.return_value.get_collections.return_value = collections
+
+    with open(item_json, "r") as f:
+        get_assets.return_value = [json.loads(f.read())]
+
+    response = app.get(
+        "/wmts",
+        params={
+            "SERVICE": "WMTS",
+            "VERSION": "1.0.0",
+            "REQUEST": "getTile",
+            "LAYER": "MAXAR_BayofBengal_Cyclone_Mocha_May_23_visual",
+            "STYLE": "default",
+            "FORMAT": "image/png",
+            "TILEMATRIXSET": "WebMercatorQuad",
+            "TILEMATRIX": 14,
+            "TILEROW": 7188,
+            "TILECOL": 12375,
+            "TIME": "2023-01-05",
+            "expression": "(where(visual_invalid >= 0))",
+        },
+    )
+    assert response.status_code == 500
+    assert "Could not find any valid assets" in response.json()["detail"]
+
+    response = app.get(
+        "/wmts",
+        params={
+            "SERVICE": "WMTS",
+            "VERSION": "1.0.0",
+            "REQUEST": "getTile",
+            "LAYER": "MAXAR_BayofBengal_Cyclone_Mocha_May_23_color",
+            "STYLE": "default",
+            "FORMAT": "image/png",
+            "TILEMATRIXSET": "WebMercatorQuad",
+            "TILEMATRIX": 14,
+            "TILEROW": 7188,
+            "TILECOL": 12375,
+            "TIME": "2023-01-05",
+            "colormap": "{invalid}",
+        },
+    )
+    assert response.status_code == 400
+    assert "Could not parse the colormap value" in response.json()["detail"]
+
+
 @patch("rio_tiler.io.rasterio.rasterio")
 @patch("titiler.stacapi.factory.STACAPIBackend.get_assets")
 @patch("titiler.stacapi.factory.Client")