test

Author: Yuri Zmytrakov

stac_fastapi/tests/api/test_api.py

@@ -1537,3 +1537,93 @@ async def test_search_max_item_limit(
     assert resp.status_code == 200
     resp_json = resp.json()
     assert int(limit) == len(resp_json["features"])
+
+
+@pytest.mark.asyncio
+async def test_use_datetime_true_finds_both_items(
+    app_client, load_test_data, txn_client, monkeypatch
+):
+    monkeypatch.setenv("USE_DATETIME", "true")
+
+    test_collection = load_test_data("test_collection.json")
+    test_collection["id"] = "test-collection-datetime-true"
+    await create_collection(txn_client, test_collection)
+
+    item = load_test_data("test_item.json")
+
+    item1 = item.copy()
+    item1["id"] = "test-item-datetime"
+    item1["collection"] = test_collection["id"]
+    item1["properties"]["datetime"] = "2020-01-01T12:00:00Z"
+    await create_item(txn_client, item1)
+
+    item2 = item.copy()
+    item2["id"] = "test-item-start-end"
+    item2["collection"] = test_collection["id"]
+    item1["properties"]["datetime"] = None
+    item2["properties"]["start_datetime"] = "2020-01-01T10:00:00Z"
+    item2["properties"]["end_datetime"] = "2020-01-01T13:00:00Z"
+    await create_item(txn_client, item2)
+
+    resp = await app_client.post(
+        "/search",
+        json={
+            "datetime": "2020-01-01T12:00:00Z",
+            "collections": [test_collection["id"]],
+        },
+    )
+
+    assert resp.status_code == 200
+    resp_json = resp.json()
+
+    found_ids = {feature["id"] for feature in resp_json["features"]}
+    assert "test-item-datetime" in found_ids
+    assert "test-item-start-end" in found_ids
+
+
+@pytest.mark.asyncio
+async def test_use_datetime_false_search(
+    app_client, load_test_data, txn_client, monkeypatch
+):
+    monkeypatch.setenv("USE_DATETIME", "false")
+
+    test_collection = load_test_data("test_collection.json")
+    test_collection["id"] = "test-collection-datetime-false"
+    await create_collection(txn_client, test_collection)
+
+    item = load_test_data("test_item.json")
+
+    # Item 1: Should NOT be found
+    item1 = item.copy()
+    item1["id"] = "test-item-datetime-only"
+    item1["collection"] = test_collection["id"]
+    item1["properties"]["datetime"] = "2020-01-01T12:00:00Z"
+    item1["properties"]["start_datetime"] = "2021-01-01T10:00:00Z"
+    item1["properties"]["end_datetime"] = "2021-01-01T14:00:00Z"
+    await create_item(txn_client, item1)
+
+    # Item 2: Should be found
+    item2 = item.copy()
+    item2["id"] = "test-item-start-end-only"
+    item2["collection"] = test_collection["id"]
+    item2["properties"]["datetime"] = None
+    item2["properties"]["start_datetime"] = "2020-01-01T10:00:00Z"
+    item2["properties"]["end_datetime"] = "2020-01-01T14:00:00Z"
+    await create_item(txn_client, item2)
+
+    resp = await app_client.post(
+        "/search",
+        json={
+            "datetime": "2020-01-01T12:00:00Z",
+            "collections": [test_collection["id"]],
+            "limit": 10,
+        },
+    )
+
+    assert resp.status_code == 200
+    resp_json = resp.json()
+
+    found_ids = {feature["id"] for feature in resp_json["features"]}
+
+    assert "test-item-datetime-only" not in found_ids
+    assert "test-item-start-end-only" in found_ids