xiaoxia-saas/tests/unit/test_generation_preview_points.py

"""视频预览生成 — v1.6.2 起免费，不扣积分"""

from __future__ import annotations

from unittest.mock import MagicMock, patch


class TestGenerationPreviewFree:
    def test_ai_video_returns_zero_cost(self):
        from packages.domain.points_rules import calculate_points_cost

        assert calculate_points_cost("ai_video", is_member=False) == 0
        assert calculate_points_cost("ai_video", is_member=True, member_type="monthly", duration_minutes=10) == 0

    def test_no_points_gate_decorator(self):
        """预览生成路由已移除 @points_gate。"""
        from app.api.routes.generation_preview import create_preview_generation_task

        # 移除装饰器后 __wrapped__ 不再存在
        assert not hasattr(create_preview_generation_task, "__wrapped__")

    def test_endpoint_does_not_deduct_points(self):
        """端点不再实例化 PointsService / 调用 deduct_points（直接走业务逻辑）。"""
        import inspect

        from app.api.routes.generation_preview import create_preview_generation_task

        src = inspect.getsource(create_preview_generation_task)
        assert "PointsService" not in src
        assert "deduct_points" not in src
        assert "calculate_points_cost" not in src