Clean up temp directory in test tear down

testPKNotInt was failing on Windows because the file was still open (cherry picked from commit 08065c3)
qgis · Jun 2, 2020 · ee0e5e9 · ee0e5e9
1 parent 3f2fb5a
commit ee0e5e9
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/src/python/test_provider_spatialite.py b/tests/src/python/test_provider_spatialite.py
@@ -964,7 +964,9 @@ def testEncodeUri(self):
     def testPKNotInt(self):
         """ Check when primary key is not an integer """
         # create test db
-        dbname = os.path.join(tempfile.mkdtemp(), "test_pknotint.sqlite")
+        tmpdir = tempfile.mkdtemp()
+        self.dirs_to_cleanup.append(tmpdir)
+        dbname = os.path.join(tmpdir, "test_pknotint.sqlite")
         con = spatialite_connect(dbname, isolation_level=None)
         cur = con.cursor()
 
@@ -1010,8 +1012,6 @@ def testPKNotInt(self):
 
         con.close()
 
-        basepath, filename = os.path.split(dbname)
-        shutil.rmtree(basepath)
 
     def testLoadStyle(self):
         """Check that we can store and load a style"""