mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-08-04 00:48:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

Improve tests for opening Sqlite by URI (GH-93047)

Browse source 
* Test with with escaped non-ascii characters
* Test read-only open of existing DB.
This commit is contained in:
Serhiy Storchaka 2022-05-21 17:38:22 +03:00 • committed by GitHub
parent 5b71b519f9
commit 4e2b664892
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 42 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

47
Lib/test/test_sqlite3/test_dbapi.py
View file

@ -661,21 +661,58 @@ class OpenTests(unittest.TestCase):
@unittest.skipIf(sys.platform == "darwin", "skipped on macOS")
@unittest.skipUnless(TESTFN_UNDECODABLE, "only works if there are undecodable paths")
def test_open_with_undecodable_path(self):
self.addCleanup(unlink, TESTFN_UNDECODABLE)
path = TESTFN_UNDECODABLE
with managed_connect(path) as cx:
self.addCleanup(unlink, path)
with managed_connect(path, in_mem=True) as cx:
self.assertTrue(os.path.exists(path))
cx.execute(self._sql)
def test_open_uri(self):
with managed_connect(TESTFN) as cx:
path = TESTFN
uri = "file:" + urllib.parse.quote(os.fsencode(path))
self.assertFalse(os.path.exists(path))
with managed_connect(uri, uri=True) as cx:
self.assertTrue(os.path.exists(path))
cx.execute(self._sql)
with managed_connect(f"file:{TESTFN}", uri=True) as cx:
def test_open_unquoted_uri(self):
path = TESTFN
uri = "file:" + path
self.assertFalse(os.path.exists(path))
with managed_connect(uri, uri=True) as cx:
self.assertTrue(os.path.exists(path))
cx.execute(self._sql)
def test_open_uri_readonly(self):
path = TESTFN
self.addCleanup(unlink, path)
uri = "file:" + urllib.parse.quote(os.fsencode(path)) + "?mode=ro"
self.assertFalse(os.path.exists(path))
# Cannot create new DB
with self.assertRaises(sqlite.OperationalError):
with managed_connect(f"file:{TESTFN}?mode=ro", uri=True) as cx:
sqlite.connect(uri, uri=True)
self.assertFalse(os.path.exists(path))
sqlite.connect(path).close()
self.assertTrue(os.path.exists(path))
# Cannot modify new DB
with managed_connect(uri, uri=True) as cx:
with self.assertRaises(sqlite.OperationalError):
cx.execute(self._sql)
@unittest.skipIf(sys.platform == "win32", "skipped on Windows")
@unittest.skipIf(sys.platform == "darwin", "skipped on macOS")
@unittest.skipUnless(TESTFN_UNDECODABLE, "only works if there are undecodable paths")
def test_open_undecodable_uri(self):
path = TESTFN_UNDECODABLE
uri = "file:" + urllib.parse.quote(path)
self.assertFalse(os.path.exists(path))
try:
with managed_connect(uri, uri=True, in_mem=True) as cx:
self.assertTrue(os.path.exists(path))
cx.execute(self._sql)
finally:
unlink(path)
def test_factory_database_arg(self):
def factory(database, *args, **kwargs):
nonlocal database_arg