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

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

from __future__ import annotations

from integrations.datadog.tools import query_datadog_metrics
from tests.tools.conftest import BaseToolContract, mock_agent_state


class TestDataDogMetricsToolContract(BaseToolContract):
    def get_tool_under_test(self):
        return query_datadog_metrics.__opensre_registered_tool__


def test_is_available_returns_false_until_implemented() -> None:
    # Stub is hidden from the planner (see issue #669) so that the LLM never
    # burns a tool-call budget slot on a tool that always returns a "not yet
    # implemented" error.  Flip this expectation back to the connection_verified
    # gate once the Metrics API v2 body is in place.
    rt = query_datadog_metrics.__opensre_registered_tool__
    assert rt.is_available({"datadog": {"connection_verified": True}}) is False
    assert rt.is_available({"datadog": {}}) is False
    assert rt.is_available({}) is False


def test_extract_params_maps_fields() -> None:
    rt = query_datadog_metrics.__opensre_registered_tool__
    sources = mock_agent_state()
    params = rt.extract_params(sources)
    assert "metric_name" in params
    assert params["api_key"] == "dd_api_key_test"


def test_run_returns_stub_unavailable() -> None:
    # This tool is a stub
    result = query_datadog_metrics(metric_name="system.cpu.user")
    assert result["available"] is False
    assert result["metric_name"] == "system.cpu.user"
    assert result["metrics"] == []


def test_run_metadata() -> None:
    rt = query_datadog_metrics.__opensre_registered_tool__
    assert rt.name == "query_datadog_metrics"
    assert rt.source == "datadog"