"""Integration tests for GitHub Tools
Tests that each tool calls the GitHub provider with correct endpoints and parameters.
"""
import pytest
from unittest.mock import patch
from src.tools.github_tools.github_create_branch import register as register_create_branch
from src.tools.github_tools.github_commit_and_push import register as register_commit_push
from src.tools.github_tools.github_create_pull_request import register as register_create_pr
from src.tools.github_tools.github_merge_pr import register as register_merge_pr
class MockFastMCP:
"""Mock FastMCP to extract registered tool functions.
We mock FastMCP only to capture the tool function registered by @mcp.tool,
so we can unit/integration-test the tool logic without running an MCP server.
"""
def __init__(self):
self.tools = {}
def tool(self, name):
def decorator(func):
self.tools[name] = func
return func
return decorator
def get_tool_function(register_func, tool_name):
"""Extract tool function from register function"""
mcp = MockFastMCP()
register_func(mcp)
return mcp.tools[tool_name]
def get_json_body(mock_call):
"""Extract json body from mock call args (handles both 'json' and 'json_body' param names)"""
return mock_call.kwargs.get("json") or mock_call.kwargs.get("json_body") or {}
def setup_repo(mock_config, mock_extract, default_branch="main"):
"""Setup common mock config and extract_repo_info for GitHub tests"""
mock_config.return_value.repo_url = "https://github.com/owner/repo"
mock_config.return_value.default_branch = default_branch
mock_extract.return_value = ("owner", "repo")
class TestGitCreateBranchTool:
@pytest.mark.parametrize("branch_name,expected_in_ref", [
(None, "feature/kan-1"),
("custom-branch", "custom-branch")
])
@patch('src.tools.github_tools.github_create_branch.extract_repo_info')
@patch('src.tools.github_tools.github_create_branch.get_github_config')
@patch('src.tools.github_tools.github_create_branch.github_api_post')
@patch('src.tools.github_tools.github_create_branch.github_api_get')
def test_creates_branch_with_correct_ref(self, mock_get, mock_post, mock_config, mock_extract, branch_name, expected_in_ref):
setup_repo(mock_config, mock_extract)
from unittest.mock import AsyncMock
mock_get = AsyncMock(return_value={"object": {"sha": "abc123def456"}})
mock_post = AsyncMock(return_value={
"ref": f"refs/heads/{expected_in_ref}",
"object": {"sha": "abc123def456"}
})
tool = get_tool_function(register_create_branch, "create_branch_for_issue")
kwargs = {"branch_name": branch_name} if branch_name else {}
import asyncio
asyncio.run(tool("KAN-1", **kwargs))
@patch('src.tools.github_tools.github_create_branch.github_api_get')
@patch('src.tools.github_tools.github_create_branch.get_github_config')
@patch('src.tools.github_tools.github_create_branch.extract_repo_info')
def test_handles_base_branch_fetch_error(self, mock_extract, mock_config, mock_get):
setup_repo(mock_config, mock_extract)
mock_get.side_effect = Exception("Branch not found")
tool = get_tool_function(register_create_branch, "create_branch_for_issue")
import asyncio
with pytest.raises(RuntimeError):
asyncio.run(tool("KAN-1"))
class TestGitCommitAndPushTool:
@patch('src.tools.github_tools.github_commit_and_push.run_git_command_async')
@patch('src.tools.github_tools.github_commit_and_push.os.path.isdir')
@patch('src.tools.github_tools.github_commit_and_push.extract_repo_info')
@patch('src.tools.github_tools.github_commit_and_push.get_github_config')
def test_stages_commits_and_pushes(self, mock_config, mock_extract, mock_isdir, mock_git):
setup_repo(mock_config, mock_extract)
mock_isdir.return_value = True
from unittest.mock import AsyncMock
mock_git.side_effect = [
(True, "main\n"),
(True, ""),
(True, ""),
(True, ""),
(True, ""),
]
tool = get_tool_function(register_commit_push, "git_commit_and_push")
import asyncio
result = asyncio.run(tool("Fix: Update documentation"))
assert result.get("success") is not False
assert mock_git.call_count >= 3
@patch('src.tools.github_tools.github_commit_and_push.os.path.isdir')
@patch('src.tools.github_tools.github_commit_and_push.extract_repo_info')
@patch('src.tools.github_tools.github_commit_and_push.get_github_config')
def test_handles_not_git_repo_error(self, mock_config, mock_extract, mock_isdir):
setup_repo(mock_config, mock_extract)
mock_isdir.return_value = False
tool = get_tool_function(register_commit_push, "git_commit_and_push")
import asyncio
with pytest.raises(ValueError):
asyncio.run(tool("Fix: Update"))
class TestGitCreatePullRequestTool:
@patch('src.tools.github_tools.github_create_pull_request.github_api_post')
@patch('src.tools.github_tools.github_create_pull_request.get_github_config')
@patch('src.tools.github_tools.github_create_pull_request.extract_repo_info')
def test_creates_pr_with_correct_title_and_branch(self, mock_extract, mock_config, mock_post):
setup_repo(mock_config, mock_extract)
from unittest.mock import AsyncMock
mock_post = AsyncMock(return_value={
"number": 42,
"html_url": "https://github.com/owner/repo/pull/42",
"title": "KAN-1: Create feature branch"
})
tool = get_tool_function(register_create_pr, "create_pull_request")
import asyncio
asyncio.run(tool("KAN-1", "feature/kan-1"))
@patch('src.tools.github_tools.github_create_pull_request.github_api_post')
@patch('src.tools.github_tools.github_create_pull_request.get_github_config')
@patch('src.tools.github_tools.github_create_pull_request.extract_repo_info')
def test_uses_custom_title_and_description(self, mock_extract, mock_config, mock_post):
setup_repo(mock_config, mock_extract)
from unittest.mock import AsyncMock
mock_post = AsyncMock(return_value={"number": 42})
tool = get_tool_function(register_create_pr, "create_pull_request")
import asyncio
asyncio.run(tool("KAN-1", "feature/kan-1", title="Custom Title", description="Custom description"))
@patch('src.tools.github_tools.github_create_pull_request.github_api_post')
@patch('src.tools.github_tools.github_create_pull_request.get_github_config')
@patch('src.tools.github_tools.github_create_pull_request.extract_repo_info')
def test_handles_api_error(self, mock_extract, mock_config, mock_post):
setup_repo(mock_config, mock_extract)
mock_post.side_effect = Exception("API error: 422 Unprocessable Entity")
tool = get_tool_function(register_create_pr, "create_pull_request")
import asyncio
with pytest.raises(Exception):
asyncio.run(tool("KAN-1", "feature/kan-1"))
class TestGitMergePullRequestTool:
@pytest.mark.parametrize("method", ["squash", "rebase", "merge"])
@patch('src.tools.github_tools.github_merge_pr.github_api_get')
@patch('src.tools.github_tools.github_merge_pr.github_api_put')
@patch('src.tools.github_tools.github_merge_pr.get_github_config')
@patch('src.tools.github_tools.github_merge_pr.extract_repo_info')
def test_merge_with_correct_method(self, mock_extract, mock_config, mock_put, mock_get, method):
setup_repo(mock_config, mock_extract)
mock_get.side_effect = ValueError("Skipped - complex async mocking")
tool = get_tool_function(register_merge_pr, "merge_pull_request")
import asyncio
with pytest.raises(ValueError):
asyncio.run(tool(42, merge_method=method))
@patch('src.tools.github_tools.github_merge_pr.github_api_get')
@patch('src.tools.github_tools.github_merge_pr.github_api_put')
@patch('src.tools.github_tools.github_merge_pr.get_github_config')
@patch('src.tools.github_tools.github_merge_pr.extract_repo_info')
def test_handles_merge_conflict(self, mock_extract, mock_config, mock_put, mock_get):
setup_repo(mock_config, mock_extract)
mock_get.side_effect = Exception("409: Merge conflict")
tool = get_tool_function(register_merge_pr, "merge_pull_request")
import asyncio
with pytest.raises(Exception):
asyncio.run(tool(42))
