pre-commit.

Author: rhysrevans3

stac_fastapi/tests/conftest.py

@@ -130,7 +130,9 @@ def test_collection() -> Dict:
 
 
 async def create_collection(txn_client: TransactionsClient, collection: Dict) -> None:
-    await txn_client.create_collection(api.Collection(**dict(collection)), request=MockRequest, refresh=True)
+    await txn_client.create_collection(
+        api.Collection(**dict(collection)), request=MockRequest, refresh=True
+    )
 
 
 async def create_item(txn_client: TransactionsClient, item: Dict) -> None:
@@ -200,14 +202,18 @@ async def app():
     settings = AsyncSettings()
 
     aggregation_extension = AggregationExtension(
-        client=EsAsyncAggregationClient(database=database, session=None, settings=settings)
+        client=EsAsyncAggregationClient(
+            database=database, session=None, settings=settings
+        )
     )
     aggregation_extension.POST = EsAggregationExtensionPostRequest
     aggregation_extension.GET = EsAggregationExtensionGetRequest
 
     search_extensions = [
         TransactionExtension(
-            client=TransactionsClient(database=database, session=None, settings=settings),
+            client=TransactionsClient(
+                database=database, session=None, settings=settings
+            ),
             settings=settings,
         ),
         SortExtension(),
@@ -241,7 +247,9 @@ async def app_client(app):
     await create_index_templates()
     await create_collection_index()
 
-    async with AsyncClient(transport=ASGITransport(app=app), base_url="http://test-server") as c:
+    async with AsyncClient(
+        transport=ASGITransport(app=app), base_url="http://test-server"
+    ) as c:
         yield c
 
 
@@ -250,14 +258,18 @@ async def app_rate_limit():
     settings = AsyncSettings()
 
     aggregation_extension = AggregationExtension(
-        client=EsAsyncAggregationClient(database=database, session=None, settings=settings)
+        client=EsAsyncAggregationClient(
+            database=database, session=None, settings=settings
+        )
     )
     aggregation_extension.POST = EsAggregationExtensionPostRequest
     aggregation_extension.GET = EsAggregationExtensionGetRequest
 
     search_extensions = [
         TransactionExtension(
-            client=TransactionsClient(database=database, session=None, settings=settings),
+            client=TransactionsClient(
+                database=database, session=None, settings=settings
+            ),
             settings=settings,
         ),
         SortExtension(),
@@ -296,7 +308,9 @@ async def app_client_rate_limit(app_rate_limit):
     await create_index_templates()
     await create_collection_index()
 
-    async with AsyncClient(transport=ASGITransport(app=app_rate_limit), base_url="http://test-server") as c:
+    async with AsyncClient(
+        transport=ASGITransport(app=app_rate_limit), base_url="http://test-server"
+    ) as c:
         yield c
 
 
@@ -338,14 +352,18 @@ async def app_basic_auth():
     settings = AsyncSettings()
 
     aggregation_extension = AggregationExtension(
-        client=EsAsyncAggregationClient(database=database, session=None, settings=settings)
+        client=EsAsyncAggregationClient(
+            database=database, session=None, settings=settings
+        )
     )
     aggregation_extension.POST = EsAggregationExtensionPostRequest
     aggregation_extension.GET = EsAggregationExtensionGetRequest
 
     search_extensions = [
         TransactionExtension(
-            client=TransactionsClient(database=database, session=None, settings=settings),
+            client=TransactionsClient(
+                database=database, session=None, settings=settings
+            ),
             settings=settings,
         ),
         SortExtension(),
@@ -382,7 +400,9 @@ async def app_client_basic_auth(app_basic_auth):
     await create_index_templates()
     await create_collection_index()
 
-    async with AsyncClient(transport=ASGITransport(app=app_basic_auth), base_url="http://test-server") as c:
+    async with AsyncClient(
+        transport=ASGITransport(app=app_basic_auth), base_url="http://test-server"
+    ) as c:
         yield c
 
 
@@ -423,7 +443,9 @@ async def route_dependencies_app():
     settings = AsyncSettings()
     extensions = [
         TransactionExtension(
-            client=TransactionsClient(database=database, session=None, settings=settings),
+            client=TransactionsClient(
+                database=database, session=None, settings=settings
+            ),
             settings=settings,
         ),
         SortExtension(),
@@ -464,10 +486,14 @@ async def route_dependencies_client(route_dependencies_app):
 
 
 def build_test_app():
-    TRANSACTIONS_EXTENSIONS = get_bool_env("ENABLE_TRANSACTIONS_EXTENSIONS", default=True)
+    TRANSACTIONS_EXTENSIONS = get_bool_env(
+        "ENABLE_TRANSACTIONS_EXTENSIONS", default=True
+    )
     settings = AsyncSettings()
     aggregation_extension = AggregationExtension(
-        client=EsAsyncAggregationClient(database=database, session=None, settings=settings)
+        client=EsAsyncAggregationClient(
+            database=database, session=None, settings=settings
+        )
     )
     aggregation_extension.POST = EsAggregationExtensionPostRequest
     aggregation_extension.GET = EsAggregationExtensionGetRequest
@@ -483,7 +509,9 @@ def build_test_app():
         search_extensions.insert(
             0,
             TransactionExtension(
-                client=TransactionsClient(database=database, session=None, settings=settings),
+                client=TransactionsClient(
+                    database=database, session=None, settings=settings
+                ),
                 settings=settings,
             ),
         )