Allow to set `load_parents` parameter while using `list_element_metadata` helper

Closes #279 (closed)

tests/test_elements_worker/test_metadata.py

@@ -508,14 +508,23 @@ def test_create_metadatas_api_error(responses, mock_elements_worker):
     ]
 
 
+def test_list_element_metadata_wrong_load_parents(fake_dummy_worker):
+    element = Element({"id": "element_id"})
+    with pytest.raises(AssertionError, match="load_parents should be of type bool"):
+        fake_dummy_worker.list_element_metadata(
+            element=element,
+            load_parents="not bool",
+        )
+
+
 def test_list_element_metadata(fake_dummy_worker):
     element = Element({"id": "element_id"})
     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
@@ -527,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