diff --git a/tests/auth/test_auth_service.py b/tests/auth/test_auth_service.py index 98f1a402..76a056a1 100644 --- a/tests/auth/test_auth_service.py +++ b/tests/auth/test_auth_service.py @@ -5,6 +5,8 @@ from auth.orm import Author @pytest.mark.asyncio async def test_ensure_user_has_reader_role(): + """Тест добавления роли reader пользователю""" + auth_service = AuthService() # Создаем тестового пользователя без роли reader @@ -18,17 +20,18 @@ async def test_ensure_user_has_reader_role(): session.commit() user_id = test_author.id - # Проверяем, что роль reader добавляется - result = await auth_service.ensure_user_has_reader_role(user_id) - assert result is True + 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 - - # Очищаем тестовые данные - with local_session() as session: - test_author = session.query(Author).filter_by(id=user_id).first() - if test_author: - session.delete(test_author) - session.commit() + # Проверяем, что при повторном вызове возвращается True + result = await auth_service.ensure_user_has_reader_role(user_id) + assert result is True + finally: + # Очищаем тестовые данные + with local_session() as session: + test_author = session.query(Author).filter_by(id=user_id).first() + if test_author: + session.delete(test_author) + session.commit()