diff --git a/tests/test_elements_worker/test_metadata.py b/tests/test_elements_worker/test_metadata.py index 98e9984223988b8dc4d1d7b5715c9c95bc6b3754..b40f9bf0fbce8d267e22522fb90b43a1cfa2ce63 100644 --- a/tests/test_elements_worker/test_metadata.py +++ b/tests/test_elements_worker/test_metadata.py @@ -522,9 +522,9 @@ def test_list_element_metadata(fake_dummy_worker): fake_dummy_worker.api_client.add_response( "ListElementMetaData", id=element.id, - response={"id": "metadata_id"}, + response=[{"id": "metadata_id"}], ) - assert fake_dummy_worker.list_element_metadata(element) == {"id": "metadata_id"} + assert fake_dummy_worker.list_element_metadata(element) == [{"id": "metadata_id"}] assert len(fake_dummy_worker.api_client.history) == 1 assert len(fake_dummy_worker.api_client.responses) == 0 @@ -536,11 +536,28 @@ def test_list_element_metadata_cached_element(mock_elements_worker_with_cache): mock_elements_worker_with_cache.api_client.add_response( "ListElementMetaData", id="element_id", - response={"id": "metadata_id"}, + response=[{"id": "metadata_id"}], ) - assert mock_elements_worker_with_cache.list_element_metadata(element) == { - "id": "metadata_id" - } + assert mock_elements_worker_with_cache.list_element_metadata(element) == [ + {"id": "metadata_id"} + ] assert len(mock_elements_worker_with_cache.api_client.history) == 1 assert len(mock_elements_worker_with_cache.api_client.responses) == 0 + + +def test_list_element_metadata_with_load_parents(fake_dummy_worker): + element = Element({"id": "element_id"}) + fake_dummy_worker.api_client.add_response( + "ListElementMetaData", + id=element.id, + load_parents=True, + response=[{"id": "metadata_id"}, {"id": "parent_metadata_id"}], + ) + assert fake_dummy_worker.list_element_metadata(element, load_parents=True) == [ + {"id": "metadata_id"}, + {"id": "parent_metadata_id"}, + ] + + assert len(fake_dummy_worker.api_client.history) == 1 + assert len(fake_dummy_worker.api_client.responses) == 0