xiaoxia-saas/tests/unit/test_lipsync_gpu_integration.py

"""LipsyncService GPU 路径集成测试."""

from __future__ import annotations

from unittest.mock import MagicMock, patch

import pytest


@pytest.fixture()
def fake_db():
    db = MagicMock()
    return db


@pytest.fixture()
def fake_mediakit():
    client = MagicMock()
    client.submit_lipsync.return_value = {"task_id": "mk-task-1"}
    return client


def _make_job(video_url="oss://video.mp4", audio_url="oss://audio.wav"):
    job = MagicMock()
    job.id = "job-1"
    job.user_id = "u1"
    job.project_id = "p1"
    job.video_url = video_url
    job.audio_url = audio_url
    job.enable_video_loop = True
    job.script_text = ""
    job.sentence_timings = None
    return job


def _make_svc(db, mediakit, use_gpu=False):
    from app.services.lipsync_service import LipsyncService

    svc = LipsyncService(db=db, client=mediakit)
    svc.settings.use_gpu_lipsync = use_gpu
    svc._sign_media_url = lambda u: (u or "") + "?signed"
    return svc


class TestGpuFallback:
    def test_switch_off_uses_mediakit(self, fake_db, fake_mediakit):
        """开关关闭时直接走 MediaKit，不调用 _submit_to_gpu."""
        svc = _make_svc(fake_db, fake_mediakit, use_gpu=False)
        job = _make_job()
        with patch.object(svc, "_submit_to_gpu") as m_sub:
            svc._submit_audio_direct(job=job)
            m_sub.assert_not_called()
        fake_mediakit.submit_lipsync.assert_called_once()
        assert job.status == "submitted"

    def test_switch_on_no_worker_falls_back(self, fake_db, fake_mediakit):
        """开关打开但 has_available_worker=False → 回退 MediaKit."""
        svc = _make_svc(fake_db, fake_mediakit, use_gpu=True)
        fake_gpu_svc = MagicMock()
        fake_gpu_svc.has_available_worker.return_value = False
        with patch("app.services.gpu_lipsync_service.GpuLipsyncService", return_value=fake_gpu_svc):
            job = _make_job()
            svc._submit_audio_direct(job=job)
        fake_gpu_svc.create_task.assert_not_called()
        fake_mediakit.submit_lipsync.assert_called_once()
        assert job.status == "submitted"

    def test_gpu_success_marks_completed(self, fake_db, fake_mediakit):
        """GPU 路径成功：job 直接 completed，不调 MediaKit."""
        svc = _make_svc(fake_db, fake_mediakit, use_gpu=True)
        gpu_done = MagicMock(
            id="gpu-task-1",
            status="done",
            result_url="oss://gpu-results/r.mp4",
            result_duration=12.5,
        )
        fake_gpu_svc = MagicMock()
        fake_gpu_svc.has_available_worker.return_value = True
        fake_gpu_svc.create_task.return_value = MagicMock(id="gpu-task-1")
        fake_gpu_svc.wait_for_result.return_value = gpu_done
        with patch("app.services.gpu_lipsync_service.GpuLipsyncService", return_value=fake_gpu_svc):
            job = _make_job()
            svc._submit_audio_direct(job=job)
        fake_gpu_svc.create_task.assert_called_once()
        fake_mediakit.submit_lipsync.assert_not_called()
        assert job.status == "completed"
        assert job.output_duration == 12.5
        assert "?signed" in job.output_video_url
        fake_db.commit.assert_called()

    def test_gpu_timeout_falls_back(self, fake_db, fake_mediakit):
        """wait_for_result 返回 None（超时）→ 回退 MediaKit."""
        svc = _make_svc(fake_db, fake_mediakit, use_gpu=True)
        fake_gpu_svc = MagicMock()
        fake_gpu_svc.has_available_worker.return_value = True
        fake_gpu_svc.create_task.return_value = MagicMock(id="gpu-t")
        fake_gpu_svc.wait_for_result.return_value = None
        with patch("app.services.gpu_lipsync_service.GpuLipsyncService", return_value=fake_gpu_svc):
            job = _make_job()
            svc._submit_audio_direct(job=job)
        fake_mediakit.submit_lipsync.assert_called_once()
        assert job.status == "submitted"

    def test_gpu_failed_status_falls_back(self, fake_db, fake_mediakit):
        """GPU 终态 failed → 回退 MediaKit."""
        svc = _make_svc(fake_db, fake_mediakit, use_gpu=True)
        fake_gpu_svc = MagicMock()
        fake_gpu_svc.has_available_worker.return_value = True
        fake_gpu_svc.create_task.return_value = MagicMock(id="gpu-t")
        fake_gpu_svc.wait_for_result.return_value = MagicMock(status="failed", error_msg="musetalk crash")
        with patch("app.services.gpu_lipsync_service.GpuLipsyncService", return_value=fake_gpu_svc):
            job = _make_job()
            svc._submit_audio_direct(job=job)
        fake_mediakit.submit_lipsync.assert_called_once()
        assert job.status == "submitted"

    def test_gpu_exception_falls_back(self, fake_db, fake_mediakit):
        """GPU 路径抛异常 → 回退 MediaKit."""
        svc = _make_svc(fake_db, fake_mediakit, use_gpu=True)
        fake_gpu_svc = MagicMock()
        fake_gpu_svc.has_available_worker.return_value = True
        fake_gpu_svc.create_task.side_effect = RuntimeError("DB down")
        with patch("app.services.gpu_lipsync_service.GpuLipsyncService", return_value=fake_gpu_svc):
            job = _make_job()
            svc._submit_audio_direct(job=job)
        fake_mediakit.submit_lipsync.assert_called_once()
        assert job.status == "submitted"


class TestGpuServiceHelpers:
    """GpuLipsyncService.has_available_worker 测试."""

    def test_no_workers(self, fake_db):
        from app.services.gpu_lipsync_service import GpuLipsyncService

        svc = GpuLipsyncService(db=fake_db)
        fake_db.query.return_value.filter.return_value.first.return_value = None
        assert svc.has_available_worker() is False

    def test_fresh_worker_available(self, fake_db):
        from app.services.gpu_lipsync_service import GpuLipsyncService

        svc = GpuLipsyncService(db=fake_db)
        svc.settings.gpu_worker_stale_seconds = 300
        # 模拟SQL filter条件成立 → first() 返回非None
        fake_db.query.return_value.filter.return_value.first.return_value = MagicMock()
        assert svc.has_available_worker() is True

    def test_stale_worker_unavailable(self, fake_db):
        from app.services.gpu_lipsync_service import GpuLipsyncService

        svc = GpuLipsyncService(db=fake_db)
        # filter条件不成立（stale）→ first() 返回None
        fake_db.query.return_value.filter.return_value.first.return_value = None
        assert svc.has_available_worker() is False