Skip to content

Commit ab2eb10

Browse files
committed
Generate item_data in a loop, Ensure correct "collection" in item_data
1 parent 3549806 commit ab2eb10

File tree

1 file changed

+12
-27
lines changed

1 file changed

+12
-27
lines changed

stac_fastapi/tests/extensions/test_filter.py

Lines changed: 12 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -648,33 +648,18 @@ async def test_queryables_enum_platform(
648648
)
649649

650650
# Create items with different platform values
651-
item_data_1 = load_test_data("test_item.json")
652-
item_data_1["id"] = "enum-test-item-1"
653-
item_data_1["properties"]["platform"] = "landsat-8"
654-
await txn_client.create_item(
655-
collection_id=collection_id,
656-
item=api.Item(**item_data_1),
657-
request=MockRequest,
658-
)
659-
660-
item_data_2 = load_test_data("test_item.json")
661-
item_data_2["id"] = "enum-test-item-2"
662-
item_data_2["properties"]["platform"] = "sentinel-2"
663-
await txn_client.create_item(
664-
collection_id=collection_id,
665-
item=api.Item(**item_data_2),
666-
request=MockRequest,
667-
)
668-
669-
item_data_3 = load_test_data("test_item.json")
670-
item_data_3["id"] = "enum-test-item-3"
671-
item_data_3["properties"]["platform"] = "landsat-8"
672-
await txn_client.create_item(
673-
collection_id=collection_id,
674-
item=api.Item(**item_data_3),
675-
request=MockRequest,
676-
refresh=True,
677-
)
651+
NUM_ITEMS = 3
652+
for i in range(1, NUM_ITEMS + 1):
653+
item_data = load_test_data("test_item.json")
654+
item_data["id"] = f"enum-test-item-{i}"
655+
item_data["collection"] = collection_id
656+
item_data["properties"]["platform"] = "landsat-8" if i % 2 else "sentinel-2"
657+
await txn_client.create_item(
658+
collection_id=collection_id,
659+
item=api.Item(**item_data),
660+
request=MockRequest,
661+
refresh=i == NUM_ITEMS,
662+
)
678663

679664
# Act
680665
# Test queryables endpoint

0 commit comments

Comments
 (0)