feat: expand MCP server tools for PR and label management, add helper CLI scripts

Closes #7
2026-06-24 00:14:47 -04:00
parent 8b1c115647
commit 82fcd5a4bc
17 changed files with 901 additions and 5 deletions
@@ -0,0 +1,64 @@
+"""Tests for merge_pr.py.
+
+Mocks api_request and credentials.
+"""
+import sys
+import unittest
+from unittest.mock import patch
+
+sys.path.insert(0, str(__import__("pathlib").Path(__file__).resolve().parent.parent))
+import merge_pr  # noqa: E402
+
+
+FAKE_CREDS = "Basic dGVzdHVzZXI6dGVzdHBhc3M="
+
+
+class TestArgParsing(unittest.TestCase):
+
+    @patch("merge_pr.api_request")
+    @patch("merge_pr.get_auth_header", return_value=FAKE_CREDS)
+    def test_minimal_required_args(self, _auth, mock_api):
+        rc = merge_pr.main(["--pr-number", "81"])
+        self.assertEqual(rc, 0)
+        mock_api.assert_called_once()
+
+    def test_missing_pr_number_exits(self):
+        with self.assertRaises(SystemExit):
+            merge_pr.main([])
+
+
+class TestAPIPayload(unittest.TestCase):
+
+    @patch("merge_pr.api_request")
+    @patch("merge_pr.get_auth_header", return_value=FAKE_CREDS)
+    def test_payload_fields(self, _auth, mock_api):
+        rc = merge_pr.main([
+            "--pr-number", "81",
+            "--do", "squash",
+            "--title", "Squash title",
+            "--message", "Squash message",
+            "--force",
+        ])
+        self.assertEqual(rc, 0)
+        mock_api.assert_called_once()
+        method, url, auth, payload = mock_api.call_args[0]
+        self.assertEqual(method, "POST")
+        self.assertEqual(auth, FAKE_CREDS)
+        self.assertEqual(payload["Do"], "squash")
+        self.assertEqual(payload["MergeTitleField"], "Squash title")
+        self.assertEqual(payload["MergeMessageField"], "Squash message")
+        self.assertEqual(payload["force_merge"], True)
+
+    @patch("merge_pr.api_request")
+    @patch("merge_pr.get_auth_header", return_value=FAKE_CREDS)
+    def test_url_construction(self, _auth, mock_api):
+        merge_pr.main(["--pr-number", "81", "--remote", "prgs"])
+        url = mock_api.call_args[0][1]
+        self.assertEqual(
+            url,
+            "https://gitea.prgs.cc/api/v1/repos/Scaled-Tech-Consulting/Timesheet/pulls/81/merge"
+        )
+
+
+if __name__ == "__main__":
+    unittest.main()