fix: add workitem ID if available in create_work_items

DSD-DBS · Dec 18, 2023 · b05d6f8 · b05d6f8
1 parent 777d8b3
commit b05d6f8
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/capella2polarion/elements/api_helper.py b/capella2polarion/elements/api_helper.py
@@ -39,7 +39,7 @@ def patch_work_item(
     old.linked_work_items = api.get_all_work_item_links(old.id)
     new.type = None
     new.status = "open"
-    new.id = old.id
+    assert new.id is not None
     try:
         api.update_work_item(new)
         handle_links(

diff --git a/capella2polarion/elements/element.py b/capella2polarion/elements/element.py
@@ -49,6 +49,8 @@ def create_work_items(
         assert work_item is not None
         assert work_item.title is not None
         assert work_item.type is not None
+        if old := ctx["POLARION_WI_MAP"].get(work_item.uuid_capella):
+            work_item.id = old.id
         if work_item.type in valid_types:
             work_items.append(work_item)
         else:

diff --git a/tests/test_elements.py b/tests/test_elements.py
@@ -49,7 +49,7 @@
     '<html><p><img style="max-width: 100%" src="data:image/svg+xml;base64,'
 )
 TEST_SER_DIAGRAM: dict[str, t.Any] = {
-    "id": None,
+    "id": "Diag-1",
     "title": "[CC] Capability",
     "description_type": "text/html",
     "type": "diagram",