LangBot/tests/unit_tests/api/service/test_user_service.py

"""
Unit tests for UserService.

Tests user management operations including:
- User initialization check
- Local user creation and authentication
- JWT token generation and verification
- Password management (reset, change, set)
- Space account management

Source: src/langbot/pkg/api/http/service/user.py
"""

from __future__ import annotations

import pytest
import jwt
import datetime

import sqlalchemy
from argon2 import PasswordHasher
from argon2.exceptions import VerifyMismatchError
from unittest.mock import AsyncMock, Mock
from types import SimpleNamespace

from langbot.pkg.api.http.service.user import (
    ControlPlaneDirectoryRequiredError,
    UserService,
)
from langbot.pkg.entity.persistence.user import AccountSource, AccountStatus, User
from langbot.pkg.entity.errors.account import (
    AccountEmailMismatchError,
    SpaceAccountBindingRequiredError,
    SpaceAccountNotRegisteredError,
)
from langbot.pkg.utils.bounded_executor import BlockingWorkCapacityError


pytestmark = pytest.mark.asyncio


async def test_password_hashing_rejects_concurrent_waiters() -> None:
    service = UserService(SimpleNamespace())
    await service._password_hash_lock.acquire()
    try:
        with pytest.raises(
            BlockingWorkCapacityError,
            match='Password hashing capacity reached',
        ):
            await service._hash_password('secret')
    finally:
        service._password_hash_lock.release()


class TestSpaceOAuthState:
    async def test_login_state_is_opaque_single_use(self):
        service = UserService(SimpleNamespace())

        state = await service.issue_space_oauth_state('login')

        assert state.count('.') == 0
        assert await service.consume_space_oauth_state(state, 'login') is None
        with pytest.raises(ValueError, match='Invalid or expired OAuth state'):
            await service.consume_space_oauth_state(state, 'login')

    async def test_bind_state_resolves_only_bound_active_account(self):
        service = UserService(SimpleNamespace())
        account = SimpleNamespace(uuid='account-a', status=AccountStatus.ACTIVE.value)
        service.get_user_by_uuid = AsyncMock(return_value=account)

        state = await service.issue_space_oauth_state('bind', account_uuid='account-a')

        assert await service.consume_space_oauth_state(state, 'bind') is account
        service.get_user_by_uuid.assert_awaited_once_with('account-a')

    async def test_state_purpose_mismatch_is_rejected_and_consumed(self):
        service = UserService(SimpleNamespace())
        state = await service.issue_space_oauth_state('login')

        with pytest.raises(ValueError, match='Invalid or expired OAuth state'):
            await service.consume_space_oauth_state(state, 'bind')
        with pytest.raises(ValueError, match='Invalid or expired OAuth state'):
            await service.consume_space_oauth_state(state, 'login')

    async def test_expired_state_is_rejected(self):
        service = UserService(SimpleNamespace())
        state = await service.issue_space_oauth_state('login')
        digest = service._space_oauth_state_digest(state)
        purpose, account_uuid, _, launch_workspace_uuid = service._space_oauth_states[digest]
        service._space_oauth_states[digest] = (purpose, account_uuid, 0, launch_workspace_uuid)

        with pytest.raises(ValueError, match='Invalid or expired OAuth state'):
            await service.consume_space_oauth_state(state, 'login')

    async def test_login_state_can_carry_launch_workspace_without_changing_normal_return(self):
        service = UserService(SimpleNamespace())
        state = await service.issue_space_oauth_state(
            'login',
            launch_workspace_uuid='workspace-a',
        )

        assert await service.consume_space_oauth_state(state, 'login') is None

        state = await service.issue_space_oauth_state(
            'login',
            launch_workspace_uuid='workspace-a',
        )
        consumed = await service.consume_space_oauth_state_details(state, 'login')
        assert consumed.account is None
        assert consumed.launch_workspace_uuid == 'workspace-a'

    async def test_issue_state_does_not_scan_all_live_states(self, monkeypatch):
        service = UserService(SimpleNamespace())
        for _ in range(512):
            await service.issue_space_oauth_state('login')

        class NoGlobalIterationDict(dict):
            def __iter__(self):
                raise AssertionError('OAuth state issuance scanned all live states')

            def keys(self):
                raise AssertionError('OAuth state issuance scanned all live states')

            def items(self):
                raise AssertionError('OAuth state issuance scanned all live states')

            def values(self):
                raise AssertionError('OAuth state issuance scanned all live states')

        guarded_states = NoGlobalIterationDict(service._space_oauth_states)
        monkeypatch.setattr(service, '_space_oauth_states', guarded_states)

        state = await service.issue_space_oauth_state('login')

        assert await service.consume_space_oauth_state(state, 'login') is None
        assert len(guarded_states) == 512


def _create_mock_user(
    email: str = '[email protected]',
    password: str = 'hashed_password',
    account_type: str = 'local',
    space_account_uuid: str = None,
) -> Mock:
    """Helper to create mock User entity."""
    user = Mock(spec=User)
    user.user = email
    user.uuid = f'account-{email}'
    user.password = password
    user.account_type = account_type
    user.space_account_uuid = space_account_uuid
    return user


def _create_mock_result(items: list = None, first_item=None):
    """Create mock result object for persistence queries."""
    result = Mock()
    result.all = Mock(return_value=items or [])
    result.first = Mock(return_value=first_item)
    return result


class TestUserServiceIsInitialized:
    """Tests for is_initialized method."""

    async def test_is_initialized_returns_true_when_users_exist(self):
        """Returns True when at least one user exists."""
        # Setup
        ap = SimpleNamespace()
        ap.persistence_mgr = SimpleNamespace()
        mock_user = _create_mock_user()
        mock_result = _create_mock_result([mock_user])
        ap.persistence_mgr.execute_async = AsyncMock(return_value=mock_result)

        service = UserService(ap)

        # Execute
        result = await service.is_initialized()

        # Verify
        assert result is True

    async def test_is_initialized_returns_false_when_no_users(self):
        """Returns False when no users exist."""
        # Setup
        ap = SimpleNamespace()
        ap.persistence_mgr = SimpleNamespace()
        mock_result = _create_mock_result([])
        ap.persistence_mgr.execute_async = AsyncMock(return_value=mock_result)

        service = UserService(ap)

        # Execute
        result = await service.is_initialized()

        # Verify
        assert result is False

    async def test_is_initialized_returns_false_on_none_result(self):
        """Returns False when result is None."""
        # Setup
        ap = SimpleNamespace()
        ap.persistence_mgr = SimpleNamespace()
        mock_result = Mock()
        mock_result.all = Mock(return_value=None)
        ap.persistence_mgr.execute_async = AsyncMock(return_value=mock_result)

        service = UserService(ap)

        # Execute
        result = await service.is_initialized()

        # Verify
        assert result is False


class TestUserServiceGetLoginCapabilities:
    """Tests for public login capability discovery."""

    async def test_uses_explicit_identity_discovery_scope(self):
        discovery_result = Mock()
        discovery_result.one = Mock(return_value=(1, 2))
        discovery_session = SimpleNamespace(execute=AsyncMock(return_value=discovery_result))

        class DiscoveryContext:
            async def __aenter__(self):
                return SimpleNamespace(session=discovery_session)

            async def __aexit__(self, exc_type, exc, tb):
                return False

        ap = SimpleNamespace()
        ap.persistence_mgr = SimpleNamespace(
            current_session=Mock(return_value=None),
            identity_discovery_uow=Mock(return_value=DiscoveryContext()),
            execute_async=AsyncMock(side_effect=AssertionError('unscoped persistence access')),
        )
        ap.workspace_service = SimpleNamespace(instance_uuid='instance-a')
        service = UserService(ap)

        result = await service.get_login_capabilities()

        assert result == {
            'password_login_enabled': True,
            'space_login_enabled': True,
        }
        ap.persistence_mgr.identity_discovery_uow.assert_called_once()
        discovery_session.execute.assert_awaited_once()
        ap.persistence_mgr.execute_async.assert_not_awaited()


class TestUserServiceGetUserByEmail:
    """Tests for get_user_by_email method."""

    async def test_get_user_by_email_found(self):
        """Returns user when found."""
        # Setup
        ap = SimpleNamespace()
        ap.persistence_mgr = SimpleNamespace()
        mock_user = _create_mock_user(email='[email protected]')
        mock_result = _create_mock_result([mock_user])
        ap.persistence_mgr.execute_async = AsyncMock(return_value=mock_result)

        service = UserService(ap)

        # Execute
        result = await service.get_user_by_email('[email protected]')

        # Verify
        assert result is not None
        assert result.user == '[email protected]'

    async def test_get_user_by_email_not_found(self):
        """Returns None when user not found."""
        # Setup
        ap = SimpleNamespace()
        ap.persistence_mgr = SimpleNamespace()
        mock_result = _create_mock_result([])
        ap.persistence_mgr.execute_async = AsyncMock(return_value=mock_result)

        service = UserService(ap)

        # Execute
        result = await service.get_user_by_email('[email protected]')

        # Verify
        assert result is None

    async def test_get_user_by_email_empty_string(self):
        """Handles empty email string."""
        # Setup
        ap = SimpleNamespace()
        ap.persistence_mgr = SimpleNamespace()
        mock_result = _create_mock_result([])
        ap.persistence_mgr.execute_async = AsyncMock(return_value=mock_result)

        service = UserService(ap)

        # Execute
        result = await service.get_user_by_email('')

        # Verify
        assert result is None


class TestUserServiceGetUserBySpaceAccountUuid:
    """Tests for get_user_by_space_account_uuid method."""

    async def test_get_user_by_space_uuid_found(self):
        """Returns user when Space UUID found."""
        # Setup
        ap = SimpleNamespace()
        ap.persistence_mgr = SimpleNamespace()
        mock_user = _create_mock_user(
            email='[email protected]',
            account_type='space',
            space_account_uuid='space-uuid-123',
        )
        mock_result = _create_mock_result([mock_user])
        ap.persistence_mgr.execute_async = AsyncMock(return_value=mock_result)

        service = UserService(ap)

        # Execute
        result = await service.get_user_by_space_account_uuid('space-uuid-123')

        # Verify
        assert result is not None
        assert result.space_account_uuid == 'space-uuid-123'

    async def test_get_user_by_space_uuid_not_found(self):
        """Returns None when Space UUID not found."""
        # Setup
        ap = SimpleNamespace()
        ap.persistence_mgr = SimpleNamespace()
        mock_result = _create_mock_result([])
        ap.persistence_mgr.execute_async = AsyncMock(return_value=mock_result)

        service = UserService(ap)

        # Execute
        result = await service.get_user_by_space_account_uuid('nonexistent-uuid')

        # Verify
        assert result is None


class TestUserServiceAuthenticate:
    """Tests for authenticate method."""

    async def test_authenticate_user_not_found_raises_error(self):
        """Raises ValueError when user not found."""
        # Setup
        ap = SimpleNamespace()
        ap.persistence_mgr = SimpleNamespace()
        mock_result = _create_mock_result([])
        ap.persistence_mgr.execute_async = AsyncMock(return_value=mock_result)
        ap.instance_config = SimpleNamespace()
        ap.instance_config.data = {'system': {'jwt': {'secret': 'test_secret', 'expire': 3600}}}

        service = UserService(ap)

        # Execute & Verify
        with pytest.raises(ValueError, match='Resource not found'):
            await service.authenticate('[email protected]', 'password')

    async def test_authenticate_space_user_without_password_raises_error(self):
        """Raises ValueError for Space user without local password."""
        # Setup
        ap = SimpleNamespace()
        ap.persistence_mgr = SimpleNamespace()
        # Space user has empty password
        mock_user = _create_mock_user(
            email='[email protected]',
            password='',  # Empty password for Space user
            account_type='space',
        )
        mock_result = _create_mock_result([mock_user])
        ap.persistence_mgr.execute_async = AsyncMock(return_value=mock_result)

        service = UserService(ap)

        # Execute & Verify
        with pytest.raises(ValueError, match='Please sign in with your LangBot Account'):
            await service.authenticate('[email protected]', 'password')


class TestUserServiceGenerateJwtToken:
    """Tests for generate_jwt_token method."""

    async def test_generate_jwt_token_returns_valid_token(self):
        """Generates valid JWT token."""
        # Setup
        ap = SimpleNamespace()
        ap.instance_config = SimpleNamespace()
        ap.instance_config.data = {'system': {'jwt': {'secret': 'test_secret', 'expire': 3600}}}

        service = UserService(ap)

        # Execute
        token = await service.generate_jwt_token('[email protected]')

        # Verify - JWT format (base64 encoded parts)
        assert token is not None
        assert len(token) > 0
        parts = token.split('.')
        assert len(parts) == 3  # JWT has 3 parts

    async def test_generate_jwt_token_custom_expire(self):
        """Generates token with custom expiry."""
        # Setup
        ap = SimpleNamespace()
        ap.instance_config = SimpleNamespace()
        ap.instance_config.data = {'system': {'jwt': {'secret': 'test_secret', 'expire': 7200}}}

        service = UserService(ap)

        # Execute
        token = await service.generate_jwt_token('[email protected]')

        # Verify
        assert token is not None


class TestUserServiceVerifyJwtToken:
    """Tests for verify_jwt_token method."""

    async def test_verify_jwt_token_valid(self):
        """Verifies valid JWT token and returns user email."""
        # Setup
        ap = SimpleNamespace()
        ap.instance_config = SimpleNamespace()
        ap.instance_config.data = {'system': {'jwt': {'secret': 'test_secret', 'expire': 3600}}}

        service = UserService(ap)

        # First generate a valid token
        token = await service.generate_jwt_token('[email protected]')

        # Execute
        user_email = await service.verify_jwt_token(token)

        # Verify
        assert user_email == '[email protected]'

    async def test_verify_jwt_token_invalid_raises_error(self):
        """Raises error for invalid JWT token."""
        # Setup
        ap = SimpleNamespace()
        ap.instance_config = SimpleNamespace()
        ap.instance_config.data = {'system': {'jwt': {'secret': 'test_secret', 'expire': 3600}}}

        service = UserService(ap)

        # Execute & Verify - invalid token should raise JWT error
        with pytest.raises(Exception):  # jwt.DecodeError or similar
            await service.verify_jwt_token('invalid.token.here')

    async def test_verify_jwt_token_rejects_foreign_audience(self):
        ap = SimpleNamespace()
        ap.instance_config = SimpleNamespace()
        ap.instance_config.data = {'system': {'jwt': {'secret': 'test_secret', 'expire': 3600}}}
        service = UserService(ap)
        token = jwt.encode(
            {
                'user': '[email protected]',
                'iss': 'langbot-core',
                'aud': 'langbot-instance:another-instance',
                'exp': datetime.datetime.now(datetime.timezone.utc) + datetime.timedelta(hours=1),
            },
            'test_secret',
            algorithm='HS256',
        )

        with pytest.raises(jwt.InvalidAudienceError):
            await service.verify_jwt_token(token)

    async def test_verify_jwt_token_accepts_legacy_community_token_only_in_oss(self):
        ap = SimpleNamespace()
        ap.instance_config = SimpleNamespace()
        ap.instance_config.data = {'system': {'jwt': {'secret': 'test_secret', 'expire': 3600}}}
        ap.workspace_service = SimpleNamespace(
            instance_uuid='instance-a',
            policy=SimpleNamespace(multi_workspace_enabled=False),
        )
        service = UserService(ap)
        legacy_token = jwt.encode(
            {
                'user': '[email protected]',
                'iss': 'LangBot-community',
                'exp': datetime.datetime.now(datetime.timezone.utc) + datetime.timedelta(hours=1),
            },
            'test_secret',
            algorithm='HS256',
        )

        assert await service.verify_jwt_token(legacy_token) == '[email protected]'

        ap.workspace_service.policy.multi_workspace_enabled = True
        with pytest.raises(jwt.MissingRequiredClaimError):
            await service.verify_jwt_token(legacy_token)


class TestUserServiceResetPassword:
    """Tests for reset_password method."""

    async def test_reset_password_updates_password(self):
        """Updates user password."""
        # Setup
        ap = SimpleNamespace()
        ap.persistence_mgr = SimpleNamespace()
        ap.persistence_mgr.execute_async = AsyncMock()

        service = UserService(ap)

        # Execute
        await service.reset_password(' [email protected] ', 'new_password')

        ap.persistence_mgr.execute_async.assert_awaited_once()
        statement = ap.persistence_mgr.execute_async.await_args.args[0]
        assert isinstance(statement, sqlalchemy.sql.dml.Update)
        assert statement.table.name == User.__tablename__
        params = statement.compile().params
        assert params['normalized_email_1'] == '[email protected]'
        assert PasswordHasher().verify(params['password'], 'new_password')
        with pytest.raises(VerifyMismatchError):
            PasswordHasher().verify(params['password'], 'old-password')


class TestUserServiceChangePassword:
    """Tests for change_password method."""

    async def test_change_password_user_not_found_raises_error(self):
        """Raises ValueError when user not found."""
        # Setup
        ap = SimpleNamespace()
        ap.persistence_mgr = SimpleNamespace()

        service = UserService(ap)

        # Mock get_user_by_email to return None
        service.get_user_by_email = AsyncMock(return_value=None)

        # Execute & Verify
        with pytest.raises(ValueError, match='User not found'):
            await service.change_password('[email protected]', 'current', 'new')

    async def test_change_password_no_local_password_raises_error(self):
        """Raises ValueError when user has no local password set."""
        # Setup
        ap = SimpleNamespace()
        ap.persistence_mgr = SimpleNamespace()

        service = UserService(ap)

        # Mock user without password
        mock_user = _create_mock_user(email='[email protected]', password=None)
        service.get_user_by_email = AsyncMock(return_value=mock_user)

        # Execute & Verify
        with pytest.raises(ValueError, match='No local password set'):
            await service.change_password('[email protected]', 'current', 'new')


class TestUserServiceGetFirstUser:
    """Tests for get_first_user method."""

    async def test_get_first_user_found(self):
        """Returns first user when exists."""
        # Setup
        ap = SimpleNamespace()
        ap.persistence_mgr = SimpleNamespace()
        mock_user = _create_mock_user(email='[email protected]')
        mock_result = _create_mock_result([mock_user])
        ap.persistence_mgr.execute_async = AsyncMock(return_value=mock_result)

        service = UserService(ap)

        # Execute
        result = await service.get_first_user()

        # Verify
        assert result is not None
        assert result.user == '[email protected]'

    async def test_get_first_user_not_found(self):
        """Returns None when no users exist."""
        # Setup
        ap = SimpleNamespace()
        ap.persistence_mgr = SimpleNamespace()
        mock_result = _create_mock_result([])
        ap.persistence_mgr.execute_async = AsyncMock(return_value=mock_result)

        service = UserService(ap)

        # Execute
        result = await service.get_first_user()

        # Verify
        assert result is None


class TestUserServiceSetPassword:
    """Tests for set_password method."""

    async def test_set_password_user_not_found_raises_error(self):
        """Raises ValueError when user not found."""
        # Setup
        ap = SimpleNamespace()
        ap.persistence_mgr = SimpleNamespace()

        service = UserService(ap)

        # Mock get_user_by_email to return None
        service.get_user_by_email = AsyncMock(return_value=None)

        # Execute & Verify
        with pytest.raises(ValueError, match='User not found'):
            await service.set_password('[email protected]', 'new_password')

    async def test_set_password_with_existing_password_requires_current(self):
        """Requires current password when user has existing password."""
        # Setup
        ap = SimpleNamespace()
        ap.persistence_mgr = SimpleNamespace()

        service = UserService(ap)

        # Mock user with existing password
        mock_user = _create_mock_user(email='[email protected]', password='hashed_old_password')
        service.get_user_by_email = AsyncMock(return_value=mock_user)

        # Execute & Verify - should raise when no current_password provided
        with pytest.raises(ValueError, match='Current password is required'):
            await service.set_password('[email protected]', 'new_password')


class TestUserServiceCreateOrUpdateSpaceUser:
    """Tests for create_or_update_space_user method."""

    async def test_create_or_update_existing_space_user(self):
        """Updates existing Space user tokens."""
        # Setup
        ap = SimpleNamespace()
        ap.persistence_mgr = SimpleNamespace()
        ap.provider_service = SimpleNamespace()
        ap.provider_service.update_space_model_provider_api_keys = AsyncMock()

        service = UserService(ap)

        # Mock existing Space user
        existing_user = _create_mock_user(
            email='[email protected]',
            account_type='space',
            space_account_uuid='existing-space-uuid',
        )
        service.get_user_by_space_account_uuid = AsyncMock(return_value=existing_user)
        service.get_user_by_email = AsyncMock(return_value=None)
        service.is_initialized = AsyncMock(return_value=True)

        ap.persistence_mgr.execute_async = AsyncMock()

        # Execute
        updated_user = await service.create_or_update_space_user(
            space_account_uuid='existing-space-uuid',
            email='[email protected]',
            access_token='new_access_token',
            refresh_token='new_refresh_token',
            api_key='new_api_key',
            expires_in=3600,
        )

        # Verify - update was called and user returned
        ap.persistence_mgr.execute_async.assert_called()
        assert updated_user.space_account_uuid == 'existing-space-uuid'

    async def test_cloud_login_updates_only_the_projected_space_account(self):
        projected = SimpleNamespace(
            uuid='projected-space-uuid',
            user='Cloud Owner',
            normalized_email='[email protected]',
            password='',
            account_type='space',
            status=AccountStatus.ACTIVE.value,
            source=AccountSource.CLOUD_PROJECTION.value,
            projection_revision=7,
            space_account_uuid='projected-space-uuid',
        )
        persistence = SimpleNamespace(execute_async=AsyncMock())
        ap = SimpleNamespace(
            persistence_mgr=persistence,
            workspace_service=SimpleNamespace(policy=SimpleNamespace(multi_workspace_enabled=True)),
        )
        service = UserService(ap)
        service.get_user_by_space_account_uuid = AsyncMock(side_effect=[projected, projected])

        result = await service.create_or_update_space_user(
            space_account_uuid='projected-space-uuid',
            email='[email protected]',
            access_token='access-token',
            refresh_token='refresh-token',
            api_key='api-key',
            expires_in=3600,
        )

        assert result is projected
        persistence.execute_async.assert_awaited_once()

    async def test_cloud_login_never_creates_an_unprojected_account(self):
        persistence = SimpleNamespace(execute_async=AsyncMock())
        ap = SimpleNamespace(
            persistence_mgr=persistence,
            workspace_service=SimpleNamespace(policy=SimpleNamespace(multi_workspace_enabled=True)),
        )
        service = UserService(ap)
        service.get_user_by_space_account_uuid = AsyncMock(return_value=None)

        with pytest.raises(
            ControlPlaneDirectoryRequiredError,
            match='verified Cloud directory',
        ):
            await service.create_or_update_space_user(
                space_account_uuid='unknown-space-uuid',
                email='[email protected]',
                access_token='access-token',
                refresh_token='refresh-token',
                api_key='api-key',
                expires_in=3600,
            )

        persistence.execute_async.assert_not_awaited()

    async def test_cloud_invitation_registration_requires_space_identity(self):
        ap = SimpleNamespace(
            workspace_service=SimpleNamespace(policy=SimpleNamespace(multi_workspace_enabled=True)),
        )
        service = UserService(ap)

        with pytest.raises(ControlPlaneDirectoryRequiredError, match='LangBot Account'):
            await service.register_invited_account('invite-token', '[email protected]', 'password')

    async def test_create_or_update_new_space_user_first_init(self):
        """Creates new Space user on first initialization."""
        # Setup
        ap = SimpleNamespace()
        ap.persistence_mgr = SimpleNamespace()
        ap.provider_service = SimpleNamespace()
        ap.provider_service.update_space_model_provider_api_keys = AsyncMock()

        service = UserService(ap)

        # Mock new user to be returned after creation
        new_user = _create_mock_user(
            email='[email protected]',
            account_type='space',
            space_account_uuid='new-space-uuid',
        )

        # First call (line 138) returns None, second call (line 194) returns new_user
        call_count = 0

        async def mock_get_by_space_uuid(uuid):
            nonlocal call_count
            call_count += 1
            if call_count == 1:  # First check for existing user
                return None
            return new_user  # After insert, return the new user

        service.get_user_by_space_account_uuid = AsyncMock(side_effect=mock_get_by_space_uuid)
        service.get_user_by_email = AsyncMock(return_value=None)
        service.is_initialized = AsyncMock(return_value=False)  # Not initialized

        ap.persistence_mgr.execute_async = AsyncMock()

        # Execute
        result = await service.create_or_update_space_user(
            space_account_uuid='new-space-uuid',
            email='[email protected]',
            access_token='access_token',
            refresh_token='refresh_token',
            api_key='api_key',
            expires_in=3600,
        )

        # Verify
        assert result.space_account_uuid == 'new-space-uuid'

    async def test_create_or_update_space_user_already_initialized_reports_unknown_space_email(self):
        """Unknown Space email is distinct from an existing local Account collision."""
        # Setup
        ap = SimpleNamespace()
        ap.persistence_mgr = SimpleNamespace()
        ap.provider_service = SimpleNamespace()
        ap.provider_service.update_space_model_provider_api_keys = AsyncMock()

        service = UserService(ap)

        # Mock system already initialized, no matching users
        service.get_user_by_space_account_uuid = AsyncMock(return_value=None)
        service.get_user_by_email = AsyncMock(return_value=None)
        service.is_initialized = AsyncMock(return_value=True)  # Already initialized

        # Execute & Verify
        with pytest.raises(SpaceAccountNotRegisteredError):
            await service.create_or_update_space_user(
                space_account_uuid='unknown-space-uuid',
                email='[email protected]',
                access_token='token',
                refresh_token='refresh',
                api_key='key',
                expires_in=3600,
            )

    async def test_unknown_space_subject_cannot_claim_existing_account_by_email(self):
        """An OAuth login collision requires the explicit account-bound bind flow."""
        existing_user = _create_mock_user(
            email='[email protected]',
            account_type='local',
            space_account_uuid=None,
        )
        ap = SimpleNamespace(
            persistence_mgr=SimpleNamespace(execute_async=AsyncMock()),
            provider_service=SimpleNamespace(update_space_model_provider_api_keys=AsyncMock()),
            space_service=SimpleNamespace(
                get_user_info_raw=AsyncMock(
                    return_value={
                        'account': {
                            'uuid': 'attacker-space-subject',
                            'email': '[email protected]',
                        },
                        'api_key': 'attacker-api-key',
                    }
                )
            ),
        )
        service = UserService(ap)
        service.get_user_by_space_account_uuid = AsyncMock(return_value=None)
        service.get_user_by_email = AsyncMock(return_value=existing_user)
        service.generate_jwt_token = AsyncMock(return_value='must-not-be-issued')

        with pytest.raises(SpaceAccountBindingRequiredError):
            await service.authenticate_space_user(
                'attacker-access-token',
                'attacker-refresh-token',
                3600,
            )

        ap.persistence_mgr.execute_async.assert_not_awaited()
        ap.provider_service.update_space_model_provider_api_keys.assert_not_awaited()
        service.generate_jwt_token.assert_not_awaited()

    async def test_oss_space_provider_refresh_requires_workspace_owner(self):
        member_account = _create_mock_user(email='[email protected]', space_account_uuid='space-member')
        access = SimpleNamespace(
            workspace=SimpleNamespace(uuid='workspace-a'),
            membership=SimpleNamespace(role='admin'),
        )
        provider_service = SimpleNamespace(update_space_model_provider_api_keys=AsyncMock())
        ap = SimpleNamespace(
            workspace_service=SimpleNamespace(policy=SimpleNamespace(multi_workspace_enabled=False)),
            workspace_collaboration_service=SimpleNamespace(list_account_workspaces=AsyncMock(return_value=[access])),
            provider_service=provider_service,
        )

        await UserService(ap)._update_space_provider_for_account(member_account, 'member-api-key')

        provider_service.update_space_model_provider_api_keys.assert_not_awaited()

    async def test_oss_space_provider_refresh_uses_workspace_owner_credentials(self):
        owner_account = _create_mock_user(email='[email protected]', space_account_uuid='space-owner')
        access = SimpleNamespace(
            workspace=SimpleNamespace(uuid='workspace-a'),
            membership=SimpleNamespace(role='owner'),
        )
        provider_service = SimpleNamespace(update_space_model_provider_api_keys=AsyncMock())
        ap = SimpleNamespace(
            workspace_service=SimpleNamespace(policy=SimpleNamespace(multi_workspace_enabled=False)),
            workspace_collaboration_service=SimpleNamespace(list_account_workspaces=AsyncMock(return_value=[access])),
            provider_service=provider_service,
        )

        await UserService(ap)._update_space_provider_for_account(owner_account, 'owner-api-key')

        provider_service.update_space_model_provider_api_keys.assert_awaited_once_with('workspace-a', 'owner-api-key')

    async def test_create_or_update_space_user_no_expiry(self):
        """Creates Space user without token expiry."""
        # Setup
        ap = SimpleNamespace()
        ap.persistence_mgr = SimpleNamespace()
        ap.provider_service = SimpleNamespace()
        ap.provider_service.update_space_model_provider_api_keys = AsyncMock()

        service = UserService(ap)

        new_user = _create_mock_user(
            email='[email protected]',
            account_type='space',
            space_account_uuid='noexpiry-uuid',
        )

        # First call (line 138) returns None, second call (line 194) returns new_user
        call_count = 0

        async def mock_get_by_space_uuid(uuid):
            nonlocal call_count
            call_count += 1
            if call_count == 1:  # First check for existing user
                return None
            return new_user  # After insert, return the new user

        service.get_user_by_space_account_uuid = AsyncMock(side_effect=mock_get_by_space_uuid)
        service.get_user_by_email = AsyncMock(return_value=None)
        service.is_initialized = AsyncMock(return_value=False)

        ap.persistence_mgr.execute_async = AsyncMock()

        # Execute with expires_in=0 (no expiry)
        result = await service.create_or_update_space_user(
            space_account_uuid='noexpiry-uuid',
            email='[email protected]',
            access_token='token',
            refresh_token='refresh',
            api_key='key',
            expires_in=0,  # No expiry
        )

        # Verify
        assert result is not None
        assert result.space_account_uuid == 'noexpiry-uuid'

    async def test_bind_space_account_rejects_different_email(self):
        service = UserService(SimpleNamespace())
        service.get_user_by_email = AsyncMock(return_value=_create_mock_user(email='[email protected]'))
        service.ap.space_service = SimpleNamespace(
            exchange_oauth_code=AsyncMock(
                return_value={'access_token': 'access', 'refresh_token': 'refresh', 'expires_in': 3600}
            ),
            get_user_info_raw=AsyncMock(
                return_value={
                    'account': {'uuid': 'space-other', 'email': '[email protected]'},
                    'api_key': 'key',
                }
            ),
        )
        service.get_user_by_space_account_uuid = AsyncMock(return_value=None)
        service._identity_execute = AsyncMock()

        with pytest.raises(AccountEmailMismatchError):
            await service.bind_space_account('[email protected]', 'code')

        service._identity_execute.assert_not_awaited()

    @pytest.mark.asyncio
    async def test_get_workspace_owner_returns_user_object_from_core_connection_result(self):
        service = UserService(SimpleNamespace())
        owner = _create_mock_user('[email protected]', password='pw')
        service.ap.persistence_mgr = SimpleNamespace(
            current_session=lambda: SimpleNamespace(scalar=AsyncMock(return_value=owner)),
        )

        resolved = await service.get_workspace_owner('workspace-1')

        assert resolved is owner


class TestUserServiceLoginCapabilities:
    async def test_capabilities_are_derived_from_all_accounts(self):
        result = SimpleNamespace(one=lambda: (2, 1))
        ap = SimpleNamespace(persistence_mgr=SimpleNamespace(execute_async=AsyncMock(return_value=result)))

        capabilities = await UserService(ap).get_login_capabilities()

        assert capabilities == {'password_login_enabled': True, 'space_login_enabled': True}

    async def test_capabilities_disable_absent_login_methods(self):
        result = SimpleNamespace(one=lambda: (0, 0))
        ap = SimpleNamespace(persistence_mgr=SimpleNamespace(execute_async=AsyncMock(return_value=result)))

        capabilities = await UserService(ap).get_login_capabilities()

        assert capabilities == {'password_login_enabled': False, 'space_login_enabled': False}


class TestUserServiceCreateUserLock:
    """Tests for create_user_lock attribute."""

    def test_create_user_lock_initialized(self):
        """Verify create_user_lock is initialized as asyncio.Lock."""
        # Setup
        ap = SimpleNamespace()

        service = UserService(ap)

        # Verify lock exists
        assert hasattr(service, '_create_user_lock')
        assert service._create_user_lock is not None