xiaoxia-saas/tests/unit/test_script_title_integration.py

"""#1894 废弃标题库整合到文案库 — 集成测试.

覆盖：
- ScriptModel 新字段 (title_text / title_category / title_config)
- ScriptService CRUD 新字段支持
- ScriptService.get_title_config_for_script 方法
- Scripts API 路由的新字段传递
- title_libraries API deprecated Warning header
"""

from __future__ import annotations

import os
import sys
import uuid
from datetime import UTC, datetime, timezone
from pathlib import Path
from unittest.mock import MagicMock, patch

import pytest
from fastapi.testclient import TestClient

# 确保 apps/api 在 sys.path 中（conftest 已加 root，但 apps/api 也需要）
_APPS_API = str(Path(__file__).resolve().parents[2] / "apps" / "api")
if _APPS_API not in sys.path:
    sys.path.insert(0, _APPS_API)
os.environ.setdefault("JWT_SECRET_KEY", "dev-secret-key-for-testing")

from main import app  # noqa: E402

# ── helpers ──────────────────────────────────────────────────────────────


def _make_script(**overrides):
    """构造一个模拟 ScriptModel 对象."""
    defaults = dict(
        id=str(uuid.uuid4()),
        user_id="user-001",
        title="测试文案",
        content="这是内容",
        segments=[],
        tags=["测试"],
        title_text="开场大标题",
        title_category="片头",
        title_config={
            "text": "开场大标题",
            "font": "思源黑体",
            "font_size": 48,
            "font_color": "#FFFFFF",
            "position": "top",
        },
        created_at=datetime(2026, 9, 1, tzinfo=timezone.utc),
        updated_at=datetime(2026, 9, 1, tzinfo=timezone.utc),
    )
    defaults.update(overrides)
    return MagicMock(**defaults)


# ── TestScriptModelNewFields ─────────────────────────────────────────────


class TestScriptModelNewFields:
    """验证 ScriptModel 新增字段的定义."""

    def test_model_has_title_text_column(self):
        from packages.adapters.sqlalchemy_impl.models import ScriptModel

        assert hasattr(ScriptModel, "title_text")
        col = ScriptModel.__table__.columns["title_text"]
        assert col is not None
        assert str(col.type) == "VARCHAR(500)"

    def test_model_has_title_category_column(self):
        from packages.adapters.sqlalchemy_impl.models import ScriptModel

        assert hasattr(ScriptModel, "title_category")
        col = ScriptModel.__table__.columns["title_category"]
        assert col is not None
        assert str(col.type) == "VARCHAR(50)"

    def test_model_has_title_config_column(self):
        from packages.adapters.sqlalchemy_impl.models import ScriptModel

        assert hasattr(ScriptModel, "title_config")
        col = ScriptModel.__table__.columns["title_config"]
        assert col is not None

    def test_model_defaults(self):
        """新字段默认值为空字符串/空 dict."""
        from packages.adapters.sqlalchemy_impl.models import ScriptModel

        s = ScriptModel(id="x", user_id="u", title="t")
        # 检查 default 值
        assert ScriptModel.__table__.columns["title_text"].default.arg == ""
        assert ScriptModel.__table__.columns["title_category"].default.arg == ""


# ── TestScriptServiceTitleConfig ─────────────────────────────────────────


class TestScriptServiceTitleConfig:
    """验证 ScriptService 新方法 get_title_config_for_script."""

    def test_get_title_config_returns_script_config(self):
        from app.services.script_service import ScriptService

        db = MagicMock()
        mock_script = _make_script(
            title_text="从文案读取",
            title_config={"text": "从文案读取", "font": "Arial", "font_size": 36},
        )
        db.query.return_value.filter.return_value.first.return_value = mock_script

        svc = ScriptService(db)
        result = svc.get_title_config_for_script("script-1", "user-001")

        assert result["text"] == "从文案读取"
        assert result["font"] == "Arial"
        assert result["font_size"] == 36

    def test_get_title_config_fills_text_from_title_text(self):
        """title_config 为空时，用 title_text 填充 text 字段."""
        from app.services.script_service import ScriptService

        db = MagicMock()
        mock_script = _make_script(
            title_text="纯文本标题",
            title_config={},
        )
        db.query.return_value.filter.return_value.first.return_value = mock_script

        svc = ScriptService(db)
        result = svc.get_title_config_for_script("script-2", "user-001")

        assert result["text"] == "纯文本标题"

    def test_get_title_config_raises_on_not_found(self):
        from app.services.script_service import ScriptNotFoundError, ScriptService

        db = MagicMock()
        db.query.return_value.filter.return_value.first.return_value = None

        svc = ScriptService(db)
        with pytest.raises(ScriptNotFoundError):
            svc.get_title_config_for_script("nonexistent", "user-001")

    def test_get_title_config_validates_user_ownership(self):
        """script 不属于当前用户时应抛异常."""
        from app.services.script_service import ScriptNotFoundError, ScriptService

        db = MagicMock()
        db.query.return_value.filter.return_value.first.return_value = None  # 不同用户查不到

        svc = ScriptService(db)
        with pytest.raises(ScriptNotFoundError):
            svc.get_title_config_for_script("script-other-user", "user-001")


# ── TestScriptServiceCreateWithNewFields ─────────────────────────────────


class TestScriptServiceCreateWithNewFields:
    """验证 create_script 和 update_script 支持新字段."""

    def test_create_script_with_title_fields(self):
        from app.services.script_service import ScriptService

        db = MagicMock()
        svc = ScriptService(db)

        script = svc.create_script(
            user_id="user-001",
            title="新文案",
            content="内容",
            title_text="标题文字",
            title_category="片尾",
            title_config={"text": "标题文字", "font_size": 24},
        )

        db.add.assert_called_once()
        db.commit.assert_called_once()
        assert script.title_text == "标题文字"
        assert script.title_category == "片尾"
        assert script.title_config == {"text": "标题文字", "font_size": 24}

    def test_update_script_title_fields(self):
        from app.services.script_service import ScriptService

        db = MagicMock()
        existing = _make_script(title_text="旧标题", title_category="旧分类", title_config={"old": True})
        db.query.return_value.filter.return_value.first.return_value = existing

        svc = ScriptService(db)
        updated = svc.update_script(
            script_id=existing.id,
            user_id="user-001",
            title_text="新标题",
            title_category="新分类",
            title_config={"new": True},
        )

        assert updated.title_text == "新标题"
        assert updated.title_category == "新分类"
        assert updated.title_config == {"new": True}


# ── TestScriptsRoutesNewFields ───────────────────────────────────────────


def _make_mock_auth_user(user_id="user-001"):
    """创建 mock 认证用户."""
    return MagicMock(user=MagicMock(id=user_id))


class TestScriptsRoutesNewFields:
    """验证 scripts API 路由正确处理新字段 — 使用 dependency_overrides 绕过真实 DB/Auth."""

    def setup_method(self):
        from app.api.routes.scripts import _get_service, get_current_user

        self._mock_svc = MagicMock()
        self._mock_user = _make_mock_auth_user()

        def _override_svc():
            return self._mock_svc

        def _override_user():
            return self._mock_user

        app.dependency_overrides[_get_service] = _override_svc
        app.dependency_overrides[get_current_user] = _override_user
        self.client = TestClient(app)

    def teardown_method(self):
        app.dependency_overrides.clear()

    def test_create_script_passes_title_fields(self):
        mock_script = _make_script(
            title_text="测试标题",
            title_category="片头",
            title_config={"text": "测试标题", "font_size": 48},
        )
        self._mock_svc.create_script.return_value = mock_script

        resp = self.client.post(
            "/api/v1/scripts",
            json={
                "title": "新文案",
                "content": "内容",
                "title_text": "测试标题",
                "title_category": "片头",
                "title_config": {"text": "测试标题", "font_size": 48},
            },
        )

        assert resp.status_code == 201, resp.text
        call_kwargs = self._mock_svc.create_script.call_args[1]
        assert call_kwargs["title_text"] == "测试标题"
        assert call_kwargs["title_category"] == "片头"
        assert call_kwargs["title_config"] == {"text": "测试标题", "font_size": 48}

    def test_get_script_response_includes_title_fields(self):
        mock_script = _make_script(
            title_text="响应标题",
            title_category="片尾",
            title_config={"text": "响应标题", "position": "bottom"},
        )
        self._mock_svc.get_script.return_value = mock_script

        resp = self.client.get("/api/v1/scripts/script-123")

        assert resp.status_code == 200, resp.text
        data = resp.json()
        assert data["title_text"] == "响应标题"
        assert data["title_category"] == "片尾"
        assert data["title_config"]["position"] == "bottom"


# ── TestTitleLibraryDeprecated ───────────────────────────────────────────


class TestTitleLibraryDeprecated:
    """验证 title_libraries API 返回 deprecated Warning header — 使用 dependency_overrides 绕过真实 DB/Auth."""

    def setup_method(self):
        from app.api.routes.titles import _get_title_repository, get_current_user
        from app.dependencies import get_user_repository

        self._mock_repo = MagicMock()
        self._mock_user_repo = MagicMock()
        self._mock_user = _make_mock_auth_user()

        app.dependency_overrides[_get_title_repository] = lambda: self._mock_repo
        app.dependency_overrides[get_user_repository] = lambda: self._mock_user_repo
        app.dependency_overrides[get_current_user] = lambda: self._mock_user
        self.client = TestClient(app)

    def teardown_method(self):
        app.dependency_overrides.clear()

    def test_list_titles_has_warning_header(self):
        # list_titles 调 use_case + repo, 注入真实用例但 mock 掉 repo 的 list/count
        self._mock_repo.list_by_user.return_value = []
        self._mock_repo.count_by_user.return_value = 0

        resp = self.client.get("/api/v1/titles")

        assert resp.status_code == 200, resp.text
        headers_lower = {k.lower(): v for k, v in resp.headers.items()}
        assert "warning" in headers_lower or "deprecation" in headers_lower
        assert "1894" in resp.headers.get("Warning", "") or "1894" in resp.headers.get("warning", "")

    def test_get_title_has_warning_header(self):
        from packages.domain.title_library import TitleLibraryItem

        mock_item = TitleLibraryItem(
            id="t1",
            user_id="user-001",
            name="测试",
            text="标题文字",
            category="通用",
            description="",
            tags=[],
            usage_count=0,
            is_active=True,
            created_at=datetime(2026, 1, 1, tzinfo=timezone.utc),
            updated_at=datetime(2026, 1, 1, tzinfo=timezone.utc),
        )
        self._mock_repo.get.return_value = mock_item

        resp = self.client.get("/api/v1/titles/t1")
        assert resp.status_code == 200, resp.text
        warning_header = resp.headers.get("Warning", "") or resp.headers.get("warning", "")
        assert "1894" in warning_header or "deprecated" in warning_header.lower()