xiaoxia-saas/tests/unit/test_ai_avatar_render_points.py

"""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