run in new ctx fix

Author: krylosov-aa

context_async_sqlalchemy/context.py

@@ -117,7 +117,7 @@ async def _new_ctx_wrapper(
     *args: Any,
     **kwargs: Any,
 ) -> AsyncCallableResult:
-    token = init_db_session_ctx()
+    token = _init_db_session_ctx()
     try:
         return await callable_func(*args, **kwargs)
     finally:

exmaples/fastapi_example/routes/multiple_session_usage.py

@@ -21,6 +21,7 @@ async def handler_multiple_sessions() -> None:
         simultaneously. For example, to run several queries concurrently.
     """
     await asyncio.gather(
+        _insert(),
         run_in_new_ctx(_insert),
         run_in_new_ctx(_insert_manual),
         _insert_non_ctx(),

exmaples/fastapi_example/tests/non_transactional/test_routes.py

@@ -98,7 +98,7 @@ async def test_example_multiple_sessions(
 
     result = await db_session_test.execute(select(ExampleTable))
     rows = result.scalars().all()
-    assert len(rows) == 4
+    assert len(rows) == 5
     for row in rows:
         assert row.text == "example_multiple_sessions"