From 73244a8519b5ed7d0b50495dceac13f8d11169b7 Mon Sep 17 00:00:00 2001
From: EvaBardou <bardou@teklia.com>
Date: Thu, 25 Jan 2024 16:30:18 +0100
Subject: [PATCH] Add test
---
tests/test_elements_worker/test_metadata.py | 29 ++++++++++++++++-----
1 file changed, 23 insertions(+), 6 deletions(-)
diff --git a/tests/test_elements_worker/test_metadata.py b/tests/test_elements_worker/test_metadata.py
index 98e99842..b40f9bf0 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
--
GitLab