conftest update

Author: jonhealy1

stac_fastapi/tests/conftest.py

@@ -18,6 +18,11 @@
     CoreClient,
     TransactionsClient,
 )
+from stac_fastapi.core.extensions import QueryExtension
+from stac_fastapi.core.extensions.aggregation import (
+    EsAggregationExtensionGetRequest,
+    EsAggregationExtensionPostRequest,
+)
 from stac_fastapi.core.rate_limit import setup_rate_limit
 from stac_fastapi.core.utilities import get_bool_env
 from stac_fastapi.sfeos_helpers.aggregation import EsAsyncBaseAggregationClient
@@ -46,7 +51,15 @@
         create_index_templates,
     )
 
-from stac_fastapi.extensions.core import TransactionExtension
+from stac_fastapi.extensions.core import (
+    AggregationExtension,
+    FieldsExtension,
+    FilterExtension,
+    FreeTextExtension,
+    SortExtension,
+    TokenPaginationExtension,
+    TransactionExtension,
+)
 from stac_fastapi.types.config import Settings
 
 DATA_DIR = os.path.join(os.path.dirname(__file__), "data")
@@ -330,7 +343,7 @@ async def route_dependencies_client(route_dependencies_app):
 
 def build_test_app():
     """Build a test app with configurable transaction extensions."""
-    # Create a copy of the base config which already has all extensions configured
+    # Create a copy of the base config
     test_config = app_config.copy()
 
     # Get transaction extensions setting
@@ -359,23 +372,27 @@ def build_test_app():
 
     # Add transaction extension if enabled
     if TRANSACTIONS_EXTENSIONS:
-        settings = AsyncSettings()
-        test_config["extensions"].extend(
-            [
-                TransactionExtension(
-                    client=TransactionsClient(
-                        database=database, session=None, settings=settings
-                    ),
-                    settings=settings,
-                ),
-                BulkTransactionExtension(
-                    client=BulkTransactionsClient(
-                        database=database, session=None, settings=settings
-                    )
+        search_extensions.append(
+            TransactionExtension(
+                client=TransactionsClient(
+                    database=database, session=None, settings=settings
                 ),
-            ]
+                settings=settings,
+            )
         )
 
+    # Update extensions in config
+    extensions = [aggregation_extension] + search_extensions
+    test_config["extensions"] = extensions
+
+    # Update client with new extensions
+    test_config["client"] = CoreClient(
+        database=database,
+        session=None,
+        extensions=extensions,
+        post_request_model=test_config["search_post_request_model"],
+    )
+
     # Create and return the app
     api = StacApi(**test_config)
     return api.app