feat(points): P2 step 2.4 - wire generation_tasks points gate (ai_video scene)
CI/CD Pipeline / Dedup Check - skip PR tests when covered by push pipeline (pull_request) Successful in 7s
CI/CD Pipeline / Check if frontend-only change (pull_request) Successful in 3s
PR Automation / Auto Approve on CI Green (pull_request) Successful in 2m56s
Preview Deploy / Deploy Preview Environment (pull_request) Successful in 2m43s
AI Code Review / AI Code Review (pull_request) Failing after 21m29s
CI/CD Pipeline / PR Build API Image (pull_request) Successful in 23s
CI/CD Pipeline / Validate - Style (pull_request) Failing after 6m34s
CI/CD Pipeline / PR Build Worker Image (pull_request) Successful in 4m4s
CI/CD Pipeline / Unit Tests (pull_request) Failing after 6m35s
CI/CD Pipeline / Integration Tests (pull_request) Successful in 7m12s
CI/CD Pipeline / Validate - Python (mypy + alembic) (pull_request) Successful in 8m23s
CI/CD Pipeline / Validate - Security (pull_request) Has been cancelled
CI/CD Pipeline / Deploy Staging (Watchtower auto-deploy) (pull_request) Has been cancelled
CI/CD Pipeline / Staging E2E Tests (pull_request) Has been cancelled
CI/CD Pipeline / Staging API Integration Tests (pull_request) Has been cancelled
CI/CD Pipeline / Build Production API Image (pull_request) Has been cancelled
CI/CD Pipeline / Build Production Web Image (pull_request) Has been cancelled
CI/CD Pipeline / Build Production Worker Image (pull_request) Has been cancelled
CI/CD Pipeline / Deploy Production (pull_request) Has been cancelled
CI/CD Pipeline / Production Browser E2E (pull_request) Has been cancelled
CI/CD Pipeline / ACR Image Cleanup (pull_request) Has been cancelled
CI/CD Pipeline / Canary Release to Production (pull_request) Has been cancelled
CI/CD Pipeline / CI Gate (pull_request) Has been cancelled
PR Automation / Auto Merge on CI Green + Approved (pull_request) Has been cancelled
CI/CD Pipeline / Build Staging Web Image (pull_request) Failing after 14h23m52s
CI/CD Pipeline / Retag skipped Staging Worker Image (pull_request) Failing after 14h12m39s
CI/CD Pipeline / Retag skipped Staging Web Image (pull_request) Failing after 14h12m39s
CI/CD Pipeline / Retag skipped Staging API Image (pull_request) Failing after 14h12m40s
CI/CD Pipeline / Build Staging Worker Image (pull_request) Failing after 14h23m9s
CI/CD Pipeline / Build Staging API Image (pull_request) Failing after 14h23m9s
CI/CD Pipeline / PR Build Web Image (pull_request) Failing after 14h23m21s
CI/CD Pipeline / Check push changed paths (pull_request) Failing after 14h48m25s
CI/CD Pipeline / Frontend Unit Tests (pull_request) Failing after 14h23m32s
CI/CD Pipeline / Frontend Lint (pull_request) Failing after 14h23m32s

This commit is contained in:
xiaoxia-agent
2026-09-15 20:20:41 +08:00
parent 20e311e341
commit 421cc3f25d
3 changed files with 67 additions and 0 deletions
@@ -4,6 +4,7 @@ from typing import Any
from app.api.routes._helpers import check_project_access
from app.auth import AuthenticatedUser, get_current_user
from packages.middleware.points_gate import points_gate
from app.core.storage import OSSStorageService, get_storage_service
from app.core.task_enqueue import (
GLOBAL_PENDING_LIMIT,
@@ -211,6 +212,7 @@ def _resolve_project_and_library(
@router.post("/tasks", response_model=BatchGenerationTaskResponse)
@points_gate("ai_video", quantity_field="count")
def create_generation_task(
request: CreateGenerationTaskRequest,
authenticated_user: AuthenticatedUser = Depends(get_current_user),
+10
View File
@@ -17,6 +17,16 @@ from packages.application.generation_tasks import (
)
from packages.domain import GenerationTask
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
@pytest.fixture
def mock_repo():
@@ -0,0 +1,55 @@
"""视频生成 积分扣点单元测试 (#1895 P2 step 2.4)"""
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 TestGenerationTasksPoints:
def test_ai_video_base_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_ai_video_quantity_scales(self):
from packages.domain.points_rules import calculate_points_cost
c1 = calculate_points_cost("ai_video", is_member=False, quantity=1)
c3 = calculate_points_cost("ai_video", is_member=False, quantity=3)
assert c3 > c1
def test_insufficient_raises_402(self):
from fastapi import HTTPException
from app.api.routes.generation_tasks import create_generation_task
from app.schemas.generation_task import CreateGenerationTaskRequest
db = MagicMock()
cu = MagicMock()
cu.user.id = "u1"
cu.user.is_member = False
cu.user.member_type = None
req = CreateGenerationTaskRequest(template_id="t1", asset_ids=["a1"], 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_generation_task(
request=req, authenticated_user=cu, db=db,
generation_task_repository=MagicMock(), project_repository=MagicMock(),
asset_library_repository=MagicMock(), asset_repository=MagicMock(),
)
assert ei.value.status_code == 402
def test_decorator_attached(self):
from app.api.routes.generation_tasks import create_generation_task
assert hasattr(create_generation_task, "__wrapped__"), "missing @points_gate"