tracer-cloud--opensre/tests/tools/test_github_repository_tree_tool.py

"""Tests for GitHubRepositoryTreeTool (function-based, @tool decorated)."""

from __future__ import annotations

from unittest.mock import MagicMock, patch

from integrations.github.tools.repository_tree import get_github_repository_tree
from tests.tools.conftest import BaseToolContract, mock_agent_state


class TestGitHubRepositoryTreeToolContract(BaseToolContract):
    def get_tool_under_test(self):
        return get_github_repository_tree.__opensre_registered_tool__


def test_is_available_requires_owner_repo() -> None:
    rt = get_github_repository_tree.__opensre_registered_tool__
    assert (
        rt.is_available({"github": {"connection_verified": True, "owner": "org", "repo": "repo"}})
        is True
    )
    assert rt.is_available({"github": {"connection_verified": True}}) is False
    assert rt.is_available({}) is False


def test_extract_params_maps_fields() -> None:
    rt = get_github_repository_tree.__opensre_registered_tool__
    sources = mock_agent_state()
    params = rt.extract_params(sources)
    assert params["owner"] == "my-org"
    assert params["repo"] == "my-repo"


def test_run_returns_unavailable_when_no_config() -> None:
    with patch("integrations.github.helpers.github_mcp_config_from_env", return_value=None):
        result = get_github_repository_tree(owner="org", repo="repo")
    assert result["available"] is False


def test_run_happy_path() -> None:
    fake_result = {
        "is_error": False,
        "tool": "get_repository_tree",
        "arguments": {},
        "text": "tree",
        "structured_content": {"tree": [{"path": "src/", "type": "tree"}]},
        "content": [],
    }
    mock_config = MagicMock()
    with (
        patch("integrations.github.helpers.github_mcp_config_from_env", return_value=None),
        patch(
            "integrations.github.helpers.build_github_mcp_config",
            return_value=mock_config,
        ),
        patch(
            "integrations.github.tools.repository_tree.call_github_mcp_tool",
            return_value=fake_result,
        ),
    ):
        result = get_github_repository_tree(
            owner="org",
            repo="repo",
            github_url="http://mcp",
            github_mode="streamable-http",
            github_token="tok",
        )
    assert result["available"] is True
    assert result["tree"] is not None