from __future__ import annotations

from cloud.planner_config import (
    DEFAULT_MODEL_BY_PROVIDER,
    DEFAULT_PROVIDER,
    DEFAULT_TIMEOUT_SECONDS,
    CloudPlannerConfig,
    build_cloud_planner_client,
    load_cloud_planner_config,
)
from runtime.tool_calling_client import (
    AnthropicToolCallingClient,
    OpenAIToolCallingClient,
)

_NO_RELEVANT_VARS = {"UNRELATED": "1"}


def test_load_cloud_planner_config_defaults_when_unset() -> None:
    config = load_cloud_planner_config(_NO_RELEVANT_VARS)

    assert config == CloudPlannerConfig(
        provider=DEFAULT_PROVIDER,
        model="",
        timeout=DEFAULT_TIMEOUT_SECONDS,
    )
    assert config.resolved_model() == DEFAULT_MODEL_BY_PROVIDER[DEFAULT_PROVIDER]


def test_load_cloud_planner_config_selects_provider_and_resolves_default_model() -> (
    None
):
    config = load_cloud_planner_config({"AI_PLANNER_PROVIDER": "openai"})

    assert config.provider == "openai"
    assert config.resolved_model() == "gpt-5.6"


def test_load_cloud_planner_config_falls_back_to_default_provider_when_unsupported() -> (
    None
):
    config = load_cloud_planner_config({"AI_PLANNER_PROVIDER": "not-a-real-provider"})

    assert config.provider == DEFAULT_PROVIDER


def test_load_cloud_planner_config_model_override_wins_regardless_of_provider() -> None:
    config = load_cloud_planner_config(
        {"AI_PLANNER_PROVIDER": "openai", "AI_PLANNER_MODEL": "custom-model"}
    )

    assert config.resolved_model() == "custom-model"


def test_load_cloud_planner_config_parses_valid_timeout() -> None:
    config = load_cloud_planner_config({"AI_PLANNER_TIMEOUT_SECONDS": "12.5"})

    assert config.timeout == 12.5


def test_load_cloud_planner_config_falls_back_to_default_timeout_when_invalid() -> None:
    for value in ["not-a-number", "0", "-5"]:
        config = load_cloud_planner_config(
            {"AI_PLANNER_TIMEOUT_SECONDS": value, **_NO_RELEVANT_VARS}
        )
        assert config.timeout == DEFAULT_TIMEOUT_SECONDS


def test_build_cloud_planner_client_selects_anthropic_by_default() -> None:
    client = build_cloud_planner_client(CloudPlannerConfig())

    assert isinstance(client, AnthropicToolCallingClient)
    assert client.model == DEFAULT_MODEL_BY_PROVIDER["anthropic"]


def test_build_cloud_planner_client_selects_openai() -> None:
    client = build_cloud_planner_client(CloudPlannerConfig(provider="openai"))

    assert isinstance(client, OpenAIToolCallingClient)
    assert client.model == DEFAULT_MODEL_BY_PROVIDER["openai"]