feat: impl migration

stac-utils · Oct 13, 2023 · 8bf6cdc · 8bf6cdc
1 parent 0399580
commit 8bf6cdc
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 2 deletions.
diff --git a/pystac/extensions/version.py b/pystac/extensions/version.py
@@ -402,8 +402,16 @@ def __init__(self, item_asset: AssetDefinition):
 
 class VersionExtensionHooks(ExtensionHooks):
     schema_uri = SCHEMA_URI
-    prev_extension_ids = {"version"}
-    stac_object_types = {STACObjectType.COLLECTION, STACObjectType.ITEM}
+    prev_extension_ids = {
+        "version",
+        "https://stac-extensions.github.io/version/v1.0.0/schema.json",
+        "https://stac-extensions.github.io/version/v1.1.0/schema.json",
+    }
+    stac_object_types = {
+        STACObjectType.COLLECTION,
+        STACObjectType.ITEM,
+        STACObjectType.CATALOG,
+    }
 
     def get_object_links(self, so: STACObject) -> list[str] | None:
         if isinstance(so, Collection) or isinstance(so, Item):

diff --git a/tests/extensions/test_version.py b/tests/extensions/test_version.py
@@ -471,3 +471,16 @@ def test_assets(item: Item) -> None:
     assert "version" in asset.extra_fields
 
     item.validate()
+
+
+@pytest.mark.parametrize("version", ["v1.0.0", "v1.1.0"])
+def test_migrate(version: str, item: Item) -> None:
+    item_dict = item.to_dict(include_self_link=False, transform_hrefs=False)
+    item_dict["stac_extensions"] = [
+        f"https://stac-extensions.github.io/version/{version}/schema.json"
+    ]
+    item = Item.from_dict(item_dict, migrate=True)
+    assert (
+        "https://stac-extensions.github.io/version/v1.2.0/schema.json"
+        in item.stac_extensions
+    )