From 48873b16c3bd1ee555b433c3dec728628df818ee Mon Sep 17 00:00:00 2001 From: "Erlend E. Aasland" Date: Tue, 14 Jun 2022 10:34:13 +0200 Subject: [PATCH] gh-93795: Use test.support TESTFN/unlink in sqlite3 tests --- Lib/test/test_sqlite3/test_transactions.py | 32 ++++++++++------------ 1 file changed, 15 insertions(+), 17 deletions(-) diff --git a/Lib/test/test_sqlite3/test_transactions.py b/Lib/test/test_sqlite3/test_transactions.py index 040ab1ee608cf1..21acad8383c29e 100644 --- a/Lib/test/test_sqlite3/test_transactions.py +++ b/Lib/test/test_sqlite3/test_transactions.py @@ -23,35 +23,33 @@ 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)")