stac-fastapi.extensions 2.5.2__tar.gz → 2.5.4__tar.gz

{stac_fastapi_extensions-2.5.2 → stac_fastapi_extensions-2.5.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: stac-fastapi.extensions
-Version: 2.5.2
+Version: 2.5.4
 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_extensions-2.5.2 → stac_fastapi_extensions-2.5.4}/stac_fastapi/extensions/core/transaction.py

@@ -31,6 +31,13 @@ class PutItem(ItemUri):
     item: stac_types.Item = attr.ib(default=Body(None))
 
 
+@attr.s
+class PutCollection(CollectionUri):
+    """Update Collection."""
+
+    collection: stac_types.Collection = attr.ib(default=Body(None))
+
+
 @attr.s
 class TransactionExtension(ApiExtension):
     """Transaction Extension.
@@ -128,7 +135,7 @@ class TransactionExtension(ApiExtension):
             response_model_exclude_none=True,
             methods=["PUT"],
             endpoint=create_async_endpoint(
-                self.client.update_collection, stac_types.Collection
+                self.client.update_collection, PutCollection
             ),
         )
 

{stac_fastapi_extensions-2.5.2 → stac_fastapi_extensions-2.5.4}/stac_fastapi/extensions/version.py

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

{stac_fastapi_extensions-2.5.2 → stac_fastapi_extensions-2.5.4}/stac_fastapi.extensions.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: stac-fastapi.extensions
-Version: 2.5.2
+Version: 2.5.4
 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_extensions-2.5.4/tests/test_transaction.py

@@ -0,0 +1,180 @@
+import json
+from typing import Iterator, Union
+
+import pytest
+from starlette.testclient import TestClient
+
+from stac_fastapi.api.app import StacApi
+from stac_fastapi.extensions.core import TransactionExtension
+from stac_fastapi.types.config import ApiSettings
+from stac_fastapi.types.core import BaseCoreClient, BaseTransactionsClient
+from stac_fastapi.types.stac import Collection, Item, ItemCollection
+
+
+class DummyCoreClient(BaseCoreClient):
+    def all_collections(self, *args, **kwargs):
+        raise NotImplementedError
+
+    def get_collection(self, *args, **kwargs):
+        raise NotImplementedError
+
+    def get_item(self, *args, **kwargs):
+        raise NotImplementedError
+
+    def get_search(self, *args, **kwargs):
+        raise NotImplementedError
+
+    def post_search(self, *args, **kwargs):
+        raise NotImplementedError
+
+    def item_collection(self, *args, **kwargs):
+        raise NotImplementedError
+
+
+class DummyTransactionsClient(BaseTransactionsClient):
+    """Dummy client returning parts of the request, rather than proper STAC items."""
+
+    def create_item(self, item: Union[Item, ItemCollection], **kwargs):
+        return {"type": item["type"]}
+
+    def update_item(self, collection_id: str, item_id: str, item: Item, **kwargs):
+        return {
+            "path_collection_id": collection_id,
+            "path_item_id": item_id,
+            "type": item["type"],
+        }
+
+    def delete_item(self, item_id: str, collection_id: str, **kwargs):
+        return {
+            "path_collection_id": collection_id,
+            "path_item_id": item_id,
+        }
+
+    def create_collection(self, collection: Collection, **kwargs):
+        return {"type": collection["type"]}
+
+    def update_collection(self, collection_id: str, collection: Collection, **kwargs):
+        return {"path_collection_id": collection_id, "type": collection["type"]}
+
+    def delete_collection(self, collection_id: str, **kwargs):
+        return {"path_collection_id": collection_id}
+
+
+def test_create_item(client: TestClient, item: Item) -> None:
+    response = client.post("/collections/a-collection/items", content=json.dumps(item))
+    assert response.is_success, response.text
+    assert response.json()["type"] == "Feature"
+
+
+def test_create_item_collection(
+    client: TestClient, item_collection: ItemCollection
+) -> None:
+    response = client.post(
+        "/collections/a-collection/items", content=json.dumps(item_collection)
+    )
+    assert response.is_success, response.text
+    assert response.json()["type"] == "FeatureCollection"
+
+
+def test_update_item(client: TestClient, item: Item) -> None:
+    response = client.put(
+        "/collections/a-collection/items/an-item", content=json.dumps(item)
+    )
+    assert response.is_success, response.text
+    assert response.json()["path_collection_id"] == "a-collection"
+    assert response.json()["path_item_id"] == "an-item"
+    assert response.json()["type"] == "Feature"
+
+
+def test_delete_item(client: TestClient) -> None:
+    response = client.delete("/collections/a-collection/items/an-item")
+    assert response.is_success, response.text
+    assert response.json()["path_collection_id"] == "a-collection"
+    assert response.json()["path_item_id"] == "an-item"
+
+
+def test_create_collection(client: TestClient, collection: Collection) -> None:
+    response = client.post("/collections", content=json.dumps(collection))
+    assert response.is_success, response.text
+    assert response.json()["type"] == "Collection"
+
+
+def test_update_collection(client: TestClient, collection: Collection) -> None:
+    response = client.put("/collections/a-collection", content=json.dumps(collection))
+    assert response.is_success, response.text
+    assert response.json()["path_collection_id"] == "a-collection"
+    assert response.json()["type"] == "Collection"
+
+
+def test_delete_collection(client: TestClient, collection: Collection) -> None:
+    response = client.delete("/collections/a-collection")
+    assert response.is_success, response.text
+    assert response.json()["path_collection_id"] == "a-collection"
+
+
+@pytest.fixture
+def client(
+    core_client: DummyCoreClient, transactions_client: DummyTransactionsClient
+) -> Iterator[TestClient]:
+    settings = ApiSettings()
+    api = StacApi(
+        settings=settings,
+        client=core_client,
+        extensions=[
+            TransactionExtension(client=transactions_client, settings=settings),
+        ],
+    )
+    with TestClient(api.app) as client:
+        yield client
+
+
+@pytest.fixture
+def core_client() -> DummyCoreClient:
+    return DummyCoreClient()
+
+
+@pytest.fixture
+def transactions_client() -> DummyTransactionsClient:
+    return DummyTransactionsClient()
+
+
+@pytest.fixture
+def item_collection(item: Item) -> ItemCollection:
+    return {
+        "type": "FeatureCollection",
+        "features": [item],
+        "links": [],
+        "context": None,
+    }
+
+
+@pytest.fixture
+def item() -> Item:
+    return {
+        "type": "Feature",
+        "stac_version": "1.0.0",
+        "stac_extensions": [],
+        "id": "test_item",
+        "geometry": {"type": "Point", "coordinates": [-105, 40]},
+        "bbox": [-105, 40, -105, 40],
+        "properties": {},
+        "links": [],
+        "assets": {},
+        "collection": "test_collection",
+    }
+
+
+@pytest.fixture
+def collection() -> Collection:
+    return {
+        "type": "Collection",
+        "stac_version": "1.0.0",
+        "stac_extensions": [],
+        "id": "test_collection",
+        "extent": {
+            "spatial": {"bbox": [[-180, -90, 180, 90]]},
+            "temporal": {
+                "interval": [["2000-01-01T00:00:00Z", "2024-01-01T00:00:00Z"]]
+            },
+        },
+    }

stac_fastapi_extensions-2.5.2/tests/test_transaction.py

@@ -1,121 +0,0 @@
-import json
-from typing import Iterator, Union
-
-import pytest
-from starlette.testclient import TestClient
-
-from stac_fastapi.api.app import StacApi
-from stac_fastapi.extensions.core import TransactionExtension
-from stac_fastapi.types.config import ApiSettings
-from stac_fastapi.types.core import BaseCoreClient, BaseTransactionsClient
-from stac_fastapi.types.stac import Item, ItemCollection
-
-
-class DummyCoreClient(BaseCoreClient):
-    def all_collections(self, *args, **kwargs):
-        raise NotImplementedError
-
-    def get_collection(self, *args, **kwargs):
-        raise NotImplementedError
-
-    def get_item(self, *args, **kwargs):
-        raise NotImplementedError
-
-    def get_search(self, *args, **kwargs):
-        raise NotImplementedError
-
-    def post_search(self, *args, **kwargs):
-        raise NotImplementedError
-
-    def item_collection(self, *args, **kwargs):
-        raise NotImplementedError
-
-
-class DummyTransactionsClient(BaseTransactionsClient):
-    """Defines a pattern for implementing the STAC transaction extension."""
-
-    def create_item(self, item: Union[Item, ItemCollection], *args, **kwargs):
-        return {"created": True, "type": item["type"]}
-
-    def update_item(self, *args, **kwargs):
-        raise NotImplementedError
-
-    def delete_item(self, *args, **kwargs):
-        raise NotImplementedError
-
-    def create_collection(self, *args, **kwargs):
-        raise NotImplementedError
-
-    def update_collection(self, *args, **kwargs):
-        raise NotImplementedError
-
-    def delete_collection(self, *args, **kwargs):
-        raise NotImplementedError
-
-
-def test_create_item(client: TestClient, item: Item) -> None:
-    response = client.post("/collections/a-collection/items", content=json.dumps(item))
-    assert response.is_success, response.text
-    assert response.json()["type"] == "Feature"
-
-
-def test_create_item_collection(
-    client: TestClient, item_collection: ItemCollection
-) -> None:
-    response = client.post(
-        "/collections/a-collection/items", content=json.dumps(item_collection)
-    )
-    assert response.is_success, response.text
-    assert response.json()["type"] == "FeatureCollection"
-
-
-@pytest.fixture
-def client(
-    core_client: DummyCoreClient, transactions_client: DummyTransactionsClient
-) -> Iterator[TestClient]:
-    settings = ApiSettings()
-    api = StacApi(
-        settings=settings,
-        client=core_client,
-        extensions=[
-            TransactionExtension(client=transactions_client, settings=settings),
-        ],
-    )
-    with TestClient(api.app) as client:
-        yield client
-
-
-@pytest.fixture
-def core_client() -> DummyCoreClient:
-    return DummyCoreClient()
-
-
-@pytest.fixture
-def transactions_client() -> DummyTransactionsClient:
-    return DummyTransactionsClient()
-
-
-@pytest.fixture
-def item_collection(item: Item) -> ItemCollection:
-    return {
-        "type": "FeatureCollection",
-        "features": [item],
-        "links": [],
-        "context": None,
-    }
-
-
-@pytest.fixture
-def item() -> Item:
-    return {
-        "type": "Feature",
-        "stac_version": "1.0.0",
-        "stac_extensions": [],
-        "id": "test_item",
-        "geometry": {"type": "Point", "coordinates": [-105, 40]},
-        "bbox": [-105, 40, -105, 40],
-        "properties": {},
-        "links": [],
-        "assets": {},
-        "collection": "test_collection",
-    }