From 70be3f4412244ac5e4febc5b4a733fb02d7821c5 Mon Sep 17 00:00:00 2001 From: Artur Shiriev Date: Fri, 9 Oct 2026 21:48:59 +0300 Subject: [PATCH] fix: annotate test context managers with Generator for ty 0.0.85 --- tests/api/helpers.py | 4 ++-- tests/conftest.py | 3 ++- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/api/helpers.py b/tests/api/helpers.py index 8a78df0..3f962bc 100644 --- a/tests/api/helpers.py +++ b/tests/api/helpers.py @@ -1,7 +1,7 @@ import contextlib import typing import uuid -from collections.abc import Iterator +from collections.abc import Generator import sqlalchemy as sa from httpx import AsyncClient @@ -41,7 +41,7 @@ async def send(client: AsyncClient, chat_id: int, text: str, key: uuid.UUID | No @contextlib.contextmanager -def count_statements(session: AsyncSession) -> Iterator[list[str]]: +def count_statements(session: AsyncSession) -> Generator[list[str]]: """Collect every SQL statement sent on the test connection behind ``session`` while the block runs.""" connection: typing.Final = session.bind assert isinstance(connection, AsyncConnection) diff --git a/tests/conftest.py b/tests/conftest.py index b667c39..dc087f5 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -1,3 +1,4 @@ +import collections.abc import contextlib import typing @@ -40,7 +41,7 @@ async def di_container(app: litestar.Litestar) -> typing.AsyncIterator[modern_di @contextlib.asynccontextmanager -async def rolled_back_db_session(di_container: modern_di.Container) -> typing.AsyncIterator[AsyncSession]: +async def rolled_back_db_session(di_container: modern_di.Container) -> collections.abc.AsyncGenerator[AsyncSession]: engine = create_database_engine() connection = await engine.connect() transaction = await connection.begin()