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

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

from __future__ import annotations

from unittest.mock import patch

from integrations.sentry.tools.sentry_issue_events_tool import list_sentry_issue_events
from tests.tools.conftest import BaseToolContract, mock_agent_state


class TestSentryIssueEventsToolContract(BaseToolContract):
    def get_tool_under_test(self):
        return list_sentry_issue_events.__opensre_registered_tool__


def test_is_available_requires_issue_id() -> None:
    rt = list_sentry_issue_events.__opensre_registered_tool__
    assert rt.is_available({"sentry": {"connection_verified": True, "issue_id": "123"}}) is True
    assert rt.is_available({"sentry": {"connection_verified": True}}) is False
    assert rt.is_available({}) is False


def test_extract_params_maps_fields() -> None:
    rt = list_sentry_issue_events.__opensre_registered_tool__
    sources = mock_agent_state()
    params = rt.extract_params(sources)
    assert params["issue_id"] == "12345"
    assert params["organization_slug"] == "my-org"


def test_run_returns_unavailable_when_no_config() -> None:
    result = list_sentry_issue_events(organization_slug="", sentry_token="", issue_id="123")
    assert result["available"] is False
    assert result["events"] == []


def test_run_happy_path() -> None:
    fake_events = [{"eventID": "e1", "dateCreated": "2024-01-01"}]
    with (
        patch(
            "integrations.sentry.tools.sentry_issue_events_tool.sentry_list_issue_events",
            return_value=fake_events,
        ),
        patch(
            "integrations.sentry.tools.sentry_search_issues_tool.sentry_config_from_env",
            return_value=None,
        ),
    ):
        result = list_sentry_issue_events(
            organization_slug="my-org", sentry_token="tok_test", issue_id="123"
        )
    assert result["available"] is True
    assert len(result["events"]) == 1