"""AI数字人渲染 积分扣点单元测试 (#1895 P2 step 2.6)""" 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 TestAiAvatarRenderPoints: def test_ai_digital_human_per_unit(self): from packages.domain.points_rules import calculate_points_cost cost = calculate_points_cost("ai_digital_human", is_member=False, duration_minutes=1) assert cost >= 15 def test_decorator_attached(self): from app.api.routes.ai_avatar_render import create_render_job assert hasattr(create_render_job, "__wrapped__"), "missing @points_gate" def test_insufficient_raises_402(self): from app.api.routes.ai_avatar_render import create_render_job from app.schemas.ai_avatar_render import CreateAiAvatarRenderRequest from fastapi import HTTPException db = MagicMock() cu = MagicMock() cu.user.id = "u1" cu.user.is_member = False cu.user.member_type = None svc = MagicMock() body = CreateAiAvatarRenderRequest(lipsync_job_id="lip1") with patch("packages.domain.points_service.PointsService") as MS: msvc = MagicMock() msvc.deduct_points.return_value = {"success": False, "balance": 0} MS.return_value = msvc with pytest.raises(HTTPException) as ei: create_render_job(body=body, current_user=cu, svc=svc, db=db) assert ei.value.status_code == 402