"""装配根与注册表的测试。"""

from __future__ import annotations

import pytest

from app.assembly import build_registry
from app.config import Settings, VolcengineSettings
from app.domain.generation import (
    GenerationRequest,
    GenerationResult,
    MediaKind,
    TaskState,
)
from app.providers.base import GenerationProvider
from app.providers.registry import ProviderRegistry
from app.providers.volcengine.video import PROVIDER_CODE

FAKE_API_KEY = "dummy-api-key-for-tests"


class StubProvider(GenerationProvider):
    def __init__(self, code: str, kind: MediaKind = MediaKind.VIDEO):
        self._code = code
        self._kind = kind

    @property
    def provider_code(self) -> str:
        return self._code

    @property
    def media_kind(self) -> MediaKind:
        return self._kind

    def submit(self, request: GenerationRequest) -> str:
        return "stub-1"

    def fetch(self, provider_request_id: str) -> GenerationResult:
        return GenerationResult(provider_request_id, TaskState.RUNNING)


class TestRegistry:
    def test_finds_by_code(self):
        registry = ProviderRegistry([StubProvider("a"), StubProvider("b")])
        assert registry.find("a").provider_code == "a"
        assert registry.codes == frozenset({"a", "b"})

    def test_unknown_code_returns_none(self):
        """对已持久化的任务，未知 code 意味着当初提交它的适配器已被移除。

        调用方应当作不可重试的失败，而不是循环重试到预算耗尽。
        """
        assert ProviderRegistry([StubProvider("a")]).find("gone") is None

    def test_duplicate_registration_is_rejected(self):
        # 静默覆盖会让「哪个实例在处理任务」取决于注册顺序，是最难查的一类问题。
        with pytest.raises(ValueError, match="重复注册"):
            ProviderRegistry([StubProvider("same"), StubProvider("same")])


class TestAssembly:
    def test_seedance_is_registered_when_credentials_exist(self):
        registry = build_registry(
            Settings(volcengine=VolcengineSettings(api_key=FAKE_API_KEY))
        )
        provider = registry.find(PROVIDER_CODE)
        assert provider is not None
        assert provider.media_kind is MediaKind.VIDEO

    def test_missing_credentials_skip_the_provider_instead_of_failing_startup(self):
        """本地只配了一家凭证时，服务仍应起得来。

        真正的失败点在调用时：find() 返回 None，调用方按「适配器不可用」处理，
        而不是拿到一个半死的实例。
        """
        registry = build_registry(Settings(volcengine=VolcengineSettings(api_key="")))
        assert registry.find(PROVIDER_CODE) is None
        assert len(registry) == 0

    def test_registered_code_matches_the_catalog_column(self):
        """注册表的键必须与 creation.generation_models.provider_code 同值。

        对不上时，Java 侧按任务行的快照来查会一直查不到适配器——而症状是任务
        永远卡在排队，离根因很远。
        """
        registry = build_registry(
            Settings(volcengine=VolcengineSettings(api_key=FAKE_API_KEY))
        )
        assert PROVIDER_CODE in registry.codes