xiaoxia-saas/tests/unit/test_1970_assembly_schema.py

"""#1970 PR3 schema 校验 + 路由辅助函数测试。"""

from __future__ import annotations

from dataclasses import dataclass, field
from types import SimpleNamespace
from unittest.mock import MagicMock

import pytest
from app.api.routes import generation_tasks as gt
from app.schemas.generation_task import CreateGenerationTaskRequest
from pydantic import ValidationError

# ── schema ─────────────────────────────────────────────────────────────────


def _base_payload(**overrides):
    payload = dict(
        template_id="tpl1",
        asset_ids=["a1", "a2"],
        duration=30,
        title_text="t",
        editing_mode="voice_over",
    )
    payload.update(overrides)
    return payload


class TestAssemblySchema:
    def test_defaults(self):
        req = CreateGenerationTaskRequest(**_base_payload())
        assert req.assembly_mode == "random"
        assert req.script_id == ""
        assert req.tts_voice_id == ""
        assert req.tts_voice_source == "preset"
        assert req.video_ratio == ""  # 空串=沿用模板默认（前端新流程显式传 9:16）
        assert req.dedup_enabled is True

    def test_narrative_accepts_fields(self):
        req = CreateGenerationTaskRequest(
            **_base_payload(
                assembly_mode="narrative",
                script_id="s1",
                tts_voice_id="longxiaochun",
                tts_voice_source="clone",
                video_ratio="16:9",
            )
        )
        assert req.assembly_mode == "narrative"
        assert req.script_id == "s1"

    def test_bad_assembly_mode_rejected(self):
        with pytest.raises(ValidationError):
            CreateGenerationTaskRequest(**_base_payload(assembly_mode="movie"))

    def test_bad_voice_source_rejected(self):
        with pytest.raises(ValidationError):
            CreateGenerationTaskRequest(**_base_payload(tts_voice_source="elevenlabs"))

    def test_bad_video_ratio_rejected(self):
        with pytest.raises(ValidationError):
            CreateGenerationTaskRequest(**_base_payload(video_ratio="4:5"))

    def test_narrative_without_script_rejected(self):
        with pytest.raises(ValidationError) as ei:
            CreateGenerationTaskRequest(**_base_payload(assembly_mode="narrative"))
        assert "script_id" in str(ei.value)

    def test_narrative_without_voice_rejected(self):
        with pytest.raises(ValidationError) as ei:
            CreateGenerationTaskRequest(**_base_payload(assembly_mode="narrative", script_id="s1"))
        assert "tts_voice_id" in str(ei.value)

    def test_random_mode_ignores_script_absence(self):
        req = CreateGenerationTaskRequest(**_base_payload())
        assert req.assembly_mode == "random"


# ── _select_assets_from_library 的叙事分支 ─────────────────────────────────


@dataclass
class _Asset:
    id: str
    status: object = field(default_factory=lambda: SimpleNamespace(value="ready"))
    mime_type: str = "video/mp4"
    tags: list[str] = field(default_factory=list)
    tag_ids: list[str] = field(default_factory=list)
    file_type: str = "video"
    quality_score: float | None = None
    duration: float = 8.0
    created_at: object = None
    metadata: dict = field(default_factory=dict)


class TestNarrativeSelectInRoute:
    def test_narrative_tags_prioritize_matched(self):
        assets = [
            _Asset("a1", tags=["工厂"]),
            _Asset("a2", tags=["旅游"]),
            _Asset("a3", tags=["工厂"]),
        ]
        picked = gt._select_assets_from_library(assets, mode="all", count=2, script_tags=["工厂"])
        assert set(picked) == {"a1", "a3"}

    def test_narrative_no_match_falls_back_to_full_pool(self):
        assets = [_Asset("a1", tags=["工厂"]), _Asset("a2", tags=["旅游"])]
        picked = gt._select_assets_from_library(assets, mode="all", count=2, script_tags=["美食"])
        assert set(picked) == {"a1", "a2"}

    def test_tag_ids_via_index(self):
        assets = [_Asset("a1", tag_ids=["t1"]), _Asset("a2", tag_ids=["t2"])]
        picked = gt._select_assets_from_library(
            assets,
            mode="all",
            count=1,
            script_tags=["教程"],
            tag_names_by_id={"a1": ["教程"], "a2": ["旅游"]},
        )
        assert picked == ["a1"]

    def test_no_script_tags_smart_path_unchanged(self):
        assets = [_Asset("a1"), _Asset("a2")]
        picked = gt._select_assets_from_library(assets, mode="smart", count=1)
        assert picked  # 非空即可，评分逻辑由 smart_match 自己的测试覆盖


# ── _load_asset_tag_names（DB 替身） ────────────────────────────────────────


class _FakeRow:
    def __init__(self, **kw):
        self.__dict__.update(kw)


class _FakeQuery:
    def __init__(self, rows):
        self._rows = rows

    def filter(self, *a, **k):
        return self

    def all(self):
        return self._rows


class _FakeDb:
    def __init__(self, name_rows, link_rows):
        self._maps = {
            "names": name_rows,
            "links": link_rows,
        }

    def query(self, *cols):
        # _load_asset_tag_names 两次查询：第一次取 (id, name)，第二次取 (asset_id, tag_id)
        keys = tuple(getattr(c, "key", None) for c in cols)
        if keys and keys[0] == "id":
            return _FakeQuery(self._maps["names"])
        return _FakeQuery(self._maps["links"])


@dataclass
class _TagIdAsset:
    id: str
    tag_ids: list[str]


class TestLoadAssetTagNames:
    def test_builds_index(self):
        assets = [_TagIdAsset("a1", ["t1", "t2"]), _TagIdAsset("a2", ["t2"])]
        db = _FakeDb(
            name_rows=[_FakeRow(id="t1", name="工厂"), _FakeRow(id="t2", name="带货")],
            link_rows=[
                ("a1", "t1"),
                ("a1", "t2"),
                ("a2", "t2"),
            ],
        )
        idx = gt._load_asset_tag_names(db, assets, "u1")
        assert idx == {"a1": ["工厂", "带货"], "a2": ["带货"]}

    def test_no_tag_ids_returns_empty(self):
        assert gt._load_asset_tag_names(_FakeDb([], []), [_TagIdAsset("a1", [])], "u1") == {}

    def test_query_failure_degrades_empty(self):
        class BoomQuery:
            def filter(self, *a, **k):
                raise RuntimeError("db down")

        class BoomDb:
            def query(self, *a):
                return BoomQuery()

        idx = gt._load_asset_tag_names(BoomDb(), [_TagIdAsset("a1", ["t1"])], "u1")
        assert idx == {}


# ── _resolve_output_dimensions ─────────────────────────────────────────────


class TestResolveOutputDimensions:
    def _req(self, ratio="", width=1280, height=720):
        return CreateGenerationTaskRequest(**_base_payload(video_ratio=ratio, output_width=width, output_height=height))

    def test_known_ratios(self):
        assert gt._resolve_output_dimensions(self._req("9:16")) == (1080, 1920)
        assert gt._resolve_output_dimensions(self._req("16:9")) == (1920, 1080)
        assert gt._resolve_output_dimensions(self._req("1:1")) == (1080, 1080)
        assert gt._resolve_output_dimensions(self._req("4:3")) == (1440, 1080)
        assert gt._resolve_output_dimensions(self._req("3:4")) == (1080, 1440)

    def test_old_call_default_kept_when_no_ratio(self):
        assert gt._resolve_output_dimensions(self._req("")) == (1280, 720)

    def test_explicit_dimensions_take_precedence(self):
        # 非旧默认值（720p）的显式分辨率优先于 ratio 映射
        req = self._req("9:16", width=1440, height=2560)
        assert gt._resolve_output_dimensions(req) == (1440, 2560)