xiaoxia-saas/tests/unit/test_extract_from_douyin_errors.py

"""验证 extract-from-douyin 在各种失败场景返回正确的 HTTP 状态码（绝不能 500）"""
from __future__ import annotations

import sys
import types
from unittest import mock

import pytest
from fastapi import HTTPException, status

from app.auth import AuthenticatedUser


class _FakeUser:
    id = "u-test"
    is_member = False
    member_type = None


@pytest.fixture
def fake_user():
    return AuthenticatedUser(user=_FakeUser())


class _FakeYDLBase:
    """通用假 yt-dlp 基类"""
    extract_info_result = None
    extract_info_raises = None
    prepare_filename_result = "/tmp/fake.mp4"

    def __init__(self, *a, **kw):
        pass

    def extract_info(self, url, download=True):
        if self.__class__.extract_info_raises:
            raise self.__class__.extract_info_raises
        return self.__class__.extract_info_result

    def prepare_filename(self, info):
        return self.__class__.prepare_filename_result


def _install_fake_ytdlp(fake_ydl_class, *, download_error_cls=None):
    """把假 yt-dlp 注入 sys.modules，函数内 import yt_dlp 会拿到我们的假版本"""
    fake_mod = types.ModuleType("yt_dlp")
    fake_mod.YoutubeDL = fake_ydl_class
    if download_error_cls is None:
        download_error_cls = type("DownloadError", (Exception,), {})
    fake_mod.DownloadError = download_error_cls
    utils_mod = types.ModuleType("yt_dlp.utils")
    utils_mod.DownloadError = download_error_cls
    fake_mod.utils = utils_mod
    sys.modules["yt_dlp"] = fake_mod
    sys.modules["yt_dlp.utils"] = utils_mod
    return fake_mod


def _import_target():
    from app.api.routes import scripts_ai
    return scripts_ai


def test_download_http404_returns_400_not_500(fake_user):
    """无效短链 / 视频 404 → 应返回 400 业务错误，不能 500"""
    scripts_ai = _import_target()
    body = scripts_ai.ExtractFromDouyinRequest(url="https://v.douyin.com/test123/")

    class DownloadError(Exception):
        pass
    class FailingYDL(_FakeYDLBase):
        extract_info_raises = DownloadError("ERROR: Unable to download webpage: HTTP Error 404: Not Found")
    _install_fake_ytdlp(FailingYDL, download_error_cls=DownloadError)

    with mock.patch.object(scripts_ai, "get_doubao_client", return_value=mock.MagicMock(is_available=True)):
        with pytest.raises(HTTPException) as exc:
            scripts_ai.extract_from_douyin(request=body, current_user=fake_user, db=mock.MagicMock())
        assert exc.value.status_code == status.HTTP_400_BAD_REQUEST, f"应为400，实际 {exc.value.status_code}: {exc.value.detail}"
        assert "无法解析" in exc.value.detail or "抖音" in exc.value.detail


def test_download_network_error_returns_502_not_500(fake_user):
    """网络错误 / 上游异常 → 502，不能 500"""
    scripts_ai = _import_target()
    body = scripts_ai.ExtractFromDouyinRequest(url="https://v.douyin.com/abc/")

    class DownloadError(Exception):
        pass
    class NetErrYDL(_FakeYDLBase):
        extract_info_raises = DownloadError("ERROR: Connection reset by peer")
    _install_fake_ytdlp(NetErrYDL, download_error_cls=DownloadError)

    with mock.patch.object(scripts_ai, "get_doubao_client", return_value=mock.MagicMock(is_available=True)):
        with pytest.raises(HTTPException) as exc:
            scripts_ai.extract_from_douyin(request=body, current_user=fake_user, db=mock.MagicMock())
        assert exc.value.status_code == status.HTTP_502_BAD_GATEWAY


def test_info_none_returns_400(fake_user):
    """yt-dlp 返回 None info → 400"""
    scripts_ai = _import_target()
    body = scripts_ai.ExtractFromDouyinRequest(url="https://v.douyin.com/abc/")

    class NoneInfoYDL(_FakeYDLBase):
        extract_info_result = None
    _install_fake_ytdlp(NoneInfoYDL)

    with mock.patch.object(scripts_ai, "get_doubao_client", return_value=mock.MagicMock(is_available=True)):
        with pytest.raises(HTTPException) as exc:
            scripts_ai.extract_from_douyin(request=body, current_user=fake_user, db=mock.MagicMock())
        assert exc.value.status_code == status.HTTP_400_BAD_REQUEST


def test_asr_not_configured_returns_503(fake_user):
    scripts_ai = _import_target()
    from app.services.script_asr_service import ASRNotConfiguredError
    body = scripts_ai.ExtractFromDouyinRequest(url="https://v.douyin.com/abc/")

    import os.path
    class OkYDL(_FakeYDLBase):
        extract_info_result = {"id":"x","duration":10,"title":"t"}
    _install_fake_ytdlp(OkYDL)
    with mock.patch.object(scripts_ai, "get_doubao_client", return_value=mock.MagicMock(is_available=True)), \
         mock.patch.object(scripts_ai.os.path, "isfile", return_value=True), \
         mock.patch.object(scripts_ai.os.path, "getsize", return_value=1024), \
         mock.patch.object(scripts_ai, "transcribe_to_text", side_effect=ASRNotConfiguredError("未配置")):
        with pytest.raises(HTTPException) as exc:
            scripts_ai.extract_from_douyin(request=body, current_user=fake_user, db=mock.MagicMock())
        assert exc.value.status_code == status.HTTP_503_SERVICE_UNAVAILABLE


def test_asr_failure_returns_502(fake_user):
    scripts_ai = _import_target()
    from app.services.script_asr_service import ASRTranscriptionError
    body = scripts_ai.ExtractFromDouyinRequest(url="https://v.douyin.com/abc/")

    class OkYDL(_FakeYDLBase):
        extract_info_result = {"id":"x","duration":10,"title":"t"}
    _install_fake_ytdlp(OkYDL)
    with mock.patch.object(scripts_ai, "get_doubao_client", return_value=mock.MagicMock(is_available=True)), \
         mock.patch.object(scripts_ai.os.path, "isfile", return_value=True), \
         mock.patch.object(scripts_ai.os.path, "getsize", return_value=1024), \
         mock.patch.object(scripts_ai, "transcribe_to_text", side_effect=ASRTranscriptionError("识别失败")):
        with pytest.raises(HTTPException) as exc:
            scripts_ai.extract_from_douyin(request=body, current_user=fake_user, db=mock.MagicMock())
        assert exc.value.status_code == status.HTTP_502_BAD_GATEWAY


def test_asr_unexpected_error_returns_502_not_500(fake_user):
    """ASR 抛未预期异常（非 ASRNotConfigured/ASRTranscriptionError）也应被兜住，不能 500"""
    scripts_ai = _import_target()
    body = scripts_ai.ExtractFromDouyinRequest(url="https://v.douyin.com/abc/")

    class OkYDL(_FakeYDLBase):
        extract_info_result = {"id":"x","duration":10,"title":"t"}
    _install_fake_ytdlp(OkYDL)
    with mock.patch.object(scripts_ai, "get_doubao_client", return_value=mock.MagicMock(is_available=True)), \
         mock.patch.object(scripts_ai.os.path, "isfile", return_value=True), \
         mock.patch.object(scripts_ai.os.path, "getsize", return_value=1024), \
         mock.patch.object(scripts_ai, "transcribe_to_text", side_effect=RuntimeError("ffmpeg crashed")):
        with pytest.raises(HTTPException) as exc:
            scripts_ai.extract_from_douyin(request=body, current_user=fake_user, db=mock.MagicMock())
        assert exc.value.status_code == status.HTTP_502_BAD_GATEWAY, f"应为502，实际 {exc.value.status_code}"


def test_missing_downloaded_file_returns_502_not_500(fake_user):
    """yt-dlp 返回 info 但文件未落地（isfile False）→ 502"""
    scripts_ai = _import_target()
    body = scripts_ai.ExtractFromDouyinRequest(url="https://v.douyin.com/abc/")

    class OkYDL(_FakeYDLBase):
        extract_info_result = {"id":"x","duration":10,"title":"t"}
    _install_fake_ytdlp(OkYDL)
    with mock.patch.object(scripts_ai, "get_doubao_client", return_value=mock.MagicMock(is_available=True)), \
         mock.patch.object(scripts_ai.os.path, "isfile", return_value=False):
        with pytest.raises(HTTPException) as exc:
            scripts_ai.extract_from_douyin(request=body, current_user=fake_user, db=mock.MagicMock())
        assert exc.value.status_code != 500
        assert "下载异常" in exc.value.detail or "文件" in exc.value.detail


def test_any_unexpected_error_does_not_return_500_raw(fake_user):
    """兜底：prepare_filename 抛未预期异常也应被捕获，返回500 code但含业务detail"""
    scripts_ai = _import_target()
    body = scripts_ai.ExtractFromDouyinRequest(url="https://v.douyin.com/abc/")

    class BuggyYDL(_FakeYDLBase):
        def extract_info(self, url, download=True):
            return {"id":"x","duration":"not_a_number","title":"t"}
        def prepare_filename(self, info):
            raise RuntimeError("some internal bug")
    _install_fake_ytdlp(BuggyYDL)
    with mock.patch.object(scripts_ai, "get_doubao_client", return_value=mock.MagicMock(is_available=True)):
        with pytest.raises(HTTPException) as exc:
            scripts_ai.extract_from_douyin(request=body, current_user=fake_user, db=mock.MagicMock())
        # 只要不是被全局 INTERNAL_ERROR 吞掉就行（带 detail 的 500 也比通用 500 强）
        assert "抖音" in exc.value.detail or "失败" in exc.value.detail or exc.value.status_code != 500