]> git.ipfire.org Git - thirdparty/Python/cpython.git/commitdiff
gh-93795: Use test.support TESTFN/unlink in sqlite3 tests (#93796)
authorErlend Egeberg Aasland <erlend.aasland@protonmail.com>
Tue, 14 Jun 2022 14:02:52 +0000 (16:02 +0200)
committerGitHub <noreply@github.com>
Tue, 14 Jun 2022 14:02:52 +0000 (16:02 +0200)
Lib/test/test_sqlite3/test_transactions.py

index 040ab1ee608cf17a35e1b583ebae2012e5d2681f..21acad8383c29ef11752b3484f343fc02a5a4493 100644 (file)
 import os, unittest
 import sqlite3 as sqlite
 
+from test.support import LOOPBACK_TIMEOUT
+from test.support.os_helper import TESTFN, unlink
+
 from test.test_sqlite3.test_dbapi import memory_database
 
-def get_db_path():
-    return "sqlite_testdb"
+
+TIMEOUT = LOOPBACK_TIMEOUT / 10
+
 
 class TransactionTests(unittest.TestCase):
     def setUp(self):
-        try:
-            os.remove(get_db_path())
-        except OSError:
-            pass
-
-        self.con1 = sqlite.connect(get_db_path(), timeout=0.1)
+        self.con1 = sqlite.connect(TESTFN, timeout=TIMEOUT)
         self.cur1 = self.con1.cursor()
 
-        self.con2 = sqlite.connect(get_db_path(), timeout=0.1)
+        self.con2 = sqlite.connect(TESTFN, timeout=TIMEOUT)
         self.cur2 = self.con2.cursor()
 
     def tearDown(self):
-        self.cur1.close()
-        self.con1.close()
+        try:
+            self.cur1.close()
+            self.con1.close()
 
-        self.cur2.close()
-        self.con2.close()
+            self.cur2.close()
+            self.con2.close()
 
-        try:
-            os.unlink(get_db_path())
-        except OSError:
-            pass
+        finally:
+            unlink(TESTFN)
 
     def test_dml_does_not_auto_commit_before(self):
         self.cur1.execute("create table test(i)")