Accumulate Time Lost from comments sent through LOVE

CHANGELOG.rst

@@ -2,6 +2,11 @@
 Version History
 ===============
 
+v7.0.2
+------
+
+* Add accumulation of time lost in Jira comments
+
 v7.0.1
 ------
 

manager/api/tests/test_jira.py

@@ -26,7 +26,12 @@
 import requests
 from django.test import TestCase, override_settings
 
-from manager.utils import handle_jira_payload, jira_comment, jira_ticket
+from manager.utils import (
+    handle_jira_payload,
+    jira_comment,
+    jira_ticket,
+    update_time_loss,
+)
 
 OLE_JIRA_OBS_COMPONENTS_FIELDS = [
     "AuxTel",
@@ -311,6 +316,27 @@ def test_needed_parameters(self):
 
         mock_jira_patcher.stop()
 
+    @patch("requests.get")
+    @patch("requests.put")
+    def test_update_time_loss(self, mock_get, mock_put):
+        """Test call to update_time_loss and verify field was updated"""
+        # patch both requests.get, requests.put
+        response = requests.Response()
+        response.status_code = 200
+        mock_put.return_value = response
+
+        response.json = lambda: {"customfield_10106": 13.6}
+        mock_get.return_value = response
+
+        # call update time lost
+        jira_response = update_time_loss(1, 3.4)
+        assert jira_response.status_code == 200
+        assert jira_response.data["ack"] == "Jira field updated"
+
+        jira_response = update_time_loss(93827, 1.23)
+        assert jira_response.status_code == 200
+        assert jira_response.data["ack"] == "Jira field updated"
+
     def test_add_comment(self):
         """Test call to jira_comment function with all needed parameters"""
         mock_jira_patcher = patch("requests.post")
@@ -319,6 +345,10 @@ def test_add_comment(self):
         response.status_code = 201
         mock_jira_client.return_value = response
 
+        mock_timeloss_patcher = patch("manager.utils.update_time_loss")
+        mock_timeloss_client = mock_timeloss_patcher.start()
+        mock_timeloss_client.return_value = response
+
         jira_response = jira_comment(self.jira_request_exposure_full_jira_comment.data)
         assert jira_response.status_code == 200
         assert jira_response.data["ack"] == "Jira comment created"

manager/manager/utils.py

@@ -505,6 +505,57 @@ def jira_ticket(request_data):
     )
 
 
+def update_time_loss(jira_id: int, add_time_loss: float = 0.0) -> Response:
+    """Connect to the Rubin Observatory JIRA Cloud REST API to
+    update a jira ticket's specific field
+
+    Params
+    ------
+    jira_id: int
+        Jira ID
+
+    Returns
+    -------
+    Response
+        define response here
+    """
+    headers = {
+        "Authorization": f"Basic {os.environ.get('JIRA_API_TOKEN')}",
+        "content-type": "application/json",
+    }
+    get_url = (
+        f"https://{os.environ.get('JIRA_API_HOSTNAME')}/rest/api/latest/issue/{jira_id}"
+    )
+    response = requests.get(get_url, headers=headers)
+    existent_time_loss = (
+        response.json().get("customfield_10106", 0.0)
+        if response.status_code == 200
+        else 0.0
+    )
+    jira_payload = {
+        "fields": {
+            "customfield_10106": float(existent_time_loss + add_time_loss),
+        },
+    }
+    put_url = f"https://{os.environ.get('JIRA_API_HOSTNAME')}/rest/api/latest/issue/{jira_id}/"
+    response = requests.put(put_url, json=jira_payload, headers=headers)
+
+    if response.status_code == 200 or response.status_code == 204:
+        return Response(
+            {
+                "ack": "Jira field updated",
+                "url": f"https://{os.environ.get('JIRA_API_HOSTNAME')}/browse/{jira_id}",
+            },
+            status=200,
+        )
+    return Response(
+        {
+            "ack": "Jira field could not be updated",
+        },
+        status=400,
+    )
+
+
 def jira_comment(request_data):
     """Connect to the Rubin Observatory JIRA Cloud REST API to
     make a comment on a previously created ticket.
@@ -547,6 +598,12 @@ def jira_comment(request_data):
     }
     url = f"https://{os.environ.get('JIRA_API_HOSTNAME')}/rest/api/latest/issue/{jira_id}/comment"
     response = requests.post(url, json=jira_payload, headers=headers)
+
+    if "time_lost" in request_data:
+        update_time_loss(
+            jira_id=jira_id, add_time_loss=request_data.get("time_lost", 0.0)
+        )
+
     if response.status_code == 201:
         return Response(
             {