# -*- coding: utf-8 -*-
"""Unit tests for execManagedNe arg normalize / detect helpers."""
from __future__ import annotations

from runtime.chat.exec_managed_ne_guard import (
    is_batch_exec_args,
    is_exec_managed_ne_tool,
    normalize_exec_managed_ne_args,
)


def test_is_exec_managed_ne_tool_aliases() -> None:
    assert is_exec_managed_ne_tool("mcp__netx__execManagedNe")
    assert is_exec_managed_ne_tool("netx_exec_managed_ne")
    assert not is_exec_managed_ne_tool("mcp__netx__listCliTargets")


def test_is_batch_exec_args() -> None:
    assert is_batch_exec_args({"ne_ids": ["a", "b"], "commands": ["show version"]})
    assert is_batch_exec_args({"ume_ne_ids": ["u1"]})
    assert is_batch_exec_args({"targets": [{"ne_id": "x"}]})
    assert is_batch_exec_args(
        {
            "targets": [
                {"ume_ne_id": "hw1", "commands": ["display optical-module brief"]},
                {"ume_ne_id": "cs1", "commands": ["show interface transceiver"]},
            ]
        }
    )
    assert not is_batch_exec_args({"ne_id": "x", "commands": ["show version"]})
    assert not is_batch_exec_args({"ne_ids": []})


def test_budget_block_payload_mentions_hetero_targets() -> None:
    from runtime.chat.exec_managed_ne_guard import budget_block_payload

    payload = budget_block_payload(
        reason="call_budget",
        lang="en",
        single_used=6,
        single_budget=6,
    )
    assert "targets=" in str(payload.get("hint") or "")
    assert isinstance(payload.get("example_hetero_targets"), dict)
    assert payload["example_hetero_targets"]["targets"]


def test_normalize_exec_managed_ne_args_defaults_and_clamps() -> None:
    assert normalize_exec_managed_ne_args({})["read_timeout_sec"] == 60
    assert normalize_exec_managed_ne_args({"read_timeout_sec": 5})["read_timeout_sec"] == 10
    assert normalize_exec_managed_ne_args({"read_timeout_sec": 999})["read_timeout_sec"] == 120
    assert normalize_exec_managed_ne_args({"read_timeout_sec": 90})["read_timeout_sec"] == 90