fix isolation level test

ydb-platform · Oct 23, 2024 · 7ba4f03 · 7ba4f03
1 parent 47cce3a
commit 7ba4f03
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/test_connection.py b/tests/test_connection.py
@@ -16,6 +16,7 @@ async def _test_isolation_level_read_only(
         isolation_level: str,
         read_only: bool,
     ) -> None:
+        connection.set_isolation_level("AUTOCOMMIT")
         async with connection.cursor() as cursor:
             with suppress(dbapi.DatabaseError):
                 await cursor.execute("DROP TABLE foo")
@@ -39,6 +40,8 @@ async def _test_isolation_level_read_only(
 
         await connection.rollback()
 
+        connection.set_isolation_level("AUTOCOMMIT")
+
         async with connection.cursor() as cursor:
             await cursor.execute("DROP TABLE foo")