From 20297618e2b0756645f7f7bdaf3668fc0cab2346 Mon Sep 17 00:00:00 2001
From: xiaoxia-agent <agent@xiaoxiajianji.com>
Date: Tue, 15 Sep 2026 20:21:06 +0800
Subject: [PATCH] feat(points): P2 step 2.5 - wire generation_preview points
 gate (ai_video scene)

---
 apps/api/app/api/routes/generation_preview.py |  2 +
 tests/unit/test_generation_preview.py         | 10 ++++
 tests/unit/test_generation_preview_points.py  | 49 +++++++++++++++++++
 3 files changed, 61 insertions(+)
 create mode 100644 tests/unit/test_generation_preview_points.py

diff --git a/apps/api/app/api/routes/generation_preview.py b/apps/api/app/api/routes/generation_preview.py
index eb2302ce7..1154de555 100755
--- a/apps/api/app/api/routes/generation_preview.py
+++ b/apps/api/app/api/routes/generation_preview.py
@@ -8,6 +8,7 @@ from __future__ import annotations
 import logging
 
 from app.auth import AuthenticatedUser, get_current_user
+from packages.middleware.points_gate import points_gate
 from app.core.storage import get_storage_service
 from app.core.task_enqueue import (
     GLOBAL_PENDING_LIMIT,
@@ -271,6 +272,7 @@ def _variant_value(values: list[str], index: int, fallback: str = "") -> str:
 
 
 @router.post("/preview", response_model=BatchPreviewGenerationTaskResponse, status_code=201)
+@points_gate("ai_video", quantity_field="preview_count")
 def create_preview_generation_task(
     request: CreatePreviewGenerationTaskRequest,
     authenticated_user: AuthenticatedUser = Depends(get_current_user),
diff --git a/tests/unit/test_generation_preview.py b/tests/unit/test_generation_preview.py
index 8da468eb5..9ac87ff79 100644
--- a/tests/unit/test_generation_preview.py
+++ b/tests/unit/test_generation_preview.py
@@ -573,6 +573,16 @@ from app.schemas.generation_task import (
     PreviewGenerationTaskResponse,
 )
 
+import packages.middleware.points_gate as _pg_module
+
+
+@pytest.fixture(autouse=True)
+def _disable_points_gate(monkeypatch):
+    """默认关闭积分闸门，避免影响既有用例。"""
+    monkeypatch.setattr(_pg_module, "_points_gate_enabled", lambda: False)
+    yield
+
+
 
 def _make_user(user_id="test_user_001"):
     """构造 mock AuthenticatedUser"""
diff --git a/tests/unit/test_generation_preview_points.py b/tests/unit/test_generation_preview_points.py
new file mode 100644
index 000000000..2767e3f79
--- /dev/null
+++ b/tests/unit/test_generation_preview_points.py
@@ -0,0 +1,49 @@
+"""视频预览生成 积分扣点单元测试 (#1895 P2 step 2.5)"""
+from __future__ import annotations
+
+from unittest.mock import MagicMock, patch
+
+import pytest
+
+import packages.middleware.points_gate as _pg_module
+
+
+@pytest.fixture(autouse=True)
+def _enable(monkeypatch):
+    monkeypatch.setattr(_pg_module, "_points_gate_enabled", lambda: True)
+    yield
+
+
+class TestGenerationPreviewPoints:
+    def test_ai_video_cost(self):
+        from packages.domain.points_rules import calculate_points_cost
+        assert calculate_points_cost("ai_video", is_member=False) == 4
+        assert calculate_points_cost("ai_video", is_member=True, member_type="monthly") == 2
+
+    def test_insufficient_raises_402(self):
+        from fastapi import HTTPException
+        from app.api.routes.generation_preview import create_preview_generation_task
+        from app.schemas.generation_task import CreatePreviewGenerationTaskRequest
+
+        db = MagicMock()
+        cu = MagicMock()
+        cu.user.id = "u1"
+        cu.user.is_member = False
+        cu.user.member_type = None
+        req = CreatePreviewGenerationTaskRequest(template_id="t1", asset_ids=["a1"], preview_count=1)
+        with patch("packages.domain.points_service.PointsService") as MS:
+            svc = MagicMock()
+            svc.check_daily_free_clip.return_value = False
+            svc.deduct_points.return_value = {"success": False, "balance": 0}
+            MS.return_value = svc
+            with pytest.raises(HTTPException) as ei:
+                create_preview_generation_task(
+                    request=req, authenticated_user=cu, db=db,
+                    generation_task_repository=MagicMock(),
+                    asset_repo=MagicMock(),
+                )
+            assert ei.value.status_code == 402
+
+    def test_decorator_attached(self):
+        from app.api.routes.generation_preview import create_preview_generation_task
+        assert hasattr(create_preview_generation_task, "__wrapped__"), "missing @points_gate"