type hint fix

Author: krylosov-aa

context_async_sqlalchemy/test_utils.py

@@ -17,7 +17,7 @@
 @asynccontextmanager
 async def rollback_session(
     connection: DBConnect,
-) -> AsyncGenerator[AsyncSession]:
+) -> AsyncGenerator[AsyncSession, None]:
     """A session that always rolls back"""
     session_maker = await connection.session_maker()
     async with session_maker() as session:

docs/api/index.html

@@ -379,7 +379,7 @@ <h3 id="rollback_session">rollback_session</h3>
 <pre><code class="language-python">@asynccontextmanager
 async def rollback_session(
     connection: DBConnect,
-) -&gt; AsyncGenerator[AsyncSession]:
+) -&gt; AsyncGenerator[AsyncSession, None]:
 </code></pre>
 <p>A context manager that creates a session which is automatically rolled
 back at the end.

docs/search/search_index.json

@@ -153,7 +153,7 @@ <h2 id="create-session-with-autorollback">Create session with autorollback</h2>
 <pre><code class="language-python">from context_async_sqlalchemy.test_utils import rollback_session
 
 @pytest_asyncio.fixture
-async def db_session_test() -&gt; AsyncGenerator[AsyncSession]:
+async def db_session_test() -&gt; AsyncGenerator[AsyncSession, None]:
     &quot;&quot;&quot;The session that is used inside the test&quot;&quot;&quot;
     async with rollback_session(connection) as session:
         yield session

docs/testing/index.html

@@ -306,7 +306,7 @@ You can read more about testing here: [Testing](testing.md)
 @asynccontextmanager
 async def rollback_session(
     connection: DBConnect,
-) -> AsyncGenerator[AsyncSession]:
+) -> AsyncGenerator[AsyncSession, None]:
 ```
 A context manager that creates a session which is automatically rolled
 back at the end.

docs_sources/docs/api.md

@@ -68,7 +68,7 @@ application and the tests.](https://github.com/krylosov-aa/context-async-sqlalch
 from context_async_sqlalchemy.test_utils import rollback_session
 
 @pytest_asyncio.fixture
-async def db_session_test() -> AsyncGenerator[AsyncSession]:
+async def db_session_test() -> AsyncGenerator[AsyncSession, None]:
     """The session that is used inside the test"""
     async with rollback_session(connection) as session:
         yield session

docs_sources/docs/testing.md

@@ -37,7 +37,7 @@ async def client(app: FastAPI) -> AsyncGenerator[AsyncClient]:
 
 
 @pytest_asyncio.fixture
-async def db_session_test() -> AsyncGenerator[AsyncSession]:
+async def db_session_test() -> AsyncGenerator[AsyncSession, None]:
     """The session that is used inside the test"""
     async with rollback_session(connection) as session:
         yield session

examples/fastapi_example/tests/conftest.py

@@ -37,7 +37,7 @@ async def client(app: FastAPI) -> AsyncGenerator[AsyncClient]:
 
 
 @pytest_asyncio.fixture
-async def db_session_test() -> AsyncGenerator[AsyncSession]:
+async def db_session_test() -> AsyncGenerator[AsyncSession, None]:
     """The session that is used inside the test"""
     async with rollback_session(connection) as session:
         yield session

examples/fastapi_with_pure_asgi_example/tests/conftest.py

@@ -37,7 +37,7 @@ async def client(app: Starlette) -> AsyncGenerator[AsyncClient]:
 
 
 @pytest_asyncio.fixture
-async def db_session_test() -> AsyncGenerator[AsyncSession]:
+async def db_session_test() -> AsyncGenerator[AsyncSession, None]:
     """The session that is used inside the test"""
     async with rollback_session(connection) as session:
         yield session