core/tests/auth/test_auth_service.py

import pytest
from services.auth import AuthService
from auth.orm import Author

@pytest.mark.asyncio
async def test_ensure_user_has_reader_role(db_session):
    """Тест добавления роли reader пользователю"""
    
    auth_service = AuthService()

    # Создаем тестового пользователя без роли reader
    test_author = Author(
        email="test_reader_role@example.com",
        slug="test_reader_role",
        password="test_password"
    )
    db_session.add(test_author)
    db_session.commit()
    user_id = test_author.id

    try:
        # Проверяем, что роль reader добавляется
        result = await auth_service.ensure_user_has_reader_role(user_id)
        assert result is True

        # Проверяем, что при повторном вызове возвращается True
        result = await auth_service.ensure_user_has_reader_role(user_id)
        assert result is True
    finally:
        # Очищаем тестовые данные
        test_author = db_session.query(Author).filter_by(id=user_id).first()
        if test_author:
            db_session.delete(test_author)
        db_session.commit()