diff options
Diffstat (limited to 'Lib/test/test_dbm_gnu.py')
| -rw-r--r-- | Lib/test/test_dbm_gnu.py | 167 |
1 files changed, 0 insertions, 167 deletions
diff --git a/Lib/test/test_dbm_gnu.py b/Lib/test/test_dbm_gnu.py deleted file mode 100644 index f1c7d34..0000000 --- a/Lib/test/test_dbm_gnu.py +++ /dev/null @@ -1,167 +0,0 @@ -from test import support -gdbm = support.import_module("dbm.gnu") #skip if not supported -import unittest -import os -from test.support import TESTFN, TESTFN_NONASCII, unlink - - -filename = TESTFN - -class TestGdbm(unittest.TestCase): - @staticmethod - def setUpClass(): - if support.verbose: - try: - from _gdbm import _GDBM_VERSION as version - except ImportError: - pass - else: - print(f"gdbm version: {version}") - - def setUp(self): - self.g = None - - def tearDown(self): - if self.g is not None: - self.g.close() - unlink(filename) - - def test_key_methods(self): - self.g = gdbm.open(filename, 'c') - self.assertEqual(self.g.keys(), []) - self.g['a'] = 'b' - self.g['12345678910'] = '019237410982340912840198242' - self.g[b'bytes'] = b'data' - key_set = set(self.g.keys()) - self.assertEqual(key_set, set([b'a', b'bytes', b'12345678910'])) - self.assertIn('a', self.g) - self.assertIn(b'a', self.g) - self.assertEqual(self.g[b'bytes'], b'data') - key = self.g.firstkey() - while key: - self.assertIn(key, key_set) - key_set.remove(key) - key = self.g.nextkey(key) - # get() and setdefault() work as in the dict interface - self.assertEqual(self.g.get(b'a'), b'b') - self.assertIsNone(self.g.get(b'xxx')) - self.assertEqual(self.g.get(b'xxx', b'foo'), b'foo') - with self.assertRaises(KeyError): - self.g['xxx'] - self.assertEqual(self.g.setdefault(b'xxx', b'foo'), b'foo') - self.assertEqual(self.g[b'xxx'], b'foo') - - def test_error_conditions(self): - # Try to open a non-existent database. - unlink(filename) - self.assertRaises(gdbm.error, gdbm.open, filename, 'r') - # Try to access a closed database. - self.g = gdbm.open(filename, 'c') - self.g.close() - self.assertRaises(gdbm.error, lambda: self.g['a']) - # try pass an invalid open flag - self.assertRaises(gdbm.error, lambda: gdbm.open(filename, 'rx').close()) - - def test_flags(self): - # Test the flag parameter open() by trying all supported flag modes. - all = set(gdbm.open_flags) - # Test standard flags (presumably "crwn"). - modes = all - set('fsu') - for mode in sorted(modes): # put "c" mode first - self.g = gdbm.open(filename, mode) - self.g.close() - - # Test additional flags (presumably "fsu"). - flags = all - set('crwn') - for mode in modes: - for flag in flags: - self.g = gdbm.open(filename, mode + flag) - self.g.close() - - def test_reorganize(self): - self.g = gdbm.open(filename, 'c') - size0 = os.path.getsize(filename) - - # bpo-33901: on macOS with gdbm 1.15, an empty database uses 16 MiB - # and adding an entry of 10,000 B has no effect on the file size. - # Add size0 bytes to make sure that the file size changes. - value_size = max(size0, 10000) - self.g['x'] = 'x' * value_size - size1 = os.path.getsize(filename) - self.assertGreater(size1, size0) - - del self.g['x'] - # 'size' is supposed to be the same even after deleting an entry. - self.assertEqual(os.path.getsize(filename), size1) - - self.g.reorganize() - size2 = os.path.getsize(filename) - self.assertLess(size2, size1) - self.assertGreaterEqual(size2, size0) - - def test_context_manager(self): - with gdbm.open(filename, 'c') as db: - db["gdbm context manager"] = "context manager" - - with gdbm.open(filename, 'r') as db: - self.assertEqual(list(db.keys()), [b"gdbm context manager"]) - - with self.assertRaises(gdbm.error) as cm: - db.keys() - self.assertEqual(str(cm.exception), - "GDBM object has already been closed") - - def test_bytes(self): - with gdbm.open(filename, 'c') as db: - db[b'bytes key \xbd'] = b'bytes value \xbd' - with gdbm.open(filename, 'r') as db: - self.assertEqual(list(db.keys()), [b'bytes key \xbd']) - self.assertTrue(b'bytes key \xbd' in db) - self.assertEqual(db[b'bytes key \xbd'], b'bytes value \xbd') - - def test_unicode(self): - with gdbm.open(filename, 'c') as db: - db['Unicode key \U0001f40d'] = 'Unicode value \U0001f40d' - with gdbm.open(filename, 'r') as db: - self.assertEqual(list(db.keys()), ['Unicode key \U0001f40d'.encode()]) - self.assertTrue('Unicode key \U0001f40d'.encode() in db) - self.assertTrue('Unicode key \U0001f40d' in db) - self.assertEqual(db['Unicode key \U0001f40d'.encode()], - 'Unicode value \U0001f40d'.encode()) - self.assertEqual(db['Unicode key \U0001f40d'], - 'Unicode value \U0001f40d'.encode()) - - def test_write_readonly_file(self): - with gdbm.open(filename, 'c') as db: - db[b'bytes key'] = b'bytes value' - with gdbm.open(filename, 'r') as db: - with self.assertRaises(gdbm.error): - del db[b'not exist key'] - with self.assertRaises(gdbm.error): - del db[b'bytes key'] - with self.assertRaises(gdbm.error): - db[b'not exist key'] = b'not exist value' - - @unittest.skipUnless(TESTFN_NONASCII, - 'requires OS support of non-ASCII encodings') - def test_nonascii_filename(self): - filename = TESTFN_NONASCII - self.addCleanup(unlink, filename) - with gdbm.open(filename, 'c') as db: - db[b'key'] = b'value' - self.assertTrue(os.path.exists(filename)) - with gdbm.open(filename, 'r') as db: - self.assertEqual(list(db.keys()), [b'key']) - self.assertTrue(b'key' in db) - self.assertEqual(db[b'key'], b'value') - - def test_nonexisting_file(self): - nonexisting_file = 'nonexisting-file' - with self.assertRaises(gdbm.error) as cm: - gdbm.open(nonexisting_file) - self.assertIn(nonexisting_file, str(cm.exception)) - self.assertEqual(cm.exception.filename, nonexisting_file) - - -if __name__ == '__main__': - unittest.main() |