| 1 | import unittest
|
|---|
| 2 | import os
|
|---|
| 3 | from test.test_support import TESTFN, run_unittest, unlink, import_module
|
|---|
| 4 | gdbm = import_module('gdbm')
|
|---|
| 5 |
|
|---|
| 6 |
|
|---|
| 7 | filename = TESTFN
|
|---|
| 8 |
|
|---|
| 9 | class TestGdbm(unittest.TestCase):
|
|---|
| 10 |
|
|---|
| 11 | def setUp(self):
|
|---|
| 12 | self.g = None
|
|---|
| 13 |
|
|---|
| 14 | def tearDown(self):
|
|---|
| 15 | if self.g is not None:
|
|---|
| 16 | self.g.close()
|
|---|
| 17 | unlink(filename)
|
|---|
| 18 |
|
|---|
| 19 | def test_key_methods(self):
|
|---|
| 20 | self.g = gdbm.open(filename, 'c')
|
|---|
| 21 | self.assertEqual(self.g.keys(), [])
|
|---|
| 22 | self.g['a'] = 'b'
|
|---|
| 23 | self.g['12345678910'] = '019237410982340912840198242'
|
|---|
| 24 | key_set = set(self.g.keys())
|
|---|
| 25 | self.assertEqual(key_set, frozenset(['a', '12345678910']))
|
|---|
| 26 | self.assertTrue(self.g.has_key('a'))
|
|---|
| 27 | key = self.g.firstkey()
|
|---|
| 28 | while key:
|
|---|
| 29 | self.assertIn(key, key_set)
|
|---|
| 30 | key_set.remove(key)
|
|---|
| 31 | key = self.g.nextkey(key)
|
|---|
| 32 | self.assertRaises(KeyError, lambda: self.g['xxx'])
|
|---|
| 33 |
|
|---|
| 34 | def test_error_conditions(self):
|
|---|
| 35 | # Try to open a non-existent database.
|
|---|
| 36 | unlink(filename)
|
|---|
| 37 | self.assertRaises(gdbm.error, gdbm.open, filename, 'r')
|
|---|
| 38 | # Try to access a closed database.
|
|---|
| 39 | self.g = gdbm.open(filename, 'c')
|
|---|
| 40 | self.g.close()
|
|---|
| 41 | self.assertRaises(gdbm.error, lambda: self.g['a'])
|
|---|
| 42 | # try pass an invalid open flag
|
|---|
| 43 | self.assertRaises(gdbm.error, lambda: gdbm.open(filename, 'rx').close())
|
|---|
| 44 |
|
|---|
| 45 | def test_flags(self):
|
|---|
| 46 | # Test the flag parameter open() by trying all supported flag modes.
|
|---|
| 47 | all = set(gdbm.open_flags)
|
|---|
| 48 | # Test standard flags (presumably "crwn").
|
|---|
| 49 | modes = all - set('fsu')
|
|---|
| 50 | for mode in sorted(modes):
|
|---|
| 51 | self.g = gdbm.open(filename, mode)
|
|---|
| 52 | self.g.close()
|
|---|
| 53 |
|
|---|
| 54 | # Test additional flags (presumably "fsu").
|
|---|
| 55 | flags = all - set('crwn')
|
|---|
| 56 | for mode in modes:
|
|---|
| 57 | for flag in flags:
|
|---|
| 58 | self.g = gdbm.open(filename, mode + flag)
|
|---|
| 59 | self.g.close()
|
|---|
| 60 |
|
|---|
| 61 | def test_reorganize(self):
|
|---|
| 62 | self.g = gdbm.open(filename, 'c')
|
|---|
| 63 | size0 = os.path.getsize(filename)
|
|---|
| 64 |
|
|---|
| 65 | self.g['x'] = 'x' * 10000
|
|---|
| 66 | size1 = os.path.getsize(filename)
|
|---|
| 67 | self.assertTrue(size0 < size1)
|
|---|
| 68 |
|
|---|
| 69 | del self.g['x']
|
|---|
| 70 | # 'size' is supposed to be the same even after deleting an entry.
|
|---|
| 71 | self.assertEqual(os.path.getsize(filename), size1)
|
|---|
| 72 |
|
|---|
| 73 | self.g.reorganize()
|
|---|
| 74 | size2 = os.path.getsize(filename)
|
|---|
| 75 | self.assertTrue(size1 > size2 >= size0)
|
|---|
| 76 |
|
|---|
| 77 |
|
|---|
| 78 | def test_main():
|
|---|
| 79 | run_unittest(TestGdbm)
|
|---|
| 80 |
|
|---|
| 81 | if __name__ == '__main__':
|
|---|
| 82 | test_main()
|
|---|
