galaxis-agent/tests/test_commit_and_open_pr.py

import pytest
from unittest.mock import AsyncMock, MagicMock, patch


def test_pr_creation_after_push():
    """push 성공 후 GiteaClient로 PR을 생성한다."""
    mock_gitea = MagicMock()
    mock_gitea.create_pull_request = AsyncMock(
        return_value={
            "number": 1,
            "html_url": "http://gitea:3000/quant/galaxis-po/pulls/1",
        }
    )
    mock_sandbox = MagicMock()
    mock_result = MagicMock(exit_code=0, output="")

    with patch(
        "agent.tools.commit_and_open_pr.get_gitea_client", return_value=mock_gitea
    ), patch(
        "agent.tools.commit_and_open_pr.get_sandbox_backend_sync",
        return_value=mock_sandbox,
    ), patch(
        "agent.tools.commit_and_open_pr.get_config",
        return_value={
            "configurable": {
                "thread_id": "test-thread",
                "repo": {"owner": "quant", "name": "galaxis-po"},
            }
        },
    ), patch(
        "agent.tools.commit_and_open_pr.resolve_repo_dir",
        return_value="/workspace/galaxis-po",
    ), patch(
        "agent.tools.commit_and_open_pr.git_has_uncommitted_changes",
        return_value=True,
    ), patch(
        "agent.tools.commit_and_open_pr.git_fetch_origin",
    ), patch(
        "agent.tools.commit_and_open_pr.git_has_unpushed_commits",
        return_value=False,
    ), patch(
        "agent.tools.commit_and_open_pr.git_current_branch",
        return_value="galaxis-agent/test-thread",
    ), patch(
        "agent.tools.commit_and_open_pr.git_checkout_branch",
    ), patch(
        "agent.tools.commit_and_open_pr.git_config_user",
    ), patch(
        "agent.tools.commit_and_open_pr.git_add_all",
    ), patch(
        "agent.tools.commit_and_open_pr.git_commit",
        return_value=mock_result,
    ), patch(
        "agent.tools.commit_and_open_pr.git_push",
        return_value=mock_result,
    ), patch.dict(
        "os.environ", {"GITEA_TOKEN": "test-token"},
    ):
        from agent.tools.commit_and_open_pr import commit_and_open_pr
        result = commit_and_open_pr(title="feat: add feature", body="PR description")
        assert result["success"] is True
        assert "pulls/1" in result["pr_url"]
        mock_gitea.create_pull_request.assert_called_once()


def test_pr_creation_converts_internal_to_external_url():
    """PR URL이 내부 URL에서 외부 URL로 변환된다."""
    mock_gitea = MagicMock()
    mock_gitea.create_pull_request = AsyncMock(
        return_value={
            "number": 5,
            "html_url": "http://gitea:3000/quant/galaxis-po/pulls/5",
        }
    )
    mock_sandbox = MagicMock()
    mock_result = MagicMock(exit_code=0, output="")

    with patch(
        "agent.tools.commit_and_open_pr.get_gitea_client", return_value=mock_gitea
    ), patch(
        "agent.tools.commit_and_open_pr.get_sandbox_backend_sync",
        return_value=mock_sandbox,
    ), patch(
        "agent.tools.commit_and_open_pr.get_config",
        return_value={
            "configurable": {
                "thread_id": "test-thread",
                "repo": {"owner": "quant", "name": "galaxis-po"},
            }
        },
    ), patch(
        "agent.tools.commit_and_open_pr.resolve_repo_dir",
        return_value="/workspace/galaxis-po",
    ), patch(
        "agent.tools.commit_and_open_pr.git_has_uncommitted_changes",
        return_value=True,
    ), patch(
        "agent.tools.commit_and_open_pr.git_fetch_origin",
    ), patch(
        "agent.tools.commit_and_open_pr.git_has_unpushed_commits",
        return_value=False,
    ), patch(
        "agent.tools.commit_and_open_pr.git_current_branch",
        return_value="galaxis-agent/test-thread",
    ), patch(
        "agent.tools.commit_and_open_pr.git_checkout_branch",
    ), patch(
        "agent.tools.commit_and_open_pr.git_config_user",
    ), patch(
        "agent.tools.commit_and_open_pr.git_add_all",
    ), patch(
        "agent.tools.commit_and_open_pr.git_commit",
        return_value=mock_result,
    ), patch(
        "agent.tools.commit_and_open_pr.git_push",
        return_value=mock_result,
    ), patch.dict(
        "os.environ",
        {
            "GITEA_TOKEN": "test-token",
            "GITEA_EXTERNAL_URL": "https://ayuriel.duckdns.org",
            "GITEA_URL": "http://gitea:3000",
        },
    ):
        from agent.tools.commit_and_open_pr import commit_and_open_pr
        result = commit_and_open_pr(title="feat: test", body="body")
        assert result["success"] is True
        assert "ayuriel.duckdns.org" in result["pr_url"]
        assert "gitea:3000" not in result["pr_url"]