xiaoxia-saas/tests/unit/test_ai_avatar_emotion_tts_lipsync.py

"""#1822 情绪/语速透传 + 对口型 TTS 直生 + 智能封面 单元测试.

CI 增量映射:
  cosyvoice_service.normalize_emotion / payload emotion
  lipsync_service TTS 直生分支（voice_id+script_text）
  ai_avatar_cover_service 智能选帧
"""

import os
from unittest.mock import MagicMock, patch

import pytest

os.environ.setdefault("JWT_SECRET_KEY", "dev-secret-key-for-testing")


# ── 情绪归一化 ──────────────────────────────────────────────────────────


def test_normalize_emotion_english_values():
    from packages.application.cosyvoice_service import normalize_emotion

    assert normalize_emotion("natural") == "natural"
    assert normalize_emotion("excited") == "excited"
    assert normalize_emotion("calm") == "calm"
    assert normalize_emotion("friendly") == "friendly"
    # 大小写 / 空白容错
    assert normalize_emotion("  Excited  ") == "excited"


def test_normalize_emotion_chinese_values():
    from packages.application.cosyvoice_service import normalize_emotion

    assert normalize_emotion("自然") == "natural"
    assert normalize_emotion("兴奋") == "excited"
    assert normalize_emotion("沉稳") == "calm"
    assert normalize_emotion("亲切") == "friendly"


def test_normalize_emotion_invalid_returns_empty():
    from packages.application.cosyvoice_service import normalize_emotion

    assert normalize_emotion("") == ""
    assert normalize_emotion("angry") == ""
    assert normalize_emotion("喜怒哀乐") == ""


# ── CosyVoice payload 携带 emotion + rate ──────────────────────────────


def _make_service_with_captured_client(captured: dict):
    """构造 CosyVoiceService，拦截 post 请求体到 captured['json']."""
    import httpx as _httpx

    from packages.application import cosyvoice_service as mod

    mock_client = MagicMock(spec=_httpx.Client)
    resp = MagicMock()
    resp.status_code = 200
    resp.json.return_value = {
        "request_id": "req-1",
        "output": {"audio": {"url": "https://tts/a.mp3", "duration": 1.0}},
    }
    resp.raise_for_status = MagicMock()

    def fake_request(method, url, headers, json, timeout):
        captured["json"] = json
        return resp

    mock_client.request.side_effect = fake_request

    with patch.object(mod, "get_shared_settings") as settings_patch:
        s = MagicMock()
        s.cosyvoice_api_key = "sk-test"
        s.cosyvoice_base_url = "https://x/api/v1"
        s.cosyvoice_model = "cosyvoice-v3-flash"
        s.cosyvoice_clone_model = "voice-enrollment"
        s.cosyvoice_format = "mp3"
        s.cosyvoice_sample_rate = 22050
        s.cosyvoice_voice = "longxiaochun"
        settings_patch.return_value = s
        svc = mod.CosyVoiceService(http_client=mock_client)
    return svc


def test_submit_synthesize_payload_includes_emotion_and_rate():
    captured: dict = {}
    svc = _make_service_with_captured_client(captured)

    svc.submit_synthesize_task(text="你好", voice_id="v-1", speed=1.5, emotion="兴奋")

    inp = captured["json"]["input"]
    assert inp["emotion"] == "excited"
    assert inp["rate"] == 1.5


def test_submit_synthesize_payload_omits_emotion_when_empty():
    captured: dict = {}
    svc = _make_service_with_captured_client(captured)

    svc.submit_synthesize_task(text="你好", voice_id="v-1")

    assert "emotion" not in captured["json"]["input"]


# ── 对口型 TTS 直生分支 ─────────────────────────────────────────────────


def _lipsync_service_with_mocks():
    from app.services.lipsync_service import LipsyncService

    db = MagicMock()
    client = MagicMock()
    client.is_available = True
    client.submit_lipsync.return_value = {
        "success": True,
        "task_id": "mk-1",
        "request_id": "req-1",
    }
    cosy = MagicMock()
    cosy.submit_synthesize_task.return_value = {
        "audio_url": "https://tts/raw.mp3",
        "request_id": "tts-req",
        "audio_duration": 3.0,
    }
    svc = LipsyncService(db, client=client, cosyvoice_service=cosy, voice_clone_repo=MagicMock())
    # _resolve_voice_id 默认原样返回（repo.get 返回 None）
    svc._voice_clone_repo.get.return_value = None
    return svc, client, cosy


def test_create_job_tts_direct_mode_synthesizes_audio():
    svc, client, cosy = _lipsync_service_with_mocks()

    with patch("app.services.lipsync_service.tts_synthesize_and_submit") as mock_task:
        mock_task.apply_async.return_value = MagicMock(id="celery-task-123")

        job = svc.create_job(
            user_id="user-1",
            video_url="https://oss/person.mp4",
            voice_id="cosy-v1",
            script_text="你好世界",
            speed=1.2,
            emotion="兴奋",
        )

    # v4: TTS 模式下 create_job 返回 tts_processing 状态，dispatch Celery 任务
    assert job.status == "tts_processing"
    assert job.emotion == "excited"
    assert job.speed == 1.2
    # 不直接调用 CosyVoice（由 Celery 任务处理）
    cosy.submit_synthesize_task.assert_not_called()
    # 不直接提交 MediaKit（由 Celery 任务处理）
    client.submit_lipsync.assert_not_called()
    # dispatch 了 Celery 任务
    mock_task.apply_async.assert_called_once()


def test_create_job_direct_audio_mode_skips_tts():
    svc, client, cosy = _lipsync_service_with_mocks()

    job = svc.create_job(
        user_id="user-1",
        video_url="https://oss/person.mp4",
        audio_url="https://oss/ready.mp3",
    )

    cosy.submit_synthesize_task.assert_not_called()
    _, submit_kwargs = client.submit_lipsync.call_args
    assert submit_kwargs["audio_url"] == "https://oss/ready.mp3"


def test_create_job_tts_failure_raises():
    """v4: TTS 模式下 create_job 不再同步失败，而是 dispatch Celery 任务。
    TTS 合成失败由 Celery 任务内部处理并更新 job 状态。"""
    svc, client, cosy = _lipsync_service_with_mocks()

    with patch("app.services.lipsync_service.tts_synthesize_and_submit") as mock_task:
        mock_task.apply_async.return_value = MagicMock(id="celery-task-456")

        job = svc.create_job(
            user_id="user-1",
            video_url="https://oss/person.mp4",
            voice_id="v-1",
            script_text="文本",
        )

    # create_job 成功返回 tts_processing，不直接调用 TTS
    assert job.status == "tts_processing"
    cosy.submit_synthesize_task.assert_not_called()
    client.submit_lipsync.assert_not_called()
    mock_task.apply_async.assert_called_once()


# ── refresh 同步中间状态 ────────────────────────────────────────────────


def test_refresh_syncs_running_status():
    from app.services.lipsync_service import LipsyncService

    db = MagicMock()
    client = MagicMock()
    client.get_task_status.return_value = {"success": True, "status": "running"}
    svc = LipsyncService(db, client=client)

    job = MagicMock()
    job.status = "submitted"
    job.mediakit_task_id = "mk-1"
    job.id = "j-1"
    svc.get_job = MagicMock(return_value=job)

    result = svc.refresh_job_status("j-1", "user-1")
    assert result.status == "running"


# ── 智能封面 ────────────────────────────────────────────────────────────


def test_smart_cover_selects_best_frame_and_persists():
    from app.services import ai_avatar_cover_service as cov

    snapshots = [
        {"image_url": "https://mk/f0.jpg"},
        {"image_url": "https://mk/f1.jpg"},
    ]
    with (
        patch("packages.shared.mediakit_client.get_mediakit_client") as mk_patch,
        patch("packages.shared.cover_frame_scorer.score_frames") as score_patch,
        patch("httpx.Client") as http_client_cls,
        patch("packages.shared.storage.get_shared_storage_service") as storage_patch,
    ):
        mk = MagicMock()
        mk.is_available = True
        mk.extract_frames.return_value = snapshots
        mk_patch.return_value = mk
        # score_frames 把 f1 选为最佳
        score_patch.side_effect = lambda cands: [
            {"url": "https://mk/f1.jpg", "score": 90.0, "image_path": cands[1]["image_path"]},
            {"url": "https://mk/f0.jpg", "score": 60.0, "image_path": cands[0]["image_path"]},
        ]
        # httpx.Client 连接池 mock
        client_instance = MagicMock()
        resp = MagicMock()
        resp.content = b"IMGDATA"
        resp.raise_for_status = MagicMock()
        client_instance.get.return_value = resp
        client_instance.__enter__ = MagicMock(return_value=client_instance)
        client_instance.__exit__ = MagicMock(return_value=False)
        http_client_cls.return_value = client_instance

        storage = MagicMock()
        storage.public_url = "https://oss.example.com"
        # video_url 不是自家 OSS，不重签
        storage.get_download_url.side_effect = lambda url, **kw: f"{url}?signed=1"
        storage.upload_file.return_value = "https://oss.example.com/cover.jpg"
        storage_patch.return_value = storage

        url = cov.generate_smart_cover("https://other-host/avatar.mp4", job_id="job-1")

    assert "signed=1" in url or url == "https://oss.example.com/cover.jpg"
    mk.extract_frames.assert_called_once()
    score_patch.assert_called_once()
    # 验证使用了增大的轮询参数
    call_kwargs = mk.extract_frames.call_args
    assert call_kwargs.kwargs.get("poll_interval") == 2.0 or call_kwargs[1].get("poll_interval") == 2.0
    assert call_kwargs.kwargs.get("max_poll_attempts") == 30 or call_kwargs[1].get("max_poll_attempts") == 30


def test_smart_cover_returns_empty_when_mediakit_unavailable():
    from app.services import ai_avatar_cover_service as cov

    with (
        patch("packages.shared.mediakit_client.get_mediakit_client") as mk_patch,
        patch("packages.shared.storage.get_shared_storage_service") as storage_patch,
    ):
        mk = MagicMock()
        mk.is_available = False
        mk_patch.return_value = mk
        storage = MagicMock()
        storage.public_url = "https://oss.example.com"
        storage_patch.return_value = storage
        url = cov.generate_smart_cover("https://oss.example.com/avatar.mp4")
    assert url == ""


def test_sign_video_url_resigns_own_oss_url():
    """自家 OSS 私有桶 URL 应被重签为长有效期预签名 URL"""
    from app.services.ai_avatar_cover_service import _sign_video_url_for_mediakit

    with patch("packages.shared.storage.get_shared_storage_service") as storage_patch:
        storage = MagicMock()
        storage.public_url = "https://oss.example.com"
        storage.get_download_url.return_value = "https://oss.example.com/file.mp4?Expires=xxx&Signature=yyy"
        storage_patch.return_value = storage

        result = _sign_video_url_for_mediakit("https://oss.example.com/file.mp4")

    assert "Signature=yyy" in result
    storage.get_download_url.assert_called_once()


def test_sign_video_url_skips_external_url():
    """外部 URL（非自家 OSS）应原样返回，不做重签"""
    from app.services.ai_avatar_cover_service import _sign_video_url_for_mediakit

    with patch("packages.shared.storage.get_shared_storage_service") as storage_patch:
        storage = MagicMock()
        storage.public_url = "https://oss.example.com"
        storage_patch.return_value = storage

        result = _sign_video_url_for_mediakit("https://external-cdn.com/video.mp4")

    assert result == "https://external-cdn.com/video.mp4"
    storage.get_download_url.assert_not_called()


def test_extract_frames_uses_extended_poll_params():
    """验证 select_best_cover_frame 使用增大后的轮询参数"""
    from app.services import ai_avatar_cover_service as cov

    with (
        patch("packages.shared.mediakit_client.get_mediakit_client") as mk_patch,
        patch("packages.shared.storage.get_shared_storage_service") as storage_patch,
    ):
        mk = MagicMock()
        mk.is_available = True
        mk.extract_frames.return_value = [{"image_url": "https://mk/f0.jpg"}]
        mk_patch.return_value = mk

        storage = MagicMock()
        storage.public_url = "https://oss.example.com"
        storage_patch.return_value = storage

        cov.select_best_cover_frame("https://other/avatar.mp4", max_frames=3)

    call_kwargs = mk.extract_frames.call_args
    assert call_kwargs.kwargs.get("poll_interval") == 2.0 or call_kwargs[1].get("poll_interval") == 2.0
    assert call_kwargs.kwargs.get("max_poll_attempts") == 30 or call_kwargs[1].get("max_poll_attempts") == 30
    assert call_kwargs.kwargs.get("max_retries") == 1 or call_kwargs[1].get("max_retries") == 1


# ── 渲染 script_id 可选（手动文案直生场景）──────────────────────────────


def test_render_request_script_id_optional():
    import os
    import sys

    sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "..", "apps", "api"))
    from app.schemas.ai_avatar_render import CreateAiAvatarRenderRequest

    # 手动文案直生：不传 script_id 也合法
    req = CreateAiAvatarRenderRequest(lipsync_job_id="j-1")
    assert req.script_id == ""

    # 空白被 strip
    req2 = CreateAiAvatarRenderRequest(lipsync_job_id="j-1", script_id="   ")
    assert req2.script_id == ""

    # title_config 是单个 dict
    req3 = CreateAiAvatarRenderRequest(
        lipsync_job_id="j-1",
        title_config={"text": "标题", "position": "top", "font_size": 40},
    )
    assert req3.title_config["position"] == "top"