242 lines
9.1 KiB
Python
242 lines
9.1 KiB
Python
"""Tests for merged PR cleanup reconciliation (#269)."""
|
|
|
|
import os
|
|
import sys
|
|
import tempfile
|
|
import unittest
|
|
from unittest.mock import patch, MagicMock
|
|
|
|
sys.path.insert(0, str(__import__("pathlib").Path(__file__).resolve().parent.parent))
|
|
|
|
import merged_cleanup_reconcile as mcr # noqa: E402
|
|
|
|
|
|
class TestMergedCleanupAssessment(unittest.TestCase):
|
|
def test_extract_linked_issue_from_closes(self):
|
|
issue = mcr.extract_linked_issue(
|
|
"feat: cleanup (Closes #269)",
|
|
"No other refs",
|
|
)
|
|
self.assertEqual(issue, 269)
|
|
|
|
def test_merged_remote_branch_safe_when_gates_pass(self):
|
|
result = mcr.assess_remote_branch_cleanup(
|
|
pr_number=42,
|
|
head_branch="feat/issue-269-merged-pr-cleanup-reconcile",
|
|
merged=True,
|
|
remote_branch_exists=True,
|
|
open_pr_heads=set(),
|
|
head_on_master=True,
|
|
delete_capability_allowed=True,
|
|
active_lock=False,
|
|
)
|
|
self.assertTrue(result["safe_to_delete_remote"])
|
|
self.assertEqual(result["block_reasons"], [])
|
|
|
|
def test_open_pr_blocks_remote_delete(self):
|
|
result = mcr.assess_remote_branch_cleanup(
|
|
pr_number=42,
|
|
head_branch="feat/issue-9-example",
|
|
merged=True,
|
|
remote_branch_exists=True,
|
|
open_pr_heads={"feat/issue-9-example"},
|
|
head_on_master=True,
|
|
delete_capability_allowed=True,
|
|
active_lock=False,
|
|
)
|
|
self.assertFalse(result["safe_to_delete_remote"])
|
|
self.assertIn("open PR", result["block_reasons"][0])
|
|
|
|
def test_protected_branch_blocks_remote_delete(self):
|
|
result = mcr.assess_remote_branch_cleanup(
|
|
pr_number=1,
|
|
head_branch="master",
|
|
merged=True,
|
|
remote_branch_exists=True,
|
|
open_pr_heads=set(),
|
|
head_on_master=True,
|
|
delete_capability_allowed=True,
|
|
active_lock=False,
|
|
)
|
|
self.assertFalse(result["safe_to_delete_remote"])
|
|
self.assertIn("protected", result["block_reasons"][0])
|
|
|
|
def test_active_lock_blocks_remote_and_worktree_cleanup(self):
|
|
remote = mcr.assess_remote_branch_cleanup(
|
|
pr_number=42,
|
|
head_branch="feat/issue-9-example",
|
|
merged=True,
|
|
remote_branch_exists=True,
|
|
open_pr_heads=set(),
|
|
head_on_master=True,
|
|
delete_capability_allowed=True,
|
|
active_lock=True,
|
|
)
|
|
local = mcr.assess_local_worktree_cleanup(
|
|
pr_number=42,
|
|
head_branch="feat/issue-9-example",
|
|
merged=True,
|
|
worktree_state={
|
|
"exists": True,
|
|
"clean": True,
|
|
"current_branch": "feat/issue-9-example",
|
|
"worktree_path": "/repo/branches/feat-issue-9-example",
|
|
},
|
|
active_lock=True,
|
|
)
|
|
self.assertFalse(remote["safe_to_delete_remote"])
|
|
self.assertFalse(local["safe_to_remove_worktree"])
|
|
self.assertIn("active issue lock", remote["block_reasons"][0])
|
|
|
|
def test_dirty_worktree_blocks_local_cleanup(self):
|
|
result = mcr.assess_local_worktree_cleanup(
|
|
pr_number=42,
|
|
head_branch="feat/issue-9-example",
|
|
merged=True,
|
|
worktree_state={
|
|
"exists": True,
|
|
"clean": False,
|
|
"dirty_files": ["gitea_mcp_server.py"],
|
|
"current_branch": "feat/issue-9-example",
|
|
"worktree_path": "/repo/branches/feat-issue-9-example",
|
|
},
|
|
active_lock=False,
|
|
)
|
|
self.assertFalse(result["safe_to_remove_worktree"])
|
|
self.assertIn("tracked edits", result["block_reasons"][0])
|
|
|
|
|
|
class TestMergedCleanupReport(unittest.TestCase):
|
|
def test_build_report_skips_unmerged_closed_prs(self):
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
report = mcr.build_reconciliation_report(
|
|
project_root=tmp,
|
|
closed_prs=[
|
|
{
|
|
"number": 10,
|
|
"title": "closed not merged",
|
|
"body": "Closes #10",
|
|
"head": {"ref": "feat/issue-10-x"},
|
|
"merged_at": None,
|
|
},
|
|
{
|
|
"number": 11,
|
|
"title": "merged",
|
|
"body": "Closes #11",
|
|
"head": {"ref": "feat/issue-11-x"},
|
|
"merged_at": "2026-07-06T12:00:00Z",
|
|
"merge_commit_sha": "abc123",
|
|
},
|
|
],
|
|
open_prs=[],
|
|
remote_branch_exists={"feat/issue-11-x": True},
|
|
head_on_master={11: True},
|
|
delete_capability_allowed=False,
|
|
)
|
|
self.assertEqual(report["merged_pr_count"], 1)
|
|
entry = report["entries"][0]
|
|
self.assertEqual(entry["pr_number"], 11)
|
|
self.assertEqual(entry["issue_number"], 11)
|
|
self.assertFalse(entry["remote_branch"]["safe_to_delete_remote"])
|
|
self.assertIn("delete_branch capability", entry["remote_branch"]["block_reasons"][0])
|
|
|
|
def test_has_active_issue_lock_reads_lock_file(self):
|
|
with tempfile.NamedTemporaryFile("w", encoding="utf-8", delete=False) as handle:
|
|
handle.write('{"branch_name": "feat/issue-9-example"}')
|
|
lock_path = handle.name
|
|
try:
|
|
self.assertTrue(
|
|
mcr.has_active_issue_lock("feat/issue-9-example", lock_path=lock_path)
|
|
)
|
|
self.assertFalse(
|
|
mcr.has_active_issue_lock("feat/other-branch", lock_path=lock_path)
|
|
)
|
|
finally:
|
|
os.remove(lock_path)
|
|
|
|
@patch("subprocess.run")
|
|
def test_discover_local_worktrees(self, mock_run):
|
|
mock_output = (
|
|
"worktree /repo/Gitea-Tools\n"
|
|
"bare\n\n"
|
|
"worktree /repo/Gitea-Tools/branches/custom-name\n"
|
|
"branch refs/heads/feat/issue-11-x\n"
|
|
"HEAD cafebabe\n\n"
|
|
)
|
|
mock_run.return_value = MagicMock(returncode=0, stdout=mock_output)
|
|
res = mcr.discover_local_worktrees("/repo/Gitea-Tools")
|
|
self.assertEqual(res, {"feat/issue-11-x": "/repo/Gitea-Tools/branches/custom-name"})
|
|
|
|
@patch("subprocess.run")
|
|
@patch("merged_cleanup_reconcile.read_local_worktree_state")
|
|
def test_build_report_discovers_non_canonical_worktree_by_branch(self, mock_state, mock_run):
|
|
mock_output = (
|
|
"worktree /repo/Gitea-Tools\n"
|
|
"bare\n\n"
|
|
"worktree /repo/Gitea-Tools/branches/custom-name\n"
|
|
"branch refs/heads/feat/issue-11-x\n"
|
|
"HEAD cafebabe\n\n"
|
|
)
|
|
mock_run.return_value = MagicMock(returncode=0, stdout=mock_output)
|
|
mock_state.return_value = {
|
|
"exists": True,
|
|
"clean": True,
|
|
"current_branch": "feat/issue-11-x",
|
|
"head_sha": "cafebabe",
|
|
"dirty_files": [],
|
|
}
|
|
|
|
report = mcr.build_reconciliation_report(
|
|
project_root="/repo/Gitea-Tools",
|
|
closed_prs=[
|
|
{
|
|
"number": 11,
|
|
"title": "merged",
|
|
"body": "Closes #11",
|
|
"head": {"ref": "feat/issue-11-x"},
|
|
"merged_at": "2026-07-06T12:00:00Z",
|
|
"merge_commit_sha": "abc123",
|
|
}
|
|
],
|
|
open_prs=[],
|
|
remote_branch_exists={"feat/issue-11-x": True},
|
|
head_on_master={11: True},
|
|
delete_capability_allowed=True,
|
|
)
|
|
self.assertEqual(report["merged_pr_count"], 1)
|
|
entry = report["entries"][0]
|
|
local = entry["local_worktree"]
|
|
self.assertEqual(local["worktree_path"], "/repo/Gitea-Tools/branches/custom-name")
|
|
self.assertEqual(local["match_type"], "branch")
|
|
|
|
@patch("subprocess.run")
|
|
def test_remove_local_worktree_uses_discovered_path(self, mock_run):
|
|
mock_output = (
|
|
"worktree /repo/Gitea-Tools\n"
|
|
"bare\n\n"
|
|
"worktree /repo/Gitea-Tools/branches/custom-name\n"
|
|
"branch refs/heads/feat/issue-11-x\n"
|
|
"HEAD cafebabe\n\n"
|
|
)
|
|
# First call is discover_local_worktrees, second is git worktree remove
|
|
mock_run.side_effect = [
|
|
MagicMock(returncode=0, stdout=mock_output),
|
|
MagicMock(returncode=0, stdout="", stderr=""),
|
|
]
|
|
|
|
with patch("os.path.isdir", return_value=True):
|
|
result = mcr.remove_local_worktree("/repo/Gitea-Tools", "feat/issue-11-x")
|
|
self.assertTrue(result["success"])
|
|
self.assertTrue(result["performed"])
|
|
self.assertEqual(result["worktree_path"], "/repo/Gitea-Tools/branches/custom-name")
|
|
# Assert git worktree remove was called with the custom path
|
|
mock_run.assert_any_call(
|
|
["git", "-C", "/repo/Gitea-Tools", "worktree", "remove", "/repo/Gitea-Tools/branches/custom-name"],
|
|
capture_output=True,
|
|
text=True,
|
|
check=False,
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main() |