feat: add PR review and edit tools to CLI and MCP server
This commit is contained in:
@@ -21,7 +21,11 @@ from mcp_server import ( # noqa: E402
|
||||
gitea_list_prs,
|
||||
gitea_view_pr,
|
||||
gitea_merge_pr,
|
||||
gitea_review_pr,
|
||||
gitea_delete_branch,
|
||||
gitea_edit_pr,
|
||||
gitea_get_file,
|
||||
gitea_commit_files,
|
||||
)
|
||||
|
||||
FAKE_AUTH = "Basic dGVzdDp0ZXN0"
|
||||
@@ -305,6 +309,47 @@ class TestMergePR(unittest.TestCase):
|
||||
self.assertEqual(payload["force_merge"], True)
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Review PR
|
||||
# ---------------------------------------------------------------------------
|
||||
class TestReviewPR(unittest.TestCase):
|
||||
|
||||
@patch("mcp_server.api_request")
|
||||
@patch("mcp_server.get_auth_header", return_value=FAKE_AUTH)
|
||||
def test_review_pr_and_merge(self, _auth, mock_api):
|
||||
# GET PR response (fetch head SHA)
|
||||
mock_api.side_effect = [
|
||||
{"head": {"sha": "sha-val-123"}}, # GET PR pulls/1
|
||||
{}, # POST review
|
||||
{}, # POST merge
|
||||
]
|
||||
result = gitea_review_pr(
|
||||
pr_number=1,
|
||||
event="APPROVE",
|
||||
body="Looks good",
|
||||
merge=True,
|
||||
merge_method="squash"
|
||||
)
|
||||
self.assertTrue(result["success"])
|
||||
self.assertIn("Successfully submitted review", result["message"])
|
||||
self.assertIn("Successfully merged", result["message"])
|
||||
|
||||
# Check call counts and arguments
|
||||
self.assertEqual(mock_api.call_count, 3)
|
||||
|
||||
# Verify GET PR
|
||||
self.assertEqual(mock_api.call_args_list[0][0][0], "GET")
|
||||
|
||||
# Verify POST review
|
||||
self.assertEqual(mock_api.call_args_list[1][0][0], "POST")
|
||||
self.assertEqual(mock_api.call_args_list[1][0][3]["event"], "APPROVE")
|
||||
self.assertEqual(mock_api.call_args_list[1][0][3]["commit_id"], "sha-val-123")
|
||||
|
||||
# Verify POST merge
|
||||
self.assertEqual(mock_api.call_args_list[2][0][0], "POST")
|
||||
self.assertEqual(mock_api.call_args_list[2][0][3]["Do"], "squash")
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Delete Branch
|
||||
# ---------------------------------------------------------------------------
|
||||
@@ -322,5 +367,104 @@ class TestDeleteBranch(unittest.TestCase):
|
||||
self.assertIn("feat%2Fbranch", url)
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Edit PR
|
||||
# ---------------------------------------------------------------------------
|
||||
class TestEditPR(unittest.TestCase):
|
||||
|
||||
@patch("mcp_server.api_request")
|
||||
@patch("mcp_server.get_auth_header", return_value=FAKE_AUTH)
|
||||
def test_edit_pr_success(self, _auth, mock_api):
|
||||
mock_api.return_value = {
|
||||
"number": 1,
|
||||
"title": "New Title",
|
||||
"body": "New description",
|
||||
"state": "open",
|
||||
"html_url": "https://gitea.example.com/pulls/1"
|
||||
}
|
||||
result = gitea_edit_pr(
|
||||
pr_number=1,
|
||||
title="New Title",
|
||||
body="New description",
|
||||
state="open"
|
||||
)
|
||||
self.assertTrue(result["success"])
|
||||
self.assertEqual(result["title"], "New Title")
|
||||
self.assertEqual(result["body"], "New description")
|
||||
|
||||
# Verify PATCH and payload
|
||||
call_args = mock_api.call_args
|
||||
self.assertEqual(call_args[0][0], "PATCH")
|
||||
self.assertEqual(call_args[0][3]["title"], "New Title")
|
||||
self.assertEqual(call_args[0][3]["body"], "New description")
|
||||
self.assertEqual(call_args[0][3]["state"], "open")
|
||||
|
||||
def test_edit_pr_no_fields_raises(self):
|
||||
with self.assertRaises(ValueError):
|
||||
gitea_edit_pr(pr_number=1)
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Get File
|
||||
# ---------------------------------------------------------------------------
|
||||
class TestGetFile(unittest.TestCase):
|
||||
|
||||
@patch("mcp_server.api_request")
|
||||
@patch("mcp_server.get_auth_header", return_value=FAKE_AUTH)
|
||||
def test_get_file_success(self, _auth, mock_api):
|
||||
mock_api.return_value = {
|
||||
"name": "README.md",
|
||||
"path": "README.md",
|
||||
"sha": "3a0b123",
|
||||
"size": 100,
|
||||
"encoding": "base64",
|
||||
"content": "SGVsbG8gV29ybGQ="
|
||||
}
|
||||
result = gitea_get_file(filepath="README.md", ref="main")
|
||||
self.assertEqual(result["name"], "README.md")
|
||||
self.assertEqual(result["sha"], "3a0b123")
|
||||
self.assertEqual(result["content"], "SGVsbG8gV29ybGQ=")
|
||||
|
||||
# Verify endpoint and GET method
|
||||
call_args = mock_api.call_args
|
||||
self.assertEqual(call_args[0][0], "GET")
|
||||
self.assertIn("contents/README.md", call_args[0][1])
|
||||
self.assertIn("ref=main", call_args[0][1])
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Commit Files
|
||||
# ---------------------------------------------------------------------------
|
||||
class TestCommitFiles(unittest.TestCase):
|
||||
|
||||
@patch("mcp_server.api_request")
|
||||
@patch("mcp_server.get_auth_header", return_value=FAKE_AUTH)
|
||||
def test_commit_files_success(self, _auth, mock_api):
|
||||
mock_api.return_value = {
|
||||
"commit": {"sha": "commit-sha-123"},
|
||||
"branch": {"name": "test-branch"}
|
||||
}
|
||||
files = [
|
||||
{"operation": "create", "path": "test.txt", "content": "SGVsbG8="}
|
||||
]
|
||||
result = gitea_commit_files(
|
||||
files=files,
|
||||
message="Initial commit",
|
||||
new_branch="test-branch"
|
||||
)
|
||||
self.assertTrue(result["success"])
|
||||
self.assertEqual(result["commit"], "commit-sha-123")
|
||||
self.assertEqual(result["branch"], "test-branch")
|
||||
|
||||
# Verify POST method and payload
|
||||
call_args = mock_api.call_args
|
||||
self.assertEqual(call_args[0][0], "POST")
|
||||
self.assertIn("/contents", call_args[0][1])
|
||||
payload = call_args[0][3]
|
||||
self.assertEqual(payload["message"], "Initial commit")
|
||||
self.assertEqual(payload["new_branch"], "test-branch")
|
||||
self.assertEqual(payload["files"], files)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
|
||||
Reference in New Issue
Block a user