Fixes #13

+ unittest

Author: mcurlej

joft/actions.py

@@ -23,7 +23,8 @@ def create_ticket(
 
     logging.info(f"New Jira ticket created: {new_issue.permalink()}")
 
-    reference_pool[action.object_id] = new_issue
+    if action.object_id:
+        reference_pool[action.object_id] = new_issue
 
 
 # TODO jira_session is not needed here. Maybe remove?
@@ -53,6 +54,9 @@ def update_ticket(
 
     logging.info(f"Ticket '{ticket_to.key}' updated.")
 
+    if action.object_id:
+        reference_pool[action.object_id] = ticket_to
+
 
 def link_issues(
     action: joft.models.LinkIssuesAction,
@@ -105,3 +109,6 @@ def transition_issue(
     jira_session.transition_issue(
         ticket_to, action.transition, action.fields, action.comment
     )
+
+    if action.object_id:
+        reference_pool[action.object_id] = ticket_to

joft/base.py

@@ -148,7 +148,8 @@ def validate_uniqueness_of_object_ids(jira_template: joft.models.JiraTemplate) -
         object_ids.append(jira_template.jira_search.object_id)
 
     for action in jira_template.jira_actions:
-        object_ids.append(action.object_id)
+        if action.object_id:
+            object_ids.append(action.object_id)
 
     # check if all the object ids are unique
     if len(object_ids) != len(set(object_ids)):

joft/models.py

@@ -19,7 +19,6 @@ class ReferenceData:
 class Action:
     # required fields
     type: str
-    object_id: str
     fields: typing.Dict[str, typing.Any]
 
     def reuse_data_must_be_list(self, reuse_data):
@@ -34,6 +33,7 @@ def reuse_data_must_be_list(self, reuse_data):
 @dataclasses.dataclass(kw_only=True)
 class CreateTicketAction(Action):
     reuse_data: dataclasses.InitVar[typing.List[ReferenceData] | None] = None
+    object_id: str | None = None
 
     reference_data: typing.List[ReferenceData] = dataclasses.field(default_factory=list)
 
@@ -48,6 +48,7 @@ def __post_init__(self, reuse_data):
 @dataclasses.dataclass(kw_only=True)
 class UpdateTicketAction(Action):
     reference_id: str
+    object_id: str | None = None
     reuse_data: dataclasses.InitVar[typing.List[ReferenceData] | None] = None
 
     reference_data: typing.List[ReferenceData] = dataclasses.field(default_factory=list)
@@ -63,6 +64,7 @@ def __post_init__(self, reuse_data):
 @dataclasses.dataclass(kw_only=True)
 class LinkIssuesAction(Action):
     reuse_data: dataclasses.InitVar[typing.List[ReferenceData] | None] = None
+    object_id: str | None = None
 
     reference_data: typing.List[ReferenceData] = dataclasses.field(default_factory=list)
 
@@ -79,6 +81,7 @@ class TransitionAction(Action):
     reference_id: str
     transition: str
     comment: str
+    object_id: str | None = None
     reuse_data: dataclasses.InitVar[typing.List[ReferenceData] | None] = None
 
     reference_data: typing.List[ReferenceData] = dataclasses.field(default_factory=list)

tests/test_base.py

@@ -9,7 +9,9 @@
 
 
 def _setup_jira_template_yaml(
-    duplicate_id: bool = False, invalid_action: bool = False
+    duplicate_id: bool = False, 
+    invalid_action: bool = False, 
+    no_object_ids: bool = False,
 ) -> dict[str, typing.Any]:
     """Setup function that provides different types of yaml structures"""
 
@@ -63,6 +65,10 @@ def _setup_jira_template_yaml(
     if invalid_action:
         jira_template_yaml["actions"][0]["type"] = "invalid_action"
 
+    if no_object_ids:
+        for action in jira_template_yaml["actions"]:
+            action.pop("object_id", None)
+
     return jira_template_yaml
 
 
@@ -350,3 +356,14 @@ def test_validate_template_success(
     assert ret_code == 0
     mock_load_and_parse_yaml.assert_called_once_with(yaml_file_path)
     mock_validate_uniqueness_object_ids.assert_called_once_with(jira_template)
+
+
+def test_object_id_not_present() -> None:
+    """Test that Object IDs are optional, but still present in the dataclasses with 
+    None value"""
+
+    jira_template_yaml = _setup_jira_template_yaml(no_object_ids=True)
+    jira_template = joft.models.JiraTemplate(**jira_template_yaml)
+
+    for action in jira_template.jira_actions:
+        assert not action.object_id