pre-commit.

Author: rhysrevans3

stac_fastapi/tests/api/test_api.py

@@ -64,7 +64,9 @@ async def test_get_search_content_type(app_client, ctx):
 @pytest.mark.asyncio
 async def test_api_headers(app_client):
     resp = await app_client.get("/api")
-    assert resp.headers["Content-Type"] == "application/vnd.oai.openapi+json;version=3.0"
+    assert (
+        resp.headers["Content-Type"] == "application/vnd.oai.openapi+json;version=3.0"
+    )
     assert resp.status_code == 200
 
 
@@ -107,7 +109,9 @@ async def test_app_context_results(app_client, txn_client, ctx, load_test_data):
     await create_collection(txn_client, test_collection)
     await create_item(txn_client, test_item)
 
-    resp = await app_client.get(f"/collections/{test_collection['id']}/items/{test_item['id']}")
+    resp = await app_client.get(
+        f"/collections/{test_collection['id']}/items/{test_item['id']}"
+    )
     assert resp.status_code == 200
     resp_json = resp.json()
     assert resp_json["id"] == test_item["id"]
@@ -157,7 +161,9 @@ async def test_app_fields_extension_query(app_client, ctx, txn_client):
 
 @pytest.mark.asyncio
 async def test_app_fields_extension_no_properties_get(app_client, ctx, txn_client):
-    resp = await app_client.get("/search", params={"collections": ["test-collection"], "fields": "-properties"})
+    resp = await app_client.get(
+        "/search", params={"collections": ["test-collection"], "fields": "-properties"}
+    )
     assert resp.status_code == 200
     resp_json = resp.json()
     assert "properties" not in resp_json["features"][0]
@@ -188,13 +194,20 @@ async def test_app_fields_extension_no_null_fields(app_client, ctx, txn_client):
         for link in feature["links"]:
             assert all(a not in link or link[a] is not None for a in ("title", "asset"))
         for asset in feature["assets"]:
-            assert all(a not in asset or asset[a] is not None for a in ("start_datetime", "created"))
+            assert all(
+                a not in asset or asset[a] is not None
+                for a in ("start_datetime", "created")
+            )
 
 
 @pytest.mark.asyncio
-async def test_app_fields_extension_return_all_properties(app_client, ctx, txn_client, load_test_data):
+async def test_app_fields_extension_return_all_properties(
+    app_client, ctx, txn_client, load_test_data
+):
     item = load_test_data("test_item.json")
-    resp = await app_client.get("/search", params={"collections": ["test-collection"], "fields": "properties"})
+    resp = await app_client.get(
+        "/search", params={"collections": ["test-collection"], "fields": "properties"}
+    )
     assert resp.status_code == 200
     resp_json = resp.json()
     feature = resp_json["features"][0]
@@ -232,7 +245,9 @@ async def test_app_query_extension_gte(app_client, ctx):
 
 @pytest.mark.asyncio
 async def test_app_query_extension_limit_lt0(app_client):
-    assert (await app_client.post("/search", json={"query": {}, "limit": -1})).status_code == 400
+    assert (
+        await app_client.post("/search", json={"query": {}, "limit": -1})
+    ).status_code == 400
 
 
 @pytest.mark.skip(reason="removal of context extension")
@@ -256,10 +271,12 @@ async def test_app_sort_extension_get_asc(app_client, txn_client, ctx):
 
     second_item = dict(first_item)
     second_item["id"] = "another-item"
-    another_item_date = datetime.strptime(first_item["properties"]["datetime"], "%Y-%m-%dT%H:%M:%SZ") - timedelta(
-        days=1
+    another_item_date = datetime.strptime(
+        first_item["properties"]["datetime"], "%Y-%m-%dT%H:%M:%SZ"
+    ) - timedelta(days=1)
+    second_item["properties"]["datetime"] = another_item_date.strftime(
+        "%Y-%m-%dT%H:%M:%SZ"
     )
-    second_item["properties"]["datetime"] = another_item_date.strftime("%Y-%m-%dT%H:%M:%SZ")
 
     await create_item(txn_client, second_item)
 
@@ -276,10 +293,12 @@ async def test_app_sort_extension_get_desc(app_client, txn_client, ctx):
 
     second_item = dict(first_item)
     second_item["id"] = "another-item"
-    another_item_date = datetime.strptime(first_item["properties"]["datetime"], "%Y-%m-%dT%H:%M:%SZ") - timedelta(
-        days=1
+    another_item_date = datetime.strptime(
+        first_item["properties"]["datetime"], "%Y-%m-%dT%H:%M:%SZ"
+    ) - timedelta(days=1)
+    second_item["properties"]["datetime"] = another_item_date.strftime(
+        "%Y-%m-%dT%H:%M:%SZ"
     )
-    second_item["properties"]["datetime"] = another_item_date.strftime("%Y-%m-%dT%H:%M:%SZ")
 
     await create_item(txn_client, second_item)
 
@@ -296,10 +315,12 @@ async def test_app_sort_extension_post_asc(app_client, txn_client, ctx):
 
     second_item = dict(first_item)
     second_item["id"] = "another-item"
-    another_item_date = datetime.strptime(first_item["properties"]["datetime"], "%Y-%m-%dT%H:%M:%SZ") - timedelta(
-        days=1
+    another_item_date = datetime.strptime(
+        first_item["properties"]["datetime"], "%Y-%m-%dT%H:%M:%SZ"
+    ) - timedelta(days=1)
+    second_item["properties"]["datetime"] = another_item_date.strftime(
+        "%Y-%m-%dT%H:%M:%SZ"
     )
-    second_item["properties"]["datetime"] = another_item_date.strftime("%Y-%m-%dT%H:%M:%SZ")
 
     await create_item(txn_client, second_item)
 
@@ -320,10 +341,12 @@ async def test_app_sort_extension_post_desc(app_client, txn_client, ctx):
 
     second_item = dict(first_item)
     second_item["id"] = "another-item"
-    another_item_date = datetime.strptime(first_item["properties"]["datetime"], "%Y-%m-%dT%H:%M:%SZ") - timedelta(
-        days=1
+    another_item_date = datetime.strptime(
+        first_item["properties"]["datetime"], "%Y-%m-%dT%H:%M:%SZ"
+    ) - timedelta(days=1)
+    second_item["properties"]["datetime"] = another_item_date.strftime(
+        "%Y-%m-%dT%H:%M:%SZ"
     )
-    second_item["properties"]["datetime"] = another_item_date.strftime("%Y-%m-%dT%H:%M:%SZ")
     await create_item(txn_client, second_item)
 
     params = {
@@ -350,7 +373,9 @@ async def test_search_invalid_date(app_client, ctx):
 
 @pytest.mark.asyncio
 async def test_search_point_intersects_get(app_client, ctx):
-    resp = await app_client.get('/search?intersects={"type":"Point","coordinates":[150.04,-33.14]}')
+    resp = await app_client.get(
+        '/search?intersects={"type":"Point","coordinates":[150.04,-33.14]}'
+    )
 
     assert resp.status_code == 200
     resp_json = resp.json()
@@ -659,11 +684,15 @@ async def test_patch_json_item(app_client, ctx):
         "properties": {"hello": "world", "proj:epsg": 1000, "landsat:column": None},
     }
 
-    resp = await app_client.patch(f"/collections/{ctx.item['collection']}/items/{ctx.item['id']}", json=data)
+    resp = await app_client.patch(
+        f"/collections/{ctx.item['collection']}/items/{ctx.item['id']}", json=data
+    )
 
     assert resp.status_code == 200
 
-    new_resp = await app_client.get(f"/collections/{ctx.item['collection']}/items/{ctx.item['id']}")
+    new_resp = await app_client.get(
+        f"/collections/{ctx.item['collection']}/items/{ctx.item['id']}"
+    )
 
     assert new_resp.status_code == 200
 
@@ -693,7 +722,9 @@ async def test_patch_operations_item(app_client, ctx):
 
     assert resp.status_code == 200
 
-    new_resp = await app_client.get(f"/collections/{ctx.item['collection']}/items/{ctx.item['id']}")
+    new_resp = await app_client.get(
+        f"/collections/{ctx.item['collection']}/items/{ctx.item['id']}"
+    )
 
     assert new_resp.status_code == 200
 
@@ -736,12 +767,16 @@ async def test_search_line_string_intersects(app_client, ctx):
         (21474836470, 1),  # Above Int Limit
     ],
 )
-async def test_big_int_eo_search(app_client, txn_client, test_item, test_collection, value, expected):
+async def test_big_int_eo_search(
+    app_client, txn_client, test_item, test_collection, value, expected
+):
     random_str = "".join(random.choice("abcdef") for _ in range(5))
     collection_id = f"test-collection-eo-{random_str}"
 
     test_collection["id"] = collection_id
-    test_collection["stac_extensions"] = ["https://stac-extensions.github.io/eo/v2.0.0/schema.json"]
+    test_collection["stac_extensions"] = [
+        "https://stac-extensions.github.io/eo/v2.0.0/schema.json"
+    ]
 
     test_item["collection"] = collection_id
     test_item["stac_extensions"] = test_collection["stac_extensions"]