fix: add required links to item collection

sqlalchemy
stac-utils · Jan 13, 2023 · b69c221 · b69c221
1 parent 3bff044
commit b69c221
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/stac_fastapi/sqlalchemy/stac_fastapi/sqlalchemy/core.py b/stac_fastapi/sqlalchemy/stac_fastapi/sqlalchemy/core.py
@@ -169,7 +169,23 @@ def item_collection(
                 else None
             )
 
-            links = []
+            links = [
+                {
+                    "rel": Relations.self.value,
+                    "type": "application/geo+json",
+                    "href": str(kwargs["request"].url),
+                },
+                {
+                    "rel": Relations.root.value,
+                    "type": "application/json",
+                    "href": str(kwargs["request"].base_url),
+                },
+                {
+                    "rel": Relations.parent.value,
+                    "type": "application/json",
+                    "href": str(kwargs["request"].base_url),
+                },
+            ]
             if page.next:
                 links.append(
                     {

diff --git a/stac_fastapi/sqlalchemy/tests/conftest.py b/stac_fastapi/sqlalchemy/tests/conftest.py
@@ -70,6 +70,7 @@ def load_file(filename: str) -> Dict:
 
 class MockStarletteRequest:
     base_url = "http://test-server"
+    url = "http://test-server/some/endpoint"
 
 
 @pytest.fixture