Close the connections opened by the iterdump virtual-table test

Author: tonghuaroot

Lib/test/test_sqlite3/test_dump.py

@@ -2,6 +2,7 @@
 
 import sqlite3
 import unittest
+from contextlib import closing
 
 from test.support.os_helper import TESTFN, unlink
 
@@ -253,7 +254,7 @@ def test_dump_virtual_tables(self):
     def test_dump_virtual_table_data_roundtrip(self):
         # gh-153729: a populated virtual table must round-trip through iterdump().
         self.addCleanup(unlink, TESTFN)
-        with sqlite3.connect(TESTFN) as src:
+        with closing(sqlite3.connect(TESTFN)) as src:
             src.execute("CREATE VIRTUAL TABLE test USING fts4(example)")
             src.execute("INSERT INTO test(example) VALUES('hello world')")
             src.execute("INSERT INTO test(example) VALUES('second row')")
@@ -266,12 +267,12 @@ def test_dump_virtual_table_data_roundtrip(self):
 
         restored_path = f"{TESTFN}.restored"
         self.addCleanup(unlink, restored_path)
-        with sqlite3.connect(restored_path) as dst:
+        with closing(sqlite3.connect(restored_path)) as dst:
             # Defensive mode blocks writable_schema writes to sqlite_master.
             dst.setconfig(sqlite3.SQLITE_DBCONFIG_DEFENSIVE, False)
             dst.setconfig(sqlite3.SQLITE_DBCONFIG_WRITABLE_SCHEMA, True)
             dst.executescript(script)
-        with sqlite3.connect(restored_path) as restored:
+        with closing(sqlite3.connect(restored_path)) as restored:
             rows = restored.execute(
                 "SELECT example FROM test ORDER BY docid"
             ).fetchall()