Ryan Malloy
d13ba744d3
Tests: 79 total (32 new), covering LRU cache eviction, BBox validation, _parse_rgb error handling, colormap ID extraction, client retry logic, WMS response validation, dimension clamping, geocode caching, classification colormaps, and scientific notation interval parsing. Shutdown: register atexit handler to close httpx connection pool when the MCP server exits, since FastMCP Middleware has no on_shutdown hook.
144 lines
5.4 KiB
Python
144 lines
5.4 KiB
Python
"""Tests for WMTS GetCapabilities XML parsing and layer search."""
|
|
|
|
from mcgibs.capabilities import (
|
|
_extract_colormap_id_from_href,
|
|
parse_capabilities,
|
|
search_layers,
|
|
)
|
|
|
|
|
|
def test_parse_capabilities_layer_count(capabilities_xml: str):
|
|
"""parse_capabilities should return exactly 3 layers from the sample XML."""
|
|
layers = parse_capabilities(capabilities_xml)
|
|
assert len(layers) == 3
|
|
|
|
|
|
def test_parse_capabilities_true_color_layer(capabilities_xml: str):
|
|
"""MODIS true color layer should have correct title, formats, and tile matrix sets."""
|
|
layers = parse_capabilities(capabilities_xml)
|
|
layer = layers["MODIS_Terra_CorrectedReflectance_TrueColor"]
|
|
|
|
assert layer.title == "Corrected Reflectance (True Color, Terra/MODIS)"
|
|
assert layer.formats == ["image/jpeg"]
|
|
assert layer.tile_matrix_sets == ["250m"]
|
|
|
|
|
|
def test_parse_capabilities_time_dimension(capabilities_xml: str):
|
|
"""True color layer time dimension should have correct start, end, period, and default."""
|
|
layers = parse_capabilities(capabilities_xml)
|
|
layer = layers["MODIS_Terra_CorrectedReflectance_TrueColor"]
|
|
|
|
assert layer.time is not None
|
|
assert layer.time.start == "2000-02-24"
|
|
assert layer.time.end == "2025-12-01"
|
|
assert layer.time.period == "P1D"
|
|
assert layer.time.default == "2025-12-01"
|
|
|
|
|
|
def test_parse_capabilities_bbox(capabilities_xml: str):
|
|
"""All sample layers have a global bounding box of -180,-90 to 180,90."""
|
|
layers = parse_capabilities(capabilities_xml)
|
|
layer = layers["MODIS_Terra_CorrectedReflectance_TrueColor"]
|
|
|
|
assert layer.bbox is not None
|
|
assert layer.bbox.west == -180.0
|
|
assert layer.bbox.south == -90.0
|
|
assert layer.bbox.east == 180.0
|
|
assert layer.bbox.north == 90.0
|
|
|
|
|
|
def test_parse_capabilities_colormap_detection(capabilities_xml: str):
|
|
"""AMSR2 layer should have a colormap; true color layer should not."""
|
|
layers = parse_capabilities(capabilities_xml)
|
|
|
|
amsr2 = layers["AMSR2_Sea_Ice_Concentration_12km_Monthly"]
|
|
assert amsr2.has_colormap is True
|
|
|
|
true_color = layers["MODIS_Terra_CorrectedReflectance_TrueColor"]
|
|
assert true_color.has_colormap is False
|
|
|
|
|
|
def test_parse_capabilities_legend_url(capabilities_xml: str):
|
|
"""AMSR2 layer should have a legend URL pointing to the PNG legend image."""
|
|
layers = parse_capabilities(capabilities_xml)
|
|
amsr2 = layers["AMSR2_Sea_Ice_Concentration_12km_Monthly"]
|
|
|
|
assert amsr2.legend_url == (
|
|
"https://gibs.earthdata.nasa.gov/legends/AMSR2_Sea_Ice_Concentration_12km.png"
|
|
)
|
|
|
|
|
|
def test_search_layers_keyword(capabilities_xml: str):
|
|
"""Searching for 'sea ice' should return the AMSR2 sea ice layer."""
|
|
index = parse_capabilities(capabilities_xml)
|
|
results = search_layers(index, "sea ice")
|
|
|
|
assert len(results) >= 1
|
|
identifiers = [r.identifier for r in results]
|
|
assert "AMSR2_Sea_Ice_Concentration_12km_Monthly" in identifiers
|
|
|
|
|
|
def test_search_layers_no_results(capabilities_xml: str):
|
|
"""Searching for a nonexistent term should return an empty list."""
|
|
index = parse_capabilities(capabilities_xml)
|
|
results = search_layers(index, "nonexistent")
|
|
|
|
assert results == []
|
|
|
|
|
|
def test_search_layers_limit(capabilities_xml: str):
|
|
"""Search with limit=1 should return at most 1 result."""
|
|
index = parse_capabilities(capabilities_xml)
|
|
# Use a broad query that could match multiple layers
|
|
results = search_layers(index, "temperature", limit=1)
|
|
|
|
assert len(results) <= 1
|
|
|
|
|
|
def test_parse_capabilities_resource_url(capabilities_xml: str):
|
|
"""True color layer should have the expected ResourceURL template."""
|
|
layers = parse_capabilities(capabilities_xml)
|
|
layer = layers["MODIS_Terra_CorrectedReflectance_TrueColor"]
|
|
|
|
expected = (
|
|
"https://gibs.earthdata.nasa.gov/wmts/epsg4326/best/"
|
|
"MODIS_Terra_CorrectedReflectance_TrueColor/default/"
|
|
"{Time}/{TileMatrixSet}/{TileMatrix}/{TileRow}/{TileCol}.jpg"
|
|
)
|
|
assert layer.resource_url_template == expected
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# _extract_colormap_id_from_href (live API colormap ID fix)
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
def test_extract_colormap_id_from_href_normal():
|
|
"""Extract colormap filename stem from a typical GIBS metadata URL."""
|
|
url = "https://gibs.earthdata.nasa.gov/colormaps/v1.3/AMSR2_Sea_Ice_Concentration.xml"
|
|
assert _extract_colormap_id_from_href(url) == "AMSR2_Sea_Ice_Concentration"
|
|
|
|
|
|
def test_extract_colormap_id_from_href_trailing_slash():
|
|
url = "https://gibs.earthdata.nasa.gov/colormaps/v1.3/Foo_Bar.xml/"
|
|
assert _extract_colormap_id_from_href(url) == "Foo_Bar"
|
|
|
|
|
|
def test_extract_colormap_id_from_href_no_xml():
|
|
url = "https://gibs.earthdata.nasa.gov/colormaps/v1.3/NoExtension"
|
|
assert _extract_colormap_id_from_href(url) is None
|
|
|
|
|
|
def test_extract_colormap_id_from_href_empty():
|
|
assert _extract_colormap_id_from_href("") is None
|
|
|
|
|
|
def test_colormap_id_differs_from_layer_id(capabilities_xml: str):
|
|
"""AMSR2 layer should have a colormap_id extracted from the metadata href,
|
|
which may differ from the layer identifier itself."""
|
|
layers = parse_capabilities(capabilities_xml)
|
|
amsr2 = layers["AMSR2_Sea_Ice_Concentration_12km_Monthly"]
|
|
# The fixture's metadata href contains the actual colormap filename
|
|
assert amsr2.has_colormap is True
|
|
assert amsr2.colormap_id is not None
|