test(LAB-3532): add add_metadata adn set_metadata integration tests

Author: paulruelle

tests/integration/presentation/test_metadata.py

@@ -1,6 +1,6 @@
 """Tests for asset metadata functions."""
 
-from typing import Any, Dict, List, cast
+from typing import Dict, List, Union, cast
 
 import pytest_mock
 
@@ -40,7 +40,10 @@ def test_add_metadata_adds_to_existing_metadata(mocker: pytest_mock.MockerFixtur
 
     project_id = "project1"
     asset_ids = ["asset1", "asset2"]
-    new_metadata: List[Dict[str, Any]] = [{"new_key1": "new_value1"}, {"new_key2": "new_value2"}]
+    new_metadata: List[Dict[str, Union[str, float, int]]] = [
+        {"new_key1": "new_value1"},
+        {"new_key2": "new_value2"},
+    ]
 
     result = mutations_asset.add_metadata(
         asset_labeling_metadata=new_metadata, asset_ids=asset_ids, project_id=project_id
@@ -104,8 +107,10 @@ def test_set_metadata_replaces_existing_metadata(mocker: pytest_mock.MockerFixtu
 
     project_id = "project1"
     asset_ids = ["asset1", "asset2"]
-    # Use Dict[str, Any] to satisfy the type checker
-    new_metadata: List[Dict[str, Any]] = [{"new_key1": "new_value1"}, {"new_key2": "new_value2"}]
+    new_metadata: List[Dict[str, Union[str, float, int]]] = [
+        {"new_key1": "new_value1"},
+        {"new_key2": "new_value2"},
+    ]
 
     result = mutations_asset.set_metadata(
         asset_labeling_metadata=new_metadata, asset_ids=asset_ids, project_id=project_id
@@ -151,8 +156,10 @@ def test_add_metadata_handles_missing_metadata(mocker: pytest_mock.MockerFixture
 
     project_id = "project1"
     asset_ids = ["asset1", "asset2"]
-    # Use Dict[str, Any] to satisfy the type checker
-    new_metadata: List[Dict[str, Any]] = [{"new_key1": "new_value1"}, {"new_key2": "new_value2"}]
+    new_metadata: List[Dict[str, Union[str, float, int]]] = [
+        {"new_key1": "new_value1"},
+        {"new_key2": "new_value2"},
+    ]
 
     mutations_asset.add_metadata(
         asset_labeling_metadata=new_metadata, asset_ids=asset_ids, project_id=project_id
@@ -196,8 +203,7 @@ def test_multiple_assets_with_different_metadata_structures(mocker: pytest_mock.
 
     project_id = "project1"
     asset_ids = ["asset1", "asset2", "asset3", "asset4"]
-    # Use Dict[str, Any] to satisfy the type checker
-    new_metadata: List[Dict[str, Any]] = [
+    new_metadata: List[Dict[str, Union[str, float, int]]] = [
         {"meta1": "value1"},
         {"meta2": "value2"},
         {"meta3": "value3"},