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

from __future__ import annotations

from unittest.mock import MagicMock, patch

import pytest

from integrations.vercel.tools import VercelLogsTool


@pytest.fixture()
def tool() -> VercelLogsTool:
    return VercelLogsTool()


def test_is_available_requires_only_connection_verified(tool: VercelLogsTool) -> None:
    # deployment_id is no longer required for is_available; it's checked inside run()
    assert (
        tool.is_available({"vercel": {"connection_verified": True, "deployment_id": "dpl_1"}})
        is True
    )
    assert tool.is_available({"vercel": {"connection_verified": True}}) is True
    assert tool.is_available({"vercel": {"deployment_id": "dpl_1"}}) is False
    assert tool.is_available({"vercel": {}}) is False
    assert tool.is_available({}) is False


def test_extract_params_maps_source_fields(tool: VercelLogsTool) -> None:
    params = tool.extract_params(
        {
            "vercel": {
                "api_token": "tok_abc",
                "team_id": "team_1",
                "project_id": "prj_1",
                "deployment_id": "dpl_xyz",
                "connection_verified": True,
            }
        }
    )
    assert params["api_token"] == "tok_abc"
    assert params["project_id"] == "prj_1"
    assert params["deployment_id"] == "dpl_xyz"
    assert params["include_runtime_logs"] is True


def test_run_returns_events_and_filters_error_events(tool: VercelLogsTool) -> None:
    events = [
        {"type": "stdout", "text": "Building dependencies...", "created": 1},
        {"type": "stderr", "text": "Error: cannot find module 'react'", "created": 2},
        {"type": "stdout", "text": "Build complete", "created": 3},
        {"type": "stderr", "text": "exception in handler", "created": 4},
    ]
    mock_client = MagicMock()
    mock_client.get_deployment.return_value = {
        "success": True,
        "deployment": {"id": "dpl_xyz", "state": "ERROR"},
    }
    mock_client.get_deployment_events.return_value = {"success": True, "events": events}
    mock_client.get_runtime_logs.return_value = {"success": True, "logs": []}

    with patch("integrations.vercel.tools.make_vercel_client", return_value=mock_client):
        result = tool.run(api_token="tok_test", deployment_id="dpl_xyz")

    assert result["available"] is True
    assert len(result["events"]) == 4
    assert len(result["error_events"]) == 2
    error_texts = {e["text"] for e in result["error_events"]}
    assert any("Error" in t for t in error_texts)
    assert any("exception" in t for t in error_texts)


def test_run_skips_runtime_logs_when_disabled(tool: VercelLogsTool) -> None:
    mock_client = MagicMock()
    mock_client.get_deployment.return_value = {"success": True, "deployment": {}}
    mock_client.get_deployment_events.return_value = {"success": True, "events": []}

    with patch("integrations.vercel.tools.make_vercel_client", return_value=mock_client):
        tool.run(api_token="tok_test", deployment_id="dpl_xyz", include_runtime_logs=False)

    mock_client.get_runtime_logs.assert_not_called()


def test_run_includes_runtime_logs_by_default(tool: VercelLogsTool) -> None:
    mock_client = MagicMock()
    mock_client.get_deployment.return_value = {"success": True, "deployment": {}}
    mock_client.get_deployment_events.return_value = {"success": True, "events": []}
    mock_client.get_runtime_logs.return_value = {
        "success": True,
        "logs": [{"id": "l1"}, {"id": "l2"}],
    }

    with patch("integrations.vercel.tools.make_vercel_client", return_value=mock_client):
        result = tool.run(api_token="tok_test", deployment_id="dpl_xyz", project_id="prj_9")

    mock_client.get_runtime_logs.assert_called_once()
    assert mock_client.get_runtime_logs.call_args.kwargs.get("project_id") == "prj_9"
    assert result["total_runtime_logs"] == 2
    assert len(result["runtime_logs"]) == 2


def test_run_gracefully_handles_deployment_fetch_failure(tool: VercelLogsTool) -> None:
    mock_client = MagicMock()
    mock_client.get_deployment.return_value = {"success": False, "error": "not found"}
    mock_client.get_deployment_events.return_value = {"success": True, "events": []}
    mock_client.get_runtime_logs.return_value = {"success": True, "logs": []}

    with patch("integrations.vercel.tools.make_vercel_client", return_value=mock_client):
        result = tool.run(api_token="tok_test", deployment_id="dpl_xyz")

    assert result["available"] is True
    assert result["deployment"] == {}


def test_run_gracefully_handles_events_fetch_failure(tool: VercelLogsTool) -> None:
    mock_client = MagicMock()
    mock_client.get_deployment.return_value = {"success": True, "deployment": {"id": "dpl_xyz"}}
    mock_client.get_deployment_events.return_value = {"success": False, "error": "rate limited"}
    mock_client.get_runtime_logs.return_value = {"success": True, "logs": []}

    with patch("integrations.vercel.tools.make_vercel_client", return_value=mock_client):
        result = tool.run(api_token="tok_test", deployment_id="dpl_xyz")

    assert result["available"] is True
    assert result["events"] == []
    assert result["error_events"] == []


def test_run_returns_unavailable_without_token(tool: VercelLogsTool) -> None:
    result = tool.run(api_token="", deployment_id="dpl_xyz")
    assert result["available"] is False
    assert result["events"] == []
    assert result["runtime_logs"] == []


def test_run_returns_unavailable_for_whitespace_only_token(tool: VercelLogsTool) -> None:
    result = tool.run(api_token="\t  \n", deployment_id="dpl_xyz")
    assert result["available"] is False
    assert result["events"] == []


def test_run_returns_informative_error_without_deployment_id(tool: VercelLogsTool) -> None:
    result = tool.run(api_token="tok_test", deployment_id="")
    assert result["available"] is False
    assert "deployment_id" in result["error"]
    assert "vercel_deployment_status" in result["error"]


def test_metadata_requires_deployment_id(tool: VercelLogsTool) -> None:
    meta = tool.metadata()
    assert meta.name == "vercel_deployment_logs"
    assert meta.source == "vercel"
    assert "deployment_id" in meta.input_schema["required"]
    assert "api_token" in meta.input_schema["required"]