feat: expand MCP server tools for PR and label management, add helper CLI scripts
Closes #7
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user