"""#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 uuid from datetime import datetime, timezone from unittest.mock import MagicMock, patch import pytest from fastapi.testclient import TestClient from main import app # ── 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 ─────────────────────────────────────────── class TestScriptsRoutesNewFields: """验证 scripts API 路由正确处理新字段.""" def test_create_script_passes_title_fields(self): mock_script = _make_script( title_text="测试标题", title_category="片头", title_config={"text": "测试标题", "font_size": 48}, ) mock_svc = MagicMock() mock_svc.create_script.return_value = mock_script from app.api.routes.scripts import _get_service from app.auth import get_current_user async def _fake_user(): return MagicMock(user=MagicMock(id="user-001")) def _fake_svc(): return mock_svc app.dependency_overrides[get_current_user] = _fake_user app.dependency_overrides[_get_service] = _fake_svc try: client = TestClient(app) resp = client.post( "/api/v1/scripts", json={ "title": "新文案", "content": "内容", "title_text": "测试标题", "title_category": "片头", "title_config": {"text": "测试标题", "font_size": 48}, }, headers={"X-User-Id": "user-001"}, ) assert resp.status_code == 201 call_kwargs = 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} finally: app.dependency_overrides.clear() def test_get_script_response_includes_title_fields(self): mock_script = _make_script( title_text="响应标题", title_category="片尾", title_config={"text": "响应标题", "position": "bottom"}, ) mock_svc = MagicMock() mock_svc.get_script.return_value = mock_script from app.api.routes.scripts import _get_service from app.auth import get_current_user async def _fake_user(): return MagicMock(user=MagicMock(id="user-001")) def _fake_svc(): return mock_svc app.dependency_overrides[get_current_user] = _fake_user app.dependency_overrides[_get_service] = _fake_svc try: client = TestClient(app) resp = client.get("/api/v1/scripts/script-123", headers={"X-User-Id": "user-001"}) assert resp.status_code == 200 data = resp.json() assert data["title_text"] == "响应标题" assert data["title_category"] == "片尾" assert data["title_config"]["position"] == "bottom" finally: app.dependency_overrides.clear() # ── TestTitleLibraryDeprecated ─────────────────────────────────────────── class TestTitleLibraryDeprecated: """验证 title_libraries API 返回 deprecated Warning header.""" def test_list_titles_has_warning_header(self): mock_repo = MagicMock() mock_repo.count_by_user.return_value = 0 from app.api.routes.titles import _get_title_repository from app.auth import get_current_user async def _fake_user(): return MagicMock(user=MagicMock(id="user-001")) def _fake_repo(): return mock_repo app.dependency_overrides[get_current_user] = _fake_user app.dependency_overrides[_get_title_repository] = _fake_repo try: client = TestClient(app) resp = client.get("/api/v1/titles", headers={"X-User-Id": "user-001"}) assert resp.status_code == 200 assert "Warning" in resp.headers or "warning" in {k.lower() for k in resp.headers} finally: app.dependency_overrides.clear() 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), ) mock_repo = MagicMock() from app.api.routes.titles import _get_title_repository from app.auth import get_current_user async def _fake_user(): return MagicMock(user=MagicMock(id="user-001")) def _fake_repo(): return mock_repo app.dependency_overrides[get_current_user] = _fake_user app.dependency_overrides[_get_title_repository] = _fake_repo try: with patch("app.api.routes.titles.GetTitleLibraryUseCase") as mock_uc: mock_uc.return_value.execute.return_value = mock_item client = TestClient(app) resp = client.get("/api/v1/titles/t1", headers={"X-User-Id": "user-001"}) assert resp.status_code == 200 warning_header = resp.headers.get("Warning", "") assert "1894" in warning_header or "deprecated" in warning_header.lower() finally: app.dependency_overrides.clear()