xiaoxia-saas/tests/unit/test_template_use_cases.py

"""
Template Use Cases 单元测试 — 剪辑计划模板 CRUD + 业务规则校验
"""

from unittest.mock import Mock

import pytest

from packages.application.template.commands import (
    CreateCategoryCommand,
    CreateTemplateCommand,
    SegmentCommand,
    UpdateTemplateCommand,
    ValidateTemplateCommand,
)
from packages.application.template.use_cases import (
    CreateCategoryUseCase,
    CreateTemplateUseCase,
    DeleteTemplateUseCase,
    GetTemplateUseCase,
    ListCategoriesUseCase,
    ListTemplatesUseCase,
    NotFoundError,
    UpdateTemplateUseCase,
    ValidateTemplateUseCase,
    ValidationError,
)
from packages.domain.template import Template, TemplateCategory, TemplateSegment


def _make_repo():
    """创建一个 mock repository."""
    repo = Mock()
    repo.list_by_user = Mock(return_value=[])
    repo.get = Mock(return_value=None)
    repo.create = Mock()
    repo.update = Mock()
    repo.delete = Mock(return_value=False)
    repo.count_by_user = Mock(return_value=0)
    repo.list_segments = Mock(return_value=[])
    repo.create_segments = Mock()
    repo.delete_segments_by_template = Mock(return_value=0)
    repo.list_categories = Mock(return_value=[])
    repo.create_category = Mock()
    repo.get_category = Mock(return_value=None)
    repo.delete_category = Mock(return_value=False)
    return repo


def _make_template(**kwargs) -> Template:
    defaults = dict(
        id="tmpl-001",
        user_id="user-001",
        name="测试模板",
        mode="pip",
        category="default",
        tags=["test"],
        title_config={"ai_auto_select": True},
        subtitle_config={"enabled": True},
        bgm_config={"enabled": False},
        estimated_duration=60.0,
        segments=[],
    )
    defaults.update(kwargs)
    return Template(**defaults)


# ── CreateTemplateUseCase ──


class TestCreateTemplateUseCase:
    @pytest.fixture
    def repo(self):
        return _make_repo()

    @pytest.fixture
    def use_case(self, repo):
        return CreateTemplateUseCase(repo)

    def test_create_basic_template(self, use_case, repo):
        """创建基础模板（无片段）."""
        repo.create.side_effect = lambda t: t  # 返回传入的 template

        command = CreateTemplateCommand(
            user_id="user-001",
            name="画中画模板",
            mode="pip",
            category="vlog",
            tags=["vlog", "pip"],
            estimated_duration=90.0,
        )
        result = use_case.execute(command)

        assert result.name == "画中画模板"
        assert result.mode == "pip"
        assert result.user_id == "user-001"
        repo.create.assert_called_once()

    def test_create_with_segments(self, use_case, repo):
        """创建模板并附带片段."""
        repo.create.side_effect = lambda t: t
        repo.create_segments.side_effect = lambda segs: segs

        command = CreateTemplateCommand(
            user_id="user-001",
            name="口播混剪模板",
            mode="voice_over",
            segments=[
                SegmentCommand(segment_order=1, duration_min=5, duration_max=15, material_type="人物"),
                SegmentCommand(segment_order=2, duration_min=10, duration_max=30, material_type="场景"),
            ],
        )
        result = use_case.execute(command)

        assert len(result.segments) == 2
        assert result.segments[0].material_type == "人物"
        repo.create_segments.assert_called_once()

    def test_create_invalid_mode_raises(self, use_case):
        """无效剪辑模式应抛出 ValidationError."""
        command = CreateTemplateCommand(
            user_id="user-001",
            name="无效模板",
            mode="invalid_mode",
        )
        with pytest.raises(ValidationError, match="无效的剪辑模式"):
            use_case.execute(command)


# ── UpdateTemplateUseCase ──


class TestUpdateTemplateUseCase:
    @pytest.fixture
    def repo(self):
        return _make_repo()

    @pytest.fixture
    def use_case(self, repo):
        return UpdateTemplateUseCase(repo)

    def test_update_name(self, use_case, repo):
        """更新模板名称."""
        existing = _make_template()
        repo.get.return_value = existing
        repo.update.side_effect = lambda t: t

        command = UpdateTemplateCommand(
            template_id="tmpl-001",
            user_id="user-001",
            name="新名称",
        )
        result = use_case.execute(command)

        assert result.name == "新名称"
        repo.update.assert_called_once()

    def test_update_not_found_raises(self, use_case, repo):
        """模板不存在时抛出 NotFoundError."""
        repo.get.return_value = None

        command = UpdateTemplateCommand(
            template_id="nonexistent",
            user_id="user-001",
            name="新名称",
        )
        with pytest.raises(NotFoundError):
            use_case.execute(command)

    def test_update_invalid_mode_raises(self, use_case, repo):
        """更新为无效模式时抛出 ValidationError."""
        existing = _make_template()
        repo.get.return_value = existing

        command = UpdateTemplateCommand(
            template_id="tmpl-001",
            user_id="user-001",
            mode="bad_mode",
        )
        with pytest.raises(ValidationError, match="无效的剪辑模式"):
            use_case.execute(command)

    def test_replace_segments(self, use_case, repo):
        """替换片段列表."""
        existing = _make_template()
        repo.get.return_value = existing
        repo.update.side_effect = lambda t: t
        repo.create_segments.side_effect = lambda segs: segs

        command = UpdateTemplateCommand(
            template_id="tmpl-001",
            user_id="user-001",
            segments=[
                SegmentCommand(segment_order=1, duration_min=5, duration_max=20, material_type=None),
            ],
        )
        result = use_case.execute(command)

        repo.delete_segments_by_template.assert_called_once_with("tmpl-001")
        repo.create_segments.assert_called_once()
        assert len(result.segments) == 1


# ── ValidateTemplateUseCase — 业务规则校验 ──


class TestValidateTemplateUseCase:
    @pytest.fixture
    def repo(self):
        return _make_repo()

    @pytest.fixture
    def use_case(self, repo):
        return ValidateTemplateUseCase(repo)

    def test_one_take_with_one_segment_ok(self, use_case, repo):
        """一镜到底 + 恰好 1 个片段 → 通过."""
        seg = TemplateSegment(
            id="seg-001",
            template_id="tmpl-001",
            segment_order=1,
            duration_min=0,
            duration_max=60,
        )
        template = _make_template(mode="one_take", segments=[seg])
        repo.get.return_value = template

        command = ValidateTemplateCommand(
            template_id="tmpl-001",
            user_id="user-001",
        )
        result = use_case.execute(command)

        assert result.template.mode == "one_take"
        assert result.warnings == []

    def test_one_take_with_two_segments_raises(self, use_case, repo):
        """一镜到底 + 2 个片段 → ValidationError."""
        segs = [
            TemplateSegment(id=f"seg-{i}", template_id="tmpl-001", segment_order=i, duration_min=0, duration_max=30)
            for i in (1, 2)
        ]
        template = _make_template(mode="one_take", segments=segs)
        repo.get.return_value = template

        command = ValidateTemplateCommand(
            template_id="tmpl-001",
            user_id="user-001",
        )
        with pytest.raises(ValidationError, match="一镜到底模式必须恰好有 1 个片段"):
            use_case.execute(command)

    def test_voice_over_all_segments_have_material_type_ok(self, use_case, repo):
        """口播+B-roll + 所有片段都有 material_type → 通过."""
        segs = [
            TemplateSegment(
                id="seg-1",
                template_id="tmpl-001",
                segment_order=1,
                duration_min=5,
                duration_max=15,
                material_type="人物",
            ),
            TemplateSegment(
                id="seg-2",
                template_id="tmpl-001",
                segment_order=2,
                duration_min=10,
                duration_max=30,
                material_type="场景",
            ),
        ]
        template = _make_template(mode="voice_over", segments=segs)
        repo.get.return_value = template

        command = ValidateTemplateCommand(
            template_id="tmpl-001",
            user_id="user-001",
        )
        result = use_case.execute(command)
        assert result.warnings == []

    def test_voice_over_missing_material_type_raises(self, use_case, repo):
        """口播+B-roll + 某片段缺少 material_type → ValidationError."""
        segs = [
            TemplateSegment(
                id="seg-1",
                template_id="tmpl-001",
                segment_order=1,
                duration_min=5,
                duration_max=15,
                material_type="人物",
            ),
            TemplateSegment(
                id="seg-2",
                template_id="tmpl-001",
                segment_order=2,
                duration_min=10,
                duration_max=30,
                material_type=None,
            ),  # 缺失
        ]
        template = _make_template(mode="voice_over", segments=segs)
        repo.get.return_value = template

        command = ValidateTemplateCommand(
            template_id="tmpl-001",
            user_id="user-001",
        )
        with pytest.raises(ValidationError, match="material_type"):
            use_case.execute(command)

    def test_voiceover_duration_within_tolerance_no_warning(self, use_case, repo):
        """配音时长在 ±30% 以内 → 无警告."""
        template = _make_template(estimated_duration=60.0)
        repo.get.return_value = template

        command = ValidateTemplateCommand(
            template_id="tmpl-001",
            user_id="user-001",
            voiceover_duration=70.0,  # 70/60 = 1.167, within ±30%
        )
        result = use_case.execute(command)
        assert result.warnings == []

    def test_voiceover_duration_exceeds_tolerance_warning(self, use_case, repo):
        """配音时长超过 ±30% → 警告."""
        template = _make_template(estimated_duration=60.0)
        repo.get.return_value = template

        command = ValidateTemplateCommand(
            template_id="tmpl-001",
            user_id="user-001",
            voiceover_duration=100.0,  # 100/60 = 1.667, exceeds +30%
        )
        result = use_case.execute(command)

        assert len(result.warnings) == 1
        assert result.warnings[0].code == "voiceover_duration_mismatch"

    def test_voiceover_duration_too_short_warning(self, use_case, repo):
        """配音时长过短（< 70%）→ 警告."""
        template = _make_template(estimated_duration=60.0)
        repo.get.return_value = template

        command = ValidateTemplateCommand(
            template_id="tmpl-001",
            user_id="user-001",
            voiceover_duration=30.0,  # 30/60 = 0.5, below -30%
        )
        result = use_case.execute(command)

        assert len(result.warnings) == 1
        assert result.warnings[0].code == "voiceover_duration_mismatch"

    def test_template_not_found_raises(self, use_case, repo):
        """模板不存在 → NotFoundError."""
        repo.get.return_value = None

        command = ValidateTemplateCommand(
            template_id="nonexistent",
            user_id="user-001",
        )
        with pytest.raises(NotFoundError):
            use_case.execute(command)


# ── Category Use Cases ──


class TestCategoryUseCases:
    @pytest.fixture
    def repo(self):
        return _make_repo()

    def test_create_category(self, repo):
        repo.create_category.side_effect = lambda c: c

        use_case = CreateCategoryUseCase(repo)
        command = CreateCategoryCommand(user_id="user-001", name="Vlog")
        result = use_case.execute(command)

        assert result.name == "Vlog"
        repo.create_category.assert_called_once()

    def test_list_categories(self, repo):
        categories = [
            TemplateCategory(id="cat-1", user_id="user-001", name="Vlog"),
            TemplateCategory(id="cat-2", user_id="user-001", name="教程"),
        ]
        repo.list_categories.return_value = categories

        use_case = ListCategoriesUseCase(repo)
        result = use_case.execute("user-001")

        assert len(result) == 2
        assert result[0].name == "Vlog"

    def test_delete_category_not_found(self, repo):
        repo.delete_category.return_value = False

        use_case = DeleteTemplateUseCase(repo)
        result = use_case.execute("nonexistent", "user-001")
        assert result is False


# ── ListTemplatesUseCase ──


class TestListTemplatesUseCase:
    def test_list_returns_templates(self):
        repo = _make_repo()
        templates = [_make_template(id=f"t-{i}") for i in range(3)]
        repo.list_by_user.return_value = templates

        use_case = ListTemplatesUseCase(repo)
        result = use_case.execute("user-001", skip=0, limit=50)

        assert len(result) == 3
        repo.list_by_user.assert_called_once_with("user-001", skip=0, limit=50)


# ── GetTemplateUseCase ──


class TestGetTemplateUseCase:
    def test_get_existing(self):
        repo = _make_repo()
        template = _make_template()
        repo.get.return_value = template

        use_case = GetTemplateUseCase(repo)
        result = use_case.execute("tmpl-001", "user-001")

        assert result.id == "tmpl-001"

    def test_get_nonexistent_returns_none(self):
        repo = _make_repo()
        repo.get.return_value = None

        use_case = GetTemplateUseCase(repo)
        result = use_case.execute("nonexistent", "user-001")

        assert result is None