Scaled-Tech-Consulting/Gitea-Tools
2
0
Fork 0
Code Issues 6 Pull Requests 1 Actions Packages Projects Releases Wiki Activity

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

Browse Source 
Closes #7
This commit is contained in:
Jason Walker
2026-06-24 00:14:47 -04:00
parent 8b1c115647
commit 82fcd5a4bc
17 changed files with 901 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_mcp_server.py
+82
View File
@@ -18,6 +18,10 @@ from mcp_server import ( # noqa: E402
gitea_view_issue,
gitea_mark_issue,
gitea_mirror_refs,
gitea_list_prs,
gitea_view_pr,
gitea_merge_pr,
gitea_delete_branch,
)
FAKE_AUTH = "Basic dGVzdDp0ZXN0"
@@ -240,5 +244,83 @@ class TestMirrorRefs(unittest.TestCase):
self.assertIn("--force", args)
# ---------------------------------------------------------------------------
# List PRs
# ---------------------------------------------------------------------------
class TestListPRs(unittest.TestCase):
@patch("mcp_server.api_request")
@patch("mcp_server.get_auth_header", return_value=FAKE_AUTH)
def test_list_prs(self, _auth, mock_api):
mock_api.return_value = [
{
"number": 1, "title": "PR 1", "state": "open",
"head": {"ref": "branch1"}, "base": {"ref": "main"},
"html_url": "http://url1", "mergeable": True
}
]
result = gitea_list_prs()
self.assertEqual(len(result), 1)
self.assertEqual(result[0]["number"], 1)
self.assertEqual(result[0]["head"], "branch1")
# ---------------------------------------------------------------------------
# View PR
# ---------------------------------------------------------------------------
class TestViewPR(unittest.TestCase):
@patch("mcp_server.api_request")
@patch("mcp_server.get_auth_header", return_value=FAKE_AUTH)
def test_view_pr(self, _auth, mock_api):
mock_api.return_value = {
"number": 1, "title": "PR 1", "state": "open",
"head": {"ref": "branch1"}, "base": {"ref": "main"},
"html_url": "http://url1", "mergeable": True, "body": "description",
"user": {"login": "user1"}
}
result = gitea_view_pr(pr_number=1)
self.assertEqual(result["number"], 1)
self.assertEqual(result["body"], "description")
self.assertEqual(result["user"], "user1")
# ---------------------------------------------------------------------------
# Merge PR
# ---------------------------------------------------------------------------
class TestMergePR(unittest.TestCase):
@patch("mcp_server.api_request")
@patch("mcp_server.get_auth_header", return_value=FAKE_AUTH)
def test_merge_pr(self, _auth, mock_api):
mock_api.return_value = {}
result = gitea_merge_pr(pr_number=1, do="squash", title="T", message="M", force=True)
self.assertTrue(result["success"])
self.assertIn("merged", result["message"])
# Check payload
payload = mock_api.call_args[0][3]
self.assertEqual(payload["Do"], "squash")
self.assertEqual(payload["MergeTitleField"], "T")
self.assertEqual(payload["MergeMessageField"], "M")
self.assertEqual(payload["force_merge"], True)
# ---------------------------------------------------------------------------
# Delete Branch
# ---------------------------------------------------------------------------
class TestDeleteBranch(unittest.TestCase):
@patch("mcp_server.api_request")
@patch("mcp_server.get_auth_header", return_value=FAKE_AUTH)
def test_delete_branch(self, _auth, mock_api):
mock_api.return_value = {}
result = gitea_delete_branch(branch="feat/branch")
self.assertTrue(result["success"])
self.assertIn("deleted", result["message"])
# Check url encoding of branch name
url = mock_api.call_args[0][1]
self.assertIn("feat%2Fbranch", url)
if __name__ == "__main__":
unittest.main()
tests/test_merge_pr.py
+64
View File
@@ -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()
tests/test_prs.py
+68
View File
@@ -0,0 +1,68 @@
"""Tests for list_prs.py, view_pr.py, and delete_branch.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 list_prs # noqa: E402
import view_pr # noqa: E402
import delete_branch # noqa: E402
FAKE_CREDS = "Basic dGVzdHVzZXI6dGVzdHBhc3M="
class TestListPRs(unittest.TestCase):
@patch("list_prs.api_request")
@patch("list_prs.get_auth_header", return_value=FAKE_CREDS)
def test_list_prs_success(self, _auth, mock_api):
mock_api.return_value = [
{"number": 1, "title": "PR 1", "head": {"ref": "branch1"}, "base": {"ref": "main"}, "html_url": "http://url1", "mergeable": True}
]
rc = list_prs.main([])
self.assertEqual(rc, 0)
mock_api.assert_called_once()
@patch("list_prs.api_request", return_value=[])
@patch("list_prs.get_auth_header", return_value=FAKE_CREDS)
def test_list_prs_empty(self, _auth, mock_api):
rc = list_prs.main(["--state", "closed"])
self.assertEqual(rc, 0)
mock_api.assert_called_once()
class TestViewPR(unittest.TestCase):
@patch("view_pr.api_request")
@patch("view_pr.get_auth_header", return_value=FAKE_CREDS)
def test_view_pr_success(self, _auth, mock_api):
mock_api.return_value = {"number": 1, "title": "PR 1", "state": "open", "user": {"login": "user1"}, "head": {"ref": "branch1"}, "base": {"ref": "main"}, "html_url": "http://url1", "mergeable": True, "body": "PR description"}
rc = view_pr.main(["1"])
self.assertEqual(rc, 0)
mock_api.assert_called_once()
def test_missing_pr_number_exits(self):
with self.assertRaises(SystemExit):
view_pr.main([])
class TestDeleteBranch(unittest.TestCase):
@patch("delete_branch.api_request")
@patch("delete_branch.get_auth_header", return_value=FAKE_CREDS)
def test_delete_branch_success(self, _auth, mock_api):
rc = delete_branch.main(["feat/my-branch"])
self.assertEqual(rc, 0)
mock_api.assert_called_once()
def test_missing_branch_exits(self):
with self.assertRaises(SystemExit):
delete_branch.main([])
if __name__ == "__main__":
unittest.main()