Add test of the TiTiler API with data

Author: mfisher87

jupyter_xarray_tiler/tests/fixtures/api.py

@@ -0,0 +1,24 @@
+from collections.abc import AsyncGenerator
+
+import pytest_asyncio
+
+from jupyter_xarray_tiler.titiler import _get_server
+
+
+async def _reset_titiler_api_for_testing() -> None:
+    # Shutdown the previous server
+    server = _get_server()
+    await server.stop_tile_server()
+    # Clear the cache so next time we'll get a fresh one
+    _get_server.cache_clear()
+
+
+@pytest_asyncio.fixture
+async def clean_titiler_api() -> AsyncGenerator[None]:
+    """Ensure a test's usage of the titiler API is not influenced by other tests.
+
+    I.e., the test will receive a fresh TiTiler server.
+    """
+    await _reset_titiler_api_for_testing()
+    yield
+    await _reset_titiler_api_for_testing()

jupyter_xarray_tiler/tests/test_titiler_api.py

@@ -1,15 +1,32 @@
 import pytest
+from xarray import DataArray
 
-from jupyter_xarray_tiler.titiler import _get_server, get_routes
+from jupyter_xarray_tiler.titiler import (
+    _get_server,
+    add_data_array,
+    get_routes,
+)
 
 
+def test_singleton_ish() -> None:
+    """Test that the API only uses one TiTiler server instance."""
+    assert id(_get_server()) == id(_get_server())
+    assert _get_server() is _get_server()
+
+
+@pytest.mark.usefixtures("clean_titiler_api")
 def test_get_routes_raises_before_server_started() -> None:
     """Test that get_routes raises an error if called before initialization."""
     with pytest.raises(RuntimeError):
         get_routes()
 
 
-def test_singleton_ish() -> None:
-    """Test that the API only uses one TiTiler server instance."""
-    assert id(_get_server()) == id(_get_server())
-    assert _get_server() is _get_server()
+@pytest.mark.usefixtures("clean_titiler_api")
+async def test_add_data_array_creates_routes(
+    random_data_array: DataArray,
+) -> None:
+    await add_data_array(
+        data_array=random_data_array,
+        colormap_name="viridis",
+    )
+    assert len(get_routes()) > 0