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

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

from __future__ import annotations

from unittest.mock import MagicMock, patch

from integrations.grafana.tools import query_grafana_metrics
from platform.common.metric_summary import summarize_prometheus_metrics
from tests.synthetic.mock_grafana_backend.backend import FixtureGrafanaBackend
from tests.synthetic.rds_postgres.scenario_loader import SUITE_DIR, load_scenario
from tests.tools.conftest import BaseToolContract, mock_agent_state


class TestGrafanaMetricsToolContract(BaseToolContract):
    def get_tool_under_test(self):
        return query_grafana_metrics.__opensre_registered_tool__


def test_is_available_requires_grafana_creds() -> None:
    rt = query_grafana_metrics.__opensre_registered_tool__
    assert rt.is_available({"grafana": {"connection_verified": True}}) is True
    assert rt.is_available({"grafana": {}}) is False
    assert rt.is_available({}) is False


def test_extract_params_maps_fields() -> None:
    rt = query_grafana_metrics.__opensre_registered_tool__
    sources = mock_agent_state()
    params = rt.extract_params(sources)
    assert "metric_name" in params
    assert params["grafana_endpoint"] == "https://grafana.example.com"


def test_extract_params_includes_basic_auth_fields() -> None:
    rt = query_grafana_metrics.__opensre_registered_tool__
    sources = mock_agent_state({"grafana": {"username": "local-user", "password": "local-pass"}})
    params = rt.extract_params(sources)
    assert params["grafana_username"] == "local-user"
    assert params["grafana_password"] == "local-pass"


def test_injected_params_include_basic_auth_fields() -> None:
    rt = query_grafana_metrics.__opensre_registered_tool__
    assert "grafana_username" in rt.injected_params
    assert "grafana_password" in rt.injected_params


def test_run_with_backend() -> None:
    mock_backend = MagicMock()
    mock_backend.query_timeseries.return_value = {
        "data": {"result": [{"metric": {}, "values": [[1000, "42"]]}]}
    }
    result = query_grafana_metrics(metric_name="pipeline_runs_total", grafana_backend=mock_backend)
    assert result["available"] is True
    assert result["total_series"] == 1


def test_rds_storage_fixture_metrics_have_compact_summaries() -> None:
    fixture = load_scenario(SUITE_DIR / "003-storage-full")
    backend = FixtureGrafanaBackend(fixture)

    result = query_grafana_metrics(
        metric_name="pipeline_runs_total",
        service_name="rds-postgres-synthetic",
        grafana_backend=backend,
    )
    summaries = summarize_prometheus_metrics(result["metrics"])

    by_name = {summary["metric_name"]: summary for summary in summaries}
    assert "FreeStorageSpace" in by_name
    assert "WriteIOPS" in by_name
    assert "orders-prod" in by_name["FreeStorageSpace"]["summary"]
    assert "decreased" in by_name["FreeStorageSpace"]["trend"]
    assert "orders-prod" in by_name["WriteIOPS"]["summary"]
    assert "8100" in by_name["WriteIOPS"]["summary"]
    assert "peak_to_latest" in by_name["WriteIOPS"]["summary"]


def test_run_no_client() -> None:
    mock_client = MagicMock()
    mock_client.is_configured = False
    with patch("integrations.grafana.tools._resolve_grafana_client", return_value=mock_client):
        result = query_grafana_metrics(metric_name="cpu_usage", grafana_endpoint="http://grafana")
    assert result["available"] is False


def test_run_no_mimir_datasource() -> None:
    mock_client = MagicMock()
    mock_client.is_configured = True
    mock_client.mimir_datasource_uid = None
    with patch("integrations.grafana.tools._resolve_grafana_client", return_value=mock_client):
        result = query_grafana_metrics(metric_name="cpu_usage", grafana_endpoint="http://grafana")
    assert result["available"] is False
    assert "Mimir" in result["error"]


def test_run_happy_path() -> None:
    mock_client = MagicMock()
    mock_client.is_configured = True
    mock_client.mimir_datasource_uid = "mimir-uid"
    mock_client.account_id = "acc-1"
    mock_client.query_mimir.return_value = {
        "success": True,
        "metrics": [{"name": "pipeline_runs_total"}],
        "total_series": 1,
    }
    with patch("integrations.grafana.tools._resolve_grafana_client", return_value=mock_client):
        result = query_grafana_metrics(
            metric_name="pipeline_runs_total", grafana_endpoint="http://grafana"
        )
    assert result["available"] is True
    assert result["total_series"] == 1