"""The startup "Agent budget" line reports the budget the turn loop enforces (#116888)."""

import logging

import pytest

from gateway.run import GatewayRunner, _bridge_max_turns_to_env, _current_max_iterations
from hermes_cli.config import TURN_LIMIT_UNLIMITED

@pytest.mark.parametrize(
    ("max_turns", "expected"),
    [({"max_turns": 50}, "50"), ({}, "90"), ({"max_turns": "unlimited"}, "unlimited")],
)
def test_agent_budget_line_matches_enforced_limit(monkeypatch, max_turns, expected):
    monkeypatch.setenv("HERMES_MAX_ITERATIONS", "90")  # .env value: config wins when set, else it is the legacy bridge
    monkeypatch.setattr("gateway.run._reload_runtime_env_preserving_config_authority", lambda: None)
    monkeypatch.setattr("gateway.run.get_hermes_home_override", lambda: None)
    _bridge_max_turns_to_env(max_turns)
    enforced = _current_max_iterations()
    records = []
    handler = logging.Handler()
    handler.emit = records.append
    startup_logger = logging.getLogger("gateway.run")
    startup_logger.addHandler(handler)
    old_level = startup_logger.level
    startup_logger.setLevel(logging.INFO)
    try:
        GatewayRunner._log_agent_budget()
    finally:
        startup_logger.removeHandler(handler)
        startup_logger.setLevel(old_level)
    line = [r.getMessage() for r in records if "Agent budget" in r.getMessage()]
    assert line, records
    assert f"max_iterations={expected}" in line[0]
    assert ("unlimited" in line[0]) == (enforced == TURN_LIMIT_UNLIMITED)