From 0994b2f874009d68c295655178b9b0d5e57bd43f Mon Sep 17 00:00:00 2001 From: Rachel Kat Date: Wed, 5 Feb 2025 20:48:32 -0500 Subject: [PATCH 1/2] async_sqla_connection --- fastapi_sqla/_pytest_plugin.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/fastapi_sqla/_pytest_plugin.py b/fastapi_sqla/_pytest_plugin.py index bd143942..2aac6bdb 100644 --- a/fastapi_sqla/_pytest_plugin.py +++ b/fastapi_sqla/_pytest_plugin.py @@ -159,6 +159,8 @@ def async_engine(async_sqlalchemy_url): async def async_sqla_connection(async_engine): async with async_engine.connect() as connection: yield connection + await connection.close() + await async_engine.dispose() @fixture async def patch_new_engine(request, async_sqla_connection): From a2c6b42af9e66d2eaecea9a335cd7cdf94be40a2 Mon Sep 17 00:00:00 2001 From: Rachel Kat Date: Thu, 6 Feb 2025 17:00:05 -0500 Subject: [PATCH 2/2] fix --- fastapi_sqla/_pytest_plugin.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/fastapi_sqla/_pytest_plugin.py b/fastapi_sqla/_pytest_plugin.py index 2aac6bdb..6dd08dea 100644 --- a/fastapi_sqla/_pytest_plugin.py +++ b/fastapi_sqla/_pytest_plugin.py @@ -153,14 +153,14 @@ def async_sqlalchemy_url(db_url): @fixture def async_engine(async_sqlalchemy_url): - return create_async_engine(async_sqlalchemy_url) + engine = create_async_engine(async_sqlalchemy_url) + yield engine + engine.dispose() @fixture async def async_sqla_connection(async_engine): async with async_engine.connect() as connection: yield connection - await connection.close() - await async_engine.dispose() @fixture async def patch_new_engine(request, async_sqla_connection):