stac-fastapi.api 2.5.4__tar.gz → 2.5.5__tar.gz

{stac_fastapi_api-2.5.4 → stac_fastapi_api-2.5.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: stac-fastapi.api
-Version: 2.5.4
+Version: 2.5.5
 Summary: An implementation of STAC API based on the FastAPI framework.
 Home-page: https://github.com/stac-utils/stac-fastapi
 Author: Arturo Engineering

{stac_fastapi_api-2.5.4 → stac_fastapi_api-2.5.5}/stac_fastapi/api/version.py

@@ -1,2 +1,2 @@
 """Library version."""
-__version__ = "2.5.4"
+__version__ = "2.5.5"

{stac_fastapi_api-2.5.4 → stac_fastapi_api-2.5.5}/stac_fastapi.api.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: stac-fastapi.api
-Version: 2.5.4
+Version: 2.5.5
 Summary: An implementation of STAC API based on the FastAPI framework.
 Home-page: https://github.com/stac-utils/stac-fastapi
 Author: Arturo Engineering

{stac_fastapi_api-2.5.4 → stac_fastapi_api-2.5.5}/stac_fastapi.api.egg-info/SOURCES.txt

@@ -17,4 +17,5 @@ stac_fastapi/api/openapi.py
 stac_fastapi/api/routes.py
 stac_fastapi/api/version.py
 tests/test_api.py
+tests/test_app_prefix.py
 tests/test_middleware.py

stac_fastapi_api-2.5.5/tests/test_app_prefix.py

@@ -0,0 +1,66 @@
+import urllib
+from typing import Optional
+
+import pytest
+from fastapi import APIRouter
+from starlette.testclient import TestClient
+
+from stac_fastapi.api.app import StacApi
+from stac_fastapi.types.config import ApiSettings
+
+
+def get_link(landing_page, rel_type, method: Optional[str] = None):
+    return next(
+        filter(
+            lambda link: link["rel"] == rel_type
+            and (not method or link.get("method") == method),
+            landing_page["links"],
+        ),
+        None,
+    )
+
+
+@pytest.mark.parametrize("prefix", ["", "/a_prefix"])
+def test_api_prefix(TestCoreClient, prefix):
+    api_settings = ApiSettings(
+        openapi_url=f"{prefix}/api",
+        docs_url=f"{prefix}/api.html",
+    )
+
+    api = StacApi(
+        settings=api_settings,
+        client=TestCoreClient(),
+        router=APIRouter(prefix=prefix),
+    )
+
+    with TestClient(api.app, base_url="http://stac.io") as client:
+        landing = client.get(f"{prefix}/")
+        assert landing.status_code == 200, landing.json()
+
+        service_doc = client.get(f"{prefix}/api.html")
+        assert service_doc.status_code == 200, service_doc.text
+
+        service_desc = client.get(f"{prefix}/api")
+        assert service_desc.status_code == 200, service_desc.json()
+
+        conformance = client.get(f"{prefix}/conformance")
+        assert conformance.status_code == 200, conformance.json()
+
+        link_tests = [
+            ("root", "application/json", "/"),
+            ("conformance", "application/json", "/conformance"),
+            ("service-doc", "text/html", "/api.html"),
+            ("service-desc", "application/vnd.oai.openapi+json;version=3.0", "/api"),
+        ]
+
+        for rel_type, expected_media_type, expected_path in link_tests:
+            link = get_link(landing.json(), rel_type)
+
+            assert link is not None, f"Missing {rel_type} link in landing page"
+            assert link.get("type") == expected_media_type
+
+            link_path = urllib.parse.urlsplit(link.get("href")).path
+            assert link_path == prefix + expected_path
+
+            resp = client.get(prefix + expected_path)
+            assert resp.status_code == 200