Fixing test.

Author: rhysrevans3

stac_fastapi/extensions/stac_fastapi/extensions/core/transaction.py

@@ -1,7 +1,7 @@
 """Transaction extension."""
 
 from enum import Enum
-from typing import List, Literal, Optional, Type, Union
+from typing import Any, List, Optional, Type, Union
 
 import attr
 from fastapi import APIRouter, Body, FastAPI, Header
@@ -15,7 +15,6 @@
 from stac_fastapi.types.config import ApiSettings
 from stac_fastapi.types.core import AsyncBaseTransactionsClient, BaseTransactionsClient
 from stac_fastapi.types.extension import ApiExtension
-from stac_fastapi.types.stac import PartialCollection, PartialItem, PatchOperation
 
 
 class TransactionConformanceClasses(str, Enum):
@@ -48,17 +47,11 @@ class PatchItem(ItemUri):
     """Patch Item."""
 
     patch: Annotated[
-        Union[PartialItem, List[PatchOperation]],
+        Any,
         Body(),
     ] = attr.ib(default=None)
     content_type: Annotated[
-        Optional[
-            Literal[
-                "application/json-patch+json",
-                "application/merge-patch+json",
-                "application/json",
-            ]
-        ],
+        Optional[str],
         Header(),
     ] = attr.ib(default="application/json")
 
@@ -75,17 +68,11 @@ class PatchCollection(CollectionUri):
     """Patch Collection."""
 
     patch: Annotated[
-        Union[PartialCollection, List[PatchOperation]],
+        Any,
         Body(),
     ] = attr.ib(default=None)
     content_type: Annotated[
-        Optional[
-            Literal[
-                "application/json-patch+json",
-                "application/merge-patch+json",
-                "application/json",
-            ]
-        ],
+        Optional[str],
         Header(),
     ] = attr.ib(default="application/json")
 

stac_fastapi/extensions/tests/test_transaction.py

@@ -119,7 +119,7 @@ def test_patch_item(client: TestClient) -> None:
     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()["patch"] == '{"patch": true}'
+    assert response.json()["patch"] == {"patch": True}
 
 
 def test_delete_item(client: TestClient) -> None:
@@ -149,7 +149,7 @@ def test_patch_collection(client: TestClient) -> None:
     )
     assert response.is_success, response.text
     assert response.json()["path_collection_id"] == "a-collection"
-    assert response.json()["patch"] == '{"patch": true}'
+    assert response.json()["patch"] == {"patch": True}
 
 
 def test_delete_collection(client: TestClient, collection: Collection) -> None:

stac_fastapi/types/stac_fastapi/types/core.py

@@ -1,11 +1,11 @@
 """Base clients."""
 
 import abc
-from typing import Annotated, Any, Dict, List, Optional, Union
+from typing import Any, Dict, List, Optional, Union
 from urllib.parse import urljoin
 
 import attr
-from fastapi import Header, Request
+from fastapi import Request
 from geojson_pydantic.geometries import Geometry
 from stac_pydantic import Collection, Item, ItemCollection
 from stac_pydantic.links import Relations
@@ -86,7 +86,7 @@ def patch_item(
         collection_id: str,
         item_id: str,
         patch: Any,
-        content_type: Annotated[Optional[str], Header()] = None,
+        content_type: Optional[str] = None,
         **kwargs,
     ) -> Optional[Union[stac.Item, Response]]:
         """Update an item from a collection.
@@ -161,7 +161,7 @@ def patch_collection(
         self,
         collection_id: str,
         patch: Any,
-        content_type: Annotated[Optional[str], Header()] = None,
+        content_type: Optional[str] = None,
         **kwargs,
     ) -> Optional[Union[stac.Collection, Response]]:
         """Update a collection.
@@ -243,7 +243,7 @@ async def patch_item(
         collection_id: str,
         item_id: str,
         patch: Any,
-        content_type: Annotated[Optional[str], Header()] = None,
+        content_type: Optional[str] = None,
         **kwargs,
     ) -> Optional[Union[stac.Item, Response]]:
         """Update an item from a collection.
@@ -318,7 +318,7 @@ async def patch_collection(
         self,
         collection_id: str,
         patch: Any,
-        content_type: Annotated[Optional[str], Header()] = None,
+        content_type: Optional[str] = None,
         **kwargs,
     ) -> Optional[Union[stac.Collection, Response]]:
         """Update a collection.