"""Unit tests for decepticon.llm.router — thin wrapper over LLMModelMapping."""

import pytest

from decepticon.llm.router import ModelRouter
from decepticon_core.types.llm import (
    AuthMethod,
    Credentials,
    LLMModelMapping,
    ModelAssignment,
    ModelProfile,
)


class TestModelRouter:
    def setup_method(self):
        # Default: all four API methods available, eco profile.
        self.mapping = LLMModelMapping.from_profile(ModelProfile.ECO)
        self.router = ModelRouter(self.mapping)

    def test_resolve_returns_primary(self):
        # recon is LOW; first method is anthropic_api → haiku.
        assert self.router.resolve("recon") == "anthropic/claude-haiku-4-5"

    def test_resolve_decepticon_is_high_tier(self):
        assert self.router.resolve("decepticon") == "anthropic/claude-opus-4-8"

    def test_resolve_with_fallback_returns_chain(self):
        chain = self.router.resolve_with_fallback("recon")
        assert chain == [
            "anthropic/claude-haiku-4-5",
            "openai/gpt-5-nano",
            "gemini/gemini-2.5-flash-lite",
            "deepseek/deepseek-v4-flash",
            "openrouter/anthropic/claude-haiku-4-5",
            "nvidia_nim/meta/llama-3.2-3b-instruct",
        ]

    def test_resolve_with_fallback_high_tier_full_chain(self):
        chain = self.router.resolve_with_fallback("decepticon")
        assert chain == [
            "anthropic/claude-opus-4-8",
            "openai/gpt-5.5",
            "gemini/gemini-2.5-pro",
            "minimax/MiniMax-M3",
            "deepseek/deepseek-v4-pro",
            "xai/grok-4.3",
            "mistral/mistral-large-latest",
            "openrouter/anthropic/claude-opus-4-8",
            "nvidia_nim/meta/llama-3.3-70b-instruct",
        ]

    def test_resolve_unknown_role_raises(self):
        with pytest.raises(KeyError, match="No model assignment"):
            self.router.resolve("nonexistent_role")

    def test_get_assignment_returns_full_config(self):
        a = self.router.get_assignment("recon")
        assert isinstance(a, ModelAssignment)
        assert a.primary == "anthropic/claude-haiku-4-5"
        assert a.temperature == 0.3

    def test_resolve_with_single_credential_no_fallback(self):
        creds = Credentials(methods=[AuthMethod.OPENAI_API])
        router = ModelRouter(LLMModelMapping.from_credentials_and_profile(creds))
        chain = router.resolve_with_fallback("decepticon")
        assert chain == ["openai/gpt-5.5"]