xiaoxia-saas/tests/unit/test_auth_register_points.py

"""注册送积分单元测试 (#1895 P2 step 3)"""

from __future__ import annotations

from unittest.mock import MagicMock, patch

import pytest


@pytest.fixture
def mock_settings(monkeypatch):
    """默认关闭 points_enabled，不影响现有用例。"""
    from app.config import settings

    monkeypatch.setattr(settings, "points_enabled", False)
    return settings


class TestRegisterBonusPoints:
    @pytest.mark.asyncio
    async def test_bonus_when_enabled(self, mock_settings):
        """开启积分时注册成功送50分。"""
        from app.api.routes import auth
        from app.api.routes.auth import RegisterRequest

        mock_settings.points_enabled = True

        mock_uc = MagicMock()
        mock_resp = MagicMock()
        mock_resp.user_id = "new-user-1"
        mock_resp.email = "n***@example.com"
        mock_resp.username = "newuser"
        mock_resp.display_name = "New User"
        mock_uc.execute.return_value = (mock_resp, None)

        mock_svc = MagicMock()

        def _mock_uc_cls(*args, **kwargs):
            return mock_uc

        db = MagicMock()

        with (
            patch("app.api.routes.auth.RegisterUserUseCase", side_effect=_mock_uc_cls),
            patch("packages.domain.points_service.PointsService", return_value=mock_svc),
        ):
            req = RegisterRequest(email="n***@example.com", password="Secret123!", username="newuser")
            resp = await auth.register(request=req, user_repository=MagicMock(), email_service=MagicMock(), db=db)
            assert resp.user_id == "new-user-1"
            mock_svc.add_points.assert_called_once()
            call_kwargs = mock_svc.add_points.call_args.kwargs
            assert call_kwargs["user_id"] == "new-user-1"
            assert call_kwargs["amount"] == 50
            assert call_kwargs["source"] == "task_reward"

    @pytest.mark.asyncio
    async def test_no_bonus_when_disabled(self, mock_settings):
        """关闭积分时不送分。"""
        from app.api.routes import auth
        from app.api.routes.auth import RegisterRequest

        mock_uc = MagicMock()
        mock_resp = MagicMock()
        mock_resp.user_id = "new-user-2"
        mock_resp.email = "n***@example.com"
        mock_resp.username = "newuser2"
        mock_resp.display_name = "New User 2"
        mock_uc.execute.return_value = (mock_resp, None)

        def _mock_uc_cls(*args, **kwargs):
            return mock_uc

        db = MagicMock()
        with (
            patch("app.api.routes.auth.RegisterUserUseCase", side_effect=_mock_uc_cls),
            patch("packages.domain.points_service.PointsService") as MockSvc,
        ):
            req = RegisterRequest(email="n***@example.com", password="Secret123!", username="newuser2")
            resp = await auth.register(request=req, user_repository=MagicMock(), email_service=MagicMock(), db=db)
            MockSvc.assert_not_called()
            assert resp.user_id == "new-user-2"

    @pytest.mark.asyncio
    async def test_bonus_failure_does_not_break_register(self, mock_settings):
        """送积分失败不应影响注册流程。"""
        from app.api.routes import auth
        from app.api.routes.auth import RegisterRequest

        mock_settings.points_enabled = True
        mock_uc = MagicMock()
        mock_resp = MagicMock()
        mock_resp.user_id = "new-user-3"
        mock_resp.email = "n***@example.com"
        mock_resp.username = "newuser3"
        mock_resp.display_name = "New User 3"
        mock_uc.execute.return_value = (mock_resp, None)

        mock_svc = MagicMock()
        mock_svc.add_points.side_effect = Exception("DB error")

        def _mock_uc_cls(*args, **kwargs):
            return mock_uc

        db = MagicMock()
        with (
            patch("app.api.routes.auth.RegisterUserUseCase", side_effect=_mock_uc_cls),
            patch("packages.domain.points_service.PointsService", return_value=mock_svc),
        ):
            req = RegisterRequest(email="n***@example.com", password="Secret123!", username="newuser3")
            resp = await auth.register(request=req, user_repository=MagicMock(), email_service=MagicMock(), db=db)
            assert resp.user_id == "new-user-3"