also test with transaction

tests/test_database.py

@@ -238,14 +238,17 @@ async def test_branch_drop_recreate(self):
         finally:
             await tb.drop_db(self.con, 'test_db_drop')
 
-    async def test_branch_drop_disconnect(self):
+    async def _test_branch_drop_disconnect(self, *, with_transaction):
         if not self.has_create_database:
             self.skipTest("create branch is not supported by the backend")
 
         await self.con.execute('CREATE EMPTY BRANCH test_db_disconnect;')
         conn = await self.connect(database='test_db_disconnect')
 
         try:
+            if with_transaction:
+                await conn.query('START TRANSACTION')
+
             dbname = await conn.query(
                 'SELECT sys::get_current_database();')
             self.assertEqual(dbname, ['test_db_disconnect'])
@@ -264,6 +267,12 @@ async def test_branch_drop_disconnect(self):
             except edgedb.UnknownDatabaseError:
                 pass
 
+    async def test_branch_drop_disconnect_01(self):
+        await self._test_branch_drop_disconnect(with_transaction=False)
+
+    async def test_branch_drop_disconnect_02(self):
+        await self._test_branch_drop_disconnect(with_transaction=True)
+
     async def test_branch_rename_disconnect(self):
         if not self.has_create_database:
             self.skipTest("create branch is not supported by the backend")