xiaoxia-saas/tests/unit/test_1294_preview_voice_injection.py

"""测试 #1294 修复：预览视频配音注入。

验证：
1. _load_task_info 正确加载 voice_ids
3. voice_ids 正确注入到 plan config 中（实际执行代码路径，diff-cover 可达）
"""

from pathlib import Path
from unittest.mock import MagicMock, patch

import pytest


class TestLoadTaskInfoVoiceIds:
    """验证 _load_task_info 包含 voice_ids"""

    def test_voice_ids_loaded_from_task(self):
        """voice_ids 从 gen_task 正确加载"""
        mock_task = MagicMock()
        mock_task.project_id = "proj_1"
        mock_task.asset_library_id = "lib_1"
        mock_task.voice_library_id = "voice_lib_1"
        mock_task.template_id = "tmpl_1"
        mock_task.strategy_id = "one_take"
        mock_task.asset_ids = ["a1", "a2"]
        mock_task.batch_id = "batch_1"
        mock_task.created_by_user_id = "user_1"
        mock_task.video_title = "test"
        mock_task.resolution = "854x480"
        mock_task.bgm_config = {}
        mock_task.is_preview = True
        mock_task.voice_ids = ["voice_1", "voice_2"]

        with patch(
            "packages.adapters.sqlalchemy_impl.generation_task_repository.SQLAlchemyGenerationTaskRepository"
        ) as MockRepo:
            mock_repo = MagicMock()
            mock_repo.get.return_value = mock_task
            MockRepo.return_value = mock_repo

            from worker_app.tasks.generation import _load_task_info

            result = _load_task_info("test_task_id")

            assert result is not None
            assert result["voice_ids"] == ["voice_1", "voice_2"]

    def test_voice_ids_empty_when_none(self):
        """voice_ids 为 None 时返回空列表"""
        mock_task = MagicMock()
        mock_task.project_id = "proj_1"
        mock_task.asset_library_id = "lib_1"
        mock_task.voice_library_id = ""
        mock_task.template_id = "tmpl_1"
        mock_task.strategy_id = "one_take"
        mock_task.asset_ids = ["a1"]
        mock_task.batch_id = ""
        mock_task.created_by_user_id = "user_1"
        mock_task.video_title = ""
        mock_task.resolution = ""
        mock_task.bgm_config = {}
        mock_task.is_preview = False
        mock_task.voice_ids = None

        with patch(
            "packages.adapters.sqlalchemy_impl.generation_task_repository.SQLAlchemyGenerationTaskRepository"
        ) as MockRepo:
            mock_repo = MagicMock()
            mock_repo.get.return_value = mock_task
            MockRepo.return_value = mock_repo

            from worker_app.tasks.generation import _load_task_info

            result = _load_task_info("test_task_id")
            assert result["voice_ids"] == []