2233admin
diff --git a/‎tests/conftest.py‎
Lines changed: 2 additions & 1 deletion b/‎tests/conftest.py‎
Lines changed: 2 additions & 1 deletion
@@ -83,8 +83,9 @@ async def db_session(db_engine) -> AsyncGenerator[AsyncSession]:
         db_engine, class_=AsyncSession, expire_on_commit=False
     )
     async with session_factory() as session:
+        rollback = session.rollback
         yield session
-        await session.rollback()
+        await rollback()
 
 
 @pytest_asyncio.fixture(scope="function")
Original file line number	Diff line number	Diff line change
`@@ -83,8 +83,9 @@ async def db_session(db_engine) -> AsyncGenerator[AsyncSession]:`
`83`	`83`	`db_engine, class_=AsyncSession, expire_on_commit=False`
`84`	`84`	`)`
`85`	`85`	`async with session_factory() as session:`
	`86`	`+ rollback = session.rollback`
`86`	`87`	`yield session`
`87`		`- await session.rollback()`
	`88`	`+ await rollback()`
`88`	`89`
`89`	`90`
`90`	`91`	`@pytest_asyncio.fixture(scope="function")`