from __future__ import annotations

from types import SimpleNamespace

import pytest

from opensquilla.engine.runtime import TurnRunner
from opensquilla.engine.types import AgentConfig
from opensquilla.gateway.config import GatewayConfig


class _SessionConfigManager:
    def __init__(self, config: object | None) -> None:
        self.config = config

    def get_session_config(self, session_key: str) -> object | None:
        return self.config


def test_resolve_agent_tool_timeout_prefers_explicit_value(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    monkeypatch.setenv("OPENSQUILLA_AGENT_TOOL_TIMEOUT", "22")
    runner = TurnRunner(
        provider_selector=None,
        session_manager=_SessionConfigManager(
            SimpleNamespace(agent_tool_timeout_seconds=11.0)
        ),
        config=GatewayConfig(agent_tool_timeout_seconds=33.0),
    )

    assert runner._resolve_agent_tool_timeout("agent:main:test", 44.0) == 44.0


def test_resolve_agent_tool_timeout_prefers_session_config(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    monkeypatch.setenv("OPENSQUILLA_AGENT_TOOL_TIMEOUT", "22")
    runner = TurnRunner(
        provider_selector=None,
        session_manager=_SessionConfigManager(
            SimpleNamespace(agent_tool_timeout_seconds=11.0)
        ),
        config=GatewayConfig(agent_tool_timeout_seconds=33.0),
    )

    assert runner._resolve_agent_tool_timeout("agent:main:test") == 11.0


def test_resolve_agent_tool_timeout_prefers_env_over_gateway_config(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    monkeypatch.setenv("OPENSQUILLA_AGENT_TOOL_TIMEOUT", "22")
    runner = TurnRunner(
        provider_selector=None,
        session_manager=_SessionConfigManager(None),
        config=GatewayConfig(agent_tool_timeout_seconds=33.0),
    )

    assert runner._resolve_agent_tool_timeout("agent:main:test") == 22.0


def test_resolve_agent_tool_timeout_uses_gateway_config(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    monkeypatch.delenv("OPENSQUILLA_AGENT_TOOL_TIMEOUT", raising=False)
    runner = TurnRunner(
        provider_selector=None,
        config=GatewayConfig(agent_tool_timeout_seconds=33.0),
    )

    assert runner._resolve_agent_tool_timeout("agent:main:test") == 33.0


def test_resolve_agent_tool_timeout_uses_agent_default_without_config(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    monkeypatch.delenv("OPENSQUILLA_AGENT_TOOL_TIMEOUT", raising=False)
    runner = TurnRunner(provider_selector=None, config=None)

    assert (
        runner._resolve_agent_tool_timeout("agent:main:test")
        == AgentConfig().tool_timeout
    )


def test_resolve_agent_tool_timeout_invalid_env_falls_through(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    monkeypatch.setenv("OPENSQUILLA_AGENT_TOOL_TIMEOUT", "not-a-float")
    runner = TurnRunner(
        provider_selector=None,
        session_manager=_SessionConfigManager(None),
        config=GatewayConfig(agent_tool_timeout_seconds=33.0),
    )

    assert runner._resolve_agent_tool_timeout("agent:main:test") == 33.0


def test_resolve_agent_tool_timeout_accepts_zero(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    monkeypatch.delenv("OPENSQUILLA_AGENT_TOOL_TIMEOUT", raising=False)
    runner = TurnRunner(provider_selector=None, config=GatewayConfig())

    assert runner._resolve_agent_tool_timeout("agent:main:test", 0.0) == 0.0


def test_resolve_agent_tool_timeout_rejects_invalid_explicit_value() -> None:
    runner = TurnRunner(provider_selector=None, config=GatewayConfig())

    with pytest.raises(ValueError, match="tool_timeout"):
        runner._resolve_agent_tool_timeout("agent:main:test", -1.0)