fix: annotate test context managers with Generator for ty 0.0.85

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)

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()