AIML-150 Update resetting the Issue as well.

Author: Jacob Mages-Haskins

src/git_handler.py

@@ -625,7 +625,7 @@ def find_issue_with_label(label: str) -> int:
         return None
 
 
-def reset_issue(issue_number: int, remediation_label: str) -> bool:
+def reset_issue(issue_number: int, issue_title: str, remediation_label: str) -> bool:
     """
     Resets a GitHub issue by:
     1. Removing all existing labels that start with "smartfix-id:"
@@ -650,7 +650,7 @@ def reset_issue(issue_number: int, remediation_label: str) -> bool:
         return False
 
     # First check if there's an open PR for this issue
-    open_pr = find_open_pr_for_issue(issue_number)
+    open_pr = find_open_pr_for_issue(issue_number, issue_title)
     if open_pr:
         pr_number = open_pr.get("number")
         pr_url = open_pr.get("url")

src/github/external_coding_agent.py

@@ -183,7 +183,7 @@ def generate_fixes(self, vuln_uuid: str, remediation_id: str, vuln_title: str, i
                 error_exit(remediation_id, FailureCategory.AGENT_FAILURE.value)
         else:
             debug_log(f"Found existing GitHub issue #{issue_number} with label {vulnerability_label}")
-            if not git_handler.reset_issue(issue_number, remediation_label):
+            if not git_handler.reset_issue(issue_number, issue_title, remediation_label):
                 log(f"Failed to reset issue #{issue_number} with labels {vulnerability_label}, {remediation_label}", is_error=True)
                 error_exit(remediation_id, FailureCategory.AGENT_FAILURE.value)
             is_existing_issue = True

test/test_external_coding_agent.py

@@ -254,7 +254,7 @@ def test_generate_fixes_with_existing_issue(self, mock_log, mock_debug_log, mock
         mock_debug_log.assert_any_call("Found existing GitHub issue #42 with label contrast-vuln-id:VULN-1234-FAKE-ABCD")
 
         # Verify reset_issue was called
-        mock_reset_issue.assert_called_once_with(42, "smartfix-id:1REM-FAKE-ABCD")
+        mock_reset_issue.assert_called_once_with(42, "Fake Vulnerability Title", "smartfix-id:1REM-FAKE-ABCD")
 
     @patch('src.git_handler.check_issues_enabled')
     @patch('src.git_handler.find_issue_with_label')

test/test_git_handler.py

@@ -248,7 +248,7 @@ def test_reset_issue_success(self, mock_config, mock_debug_log, mock_log, mock_e
         _ = get_config(testing=True)
 
         # Execute
-        result = git_handler.reset_issue(issue_number, remediation_label)
+        result = git_handler.reset_issue(issue_number, "Test Issue Title", remediation_label)
 
         # Assert
         mock_check_issues.assert_called_once()
@@ -275,7 +275,7 @@ def test_reset_issue_failure(self, mock_log, mock_run_command, mock_find_pr, moc
         _ = get_config(testing=True)
 
         # Execute
-        result = git_handler.reset_issue(issue_number, remediation_label)
+        result = git_handler.reset_issue(issue_number, "Test Issue Title", remediation_label)
 
         # Assert
         mock_check_issues.assert_called_once()
@@ -302,10 +302,10 @@ def test_reset_issue_with_open_pr(self, mock_log, mock_find_open_pr):
         _ = get_config(testing=True)
 
         # Execute
-        result = git_handler.reset_issue(issue_number, remediation_label)
+        result = git_handler.reset_issue(issue_number, "Test Issue Title", remediation_label)
 
         # Assert
-        mock_find_open_pr.assert_called_once_with(issue_number)
+        mock_find_open_pr.assert_called_once_with(issue_number, "Test Issue Title")
         self.assertFalse(result)
         mock_log.assert_any_call(
             "Cannot reset issue #42 because it has an open PR #123: https://github.com/mock/repo/pull/123",
@@ -347,7 +347,7 @@ def test_reset_issue_claude_code(self, mock_config, mock_debug_log, mock_log, mo
         mock_ensure_label.return_value = True
 
         # Execute
-        result = git_handler.reset_issue(issue_number, remediation_label)
+        result = git_handler.reset_issue(issue_number, "Test Issue Title", remediation_label)
 
         # Assert
         mock_check_issues.assert_called_once()
@@ -407,7 +407,7 @@ def test_reset_issue_claude_code_error(self, mock_config, mock_log, mock_ensure_
         mock_ensure_label.return_value = True
 
         # Execute
-        result = git_handler.reset_issue(issue_number, remediation_label)
+        result = git_handler.reset_issue(issue_number, "Test Issue Title", remediation_label)
 
         # Assert
         mock_check_issues.assert_called_once()
@@ -941,7 +941,7 @@ def test_reset_issue_issues_disabled(self, mock_log, mock_find_open_pr, mock_che
         _ = get_config(testing=True)
 
         # Execute
-        result = git_handler.reset_issue(issue_number, remediation_label)
+        result = git_handler.reset_issue(issue_number, "Test Issue Title", remediation_label)
 
         # Assert
         self.assertFalse(result)