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