Benjamin Peterson | 90f5ba5 | 2010-03-11 22:53:45 +0000 | [diff] [blame] | 1 | #! /usr/bin/env python3 |
Guido van Rossum | 2f1064c | 2001-01-18 16:46:52 +0000 | [diff] [blame] | 2 | """Test script for the dumbdbm module |
| 3 | Original by Roger E. Masse |
| 4 | """ |
| 5 | |
Guido van Rossum | 6252e10 | 2007-05-23 20:51:02 +0000 | [diff] [blame] | 6 | import io |
Guido van Rossum | 2f1064c | 2001-01-18 16:46:52 +0000 | [diff] [blame] | 7 | import os |
Barry Warsaw | 3ca656f | 2001-11-13 20:16:52 +0000 | [diff] [blame] | 8 | import unittest |
Georg Brandl | 0a7ac7d | 2008-05-26 10:29:35 +0000 | [diff] [blame] | 9 | import dbm.dumb as dumbdbm |
Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 10 | from test import support |
Guido van Rossum | 2f1064c | 2001-01-18 16:46:52 +0000 | [diff] [blame] | 11 | |
Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 12 | _fname = support.TESTFN |
Tim Peters | d9fbf35 | 2002-01-30 07:32:53 +0000 | [diff] [blame] | 13 | |
| 14 | def _delete_files(): |
| 15 | for ext in [".dir", ".dat", ".bak"]: |
| 16 | try: |
| 17 | os.unlink(_fname + ext) |
| 18 | except OSError: |
| 19 | pass |
| 20 | |
Barry Warsaw | 3ca656f | 2001-11-13 20:16:52 +0000 | [diff] [blame] | 21 | class DumbDBMTestCase(unittest.TestCase): |
Brett Cannon | 58425d3 | 2008-11-21 00:17:53 +0000 | [diff] [blame] | 22 | _dict = {b'0': b'', |
| 23 | b'a': b'Python:', |
| 24 | b'b': b'Programming', |
| 25 | b'c': b'the', |
| 26 | b'd': b'way', |
| 27 | b'f': b'Guido', |
| 28 | b'g': b'intended', |
| 29 | '\u00fc'.encode('utf-8') : b'!', |
Barry Warsaw | 3ca656f | 2001-11-13 20:16:52 +0000 | [diff] [blame] | 30 | } |
| 31 | |
| 32 | def __init__(self, *args): |
| 33 | unittest.TestCase.__init__(self, *args) |
Tim Peters | 8876848 | 2001-11-13 21:51:26 +0000 | [diff] [blame] | 34 | |
Barry Warsaw | 3ca656f | 2001-11-13 20:16:52 +0000 | [diff] [blame] | 35 | def test_dumbdbm_creation(self): |
Tim Peters | d9fbf35 | 2002-01-30 07:32:53 +0000 | [diff] [blame] | 36 | f = dumbdbm.open(_fname, 'c') |
Brett Cannon | c1b22a0 | 2007-02-22 05:04:32 +0000 | [diff] [blame] | 37 | self.assertEqual(list(f.keys()), []) |
Barry Warsaw | 3ca656f | 2001-11-13 20:16:52 +0000 | [diff] [blame] | 38 | for key in self._dict: |
Brett Cannon | 58425d3 | 2008-11-21 00:17:53 +0000 | [diff] [blame] | 39 | f[key] = self._dict[key] |
Barry Warsaw | 3ca656f | 2001-11-13 20:16:52 +0000 | [diff] [blame] | 40 | self.read_helper(f) |
| 41 | f.close() |
| 42 | |
Thomas Wouters | 902d6eb | 2007-01-09 23:18:33 +0000 | [diff] [blame] | 43 | def test_dumbdbm_creation_mode(self): |
| 44 | # On platforms without chmod, don't do anything. |
| 45 | if not (hasattr(os, 'chmod') and hasattr(os, 'umask')): |
| 46 | return |
| 47 | |
| 48 | try: |
Guido van Rossum | cd16bf6 | 2007-06-13 18:07:49 +0000 | [diff] [blame] | 49 | old_umask = os.umask(0o002) |
| 50 | f = dumbdbm.open(_fname, 'c', 0o637) |
Thomas Wouters | 902d6eb | 2007-01-09 23:18:33 +0000 | [diff] [blame] | 51 | f.close() |
| 52 | finally: |
| 53 | os.umask(old_umask) |
Thomas Wouters | 9fe394c | 2007-02-05 01:24:16 +0000 | [diff] [blame] | 54 | |
Guido van Rossum | cd16bf6 | 2007-06-13 18:07:49 +0000 | [diff] [blame] | 55 | expected_mode = 0o635 |
Thomas Wouters | b213704 | 2007-02-01 18:02:27 +0000 | [diff] [blame] | 56 | if os.name != 'posix': |
| 57 | # Windows only supports setting the read-only attribute. |
| 58 | # This shouldn't fail, but doesn't work like Unix either. |
Guido van Rossum | cd16bf6 | 2007-06-13 18:07:49 +0000 | [diff] [blame] | 59 | expected_mode = 0o666 |
Thomas Wouters | b213704 | 2007-02-01 18:02:27 +0000 | [diff] [blame] | 60 | |
Thomas Wouters | 902d6eb | 2007-01-09 23:18:33 +0000 | [diff] [blame] | 61 | import stat |
| 62 | st = os.stat(_fname + '.dat') |
Thomas Wouters | b213704 | 2007-02-01 18:02:27 +0000 | [diff] [blame] | 63 | self.assertEqual(stat.S_IMODE(st.st_mode), expected_mode) |
Thomas Wouters | 902d6eb | 2007-01-09 23:18:33 +0000 | [diff] [blame] | 64 | st = os.stat(_fname + '.dir') |
Thomas Wouters | b213704 | 2007-02-01 18:02:27 +0000 | [diff] [blame] | 65 | self.assertEqual(stat.S_IMODE(st.st_mode), expected_mode) |
Thomas Wouters | 9fe394c | 2007-02-05 01:24:16 +0000 | [diff] [blame] | 66 | |
Tim Peters | 7a6c733 | 2003-07-13 17:21:10 +0000 | [diff] [blame] | 67 | def test_close_twice(self): |
| 68 | f = dumbdbm.open(_fname) |
Martin v. Löwis | e656869 | 2007-08-07 05:37:39 +0000 | [diff] [blame] | 69 | f[b'a'] = b'b' |
| 70 | self.assertEqual(f[b'a'], b'b') |
Tim Peters | 7a6c733 | 2003-07-13 17:21:10 +0000 | [diff] [blame] | 71 | f.close() |
| 72 | f.close() |
| 73 | |
Barry Warsaw | 3ca656f | 2001-11-13 20:16:52 +0000 | [diff] [blame] | 74 | def test_dumbdbm_modification(self): |
Skip Montanaro | c08fe82 | 2002-03-17 23:03:42 +0000 | [diff] [blame] | 75 | self.init_db() |
Tim Peters | d9fbf35 | 2002-01-30 07:32:53 +0000 | [diff] [blame] | 76 | f = dumbdbm.open(_fname, 'w') |
Brett Cannon | 58425d3 | 2008-11-21 00:17:53 +0000 | [diff] [blame] | 77 | self._dict[b'g'] = f[b'g'] = b"indented" |
Barry Warsaw | 3ca656f | 2001-11-13 20:16:52 +0000 | [diff] [blame] | 78 | self.read_helper(f) |
| 79 | f.close() |
| 80 | |
| 81 | def test_dumbdbm_read(self): |
Skip Montanaro | c08fe82 | 2002-03-17 23:03:42 +0000 | [diff] [blame] | 82 | self.init_db() |
Tim Peters | d9fbf35 | 2002-01-30 07:32:53 +0000 | [diff] [blame] | 83 | f = dumbdbm.open(_fname, 'r') |
Barry Warsaw | 3ca656f | 2001-11-13 20:16:52 +0000 | [diff] [blame] | 84 | self.read_helper(f) |
| 85 | f.close() |
| 86 | |
| 87 | def test_dumbdbm_keys(self): |
Skip Montanaro | c08fe82 | 2002-03-17 23:03:42 +0000 | [diff] [blame] | 88 | self.init_db() |
Tim Peters | d9fbf35 | 2002-01-30 07:32:53 +0000 | [diff] [blame] | 89 | f = dumbdbm.open(_fname) |
Barry Warsaw | 3ca656f | 2001-11-13 20:16:52 +0000 | [diff] [blame] | 90 | keys = self.keys_helper(f) |
| 91 | f.close() |
| 92 | |
Gregory P. Smith | 178fefb | 2007-08-24 21:59:45 +0000 | [diff] [blame] | 93 | def test_write_contains(self): |
| 94 | f = dumbdbm.open(_fname) |
| 95 | f[b'1'] = b'hello' |
Benjamin Peterson | 577473f | 2010-01-19 00:09:57 +0000 | [diff] [blame] | 96 | self.assertIn(b'1', f) |
Gregory P. Smith | 178fefb | 2007-08-24 21:59:45 +0000 | [diff] [blame] | 97 | f.close() |
| 98 | |
Skip Montanaro | c08fe82 | 2002-03-17 23:03:42 +0000 | [diff] [blame] | 99 | def test_write_write_read(self): |
| 100 | # test for bug #482460 |
| 101 | f = dumbdbm.open(_fname) |
Martin v. Löwis | e656869 | 2007-08-07 05:37:39 +0000 | [diff] [blame] | 102 | f[b'1'] = b'hello' |
| 103 | f[b'1'] = b'hello2' |
Skip Montanaro | c08fe82 | 2002-03-17 23:03:42 +0000 | [diff] [blame] | 104 | f.close() |
| 105 | f = dumbdbm.open(_fname) |
Guido van Rossum | 7b4beea | 2007-08-28 00:09:54 +0000 | [diff] [blame] | 106 | self.assertEqual(f[b'1'], b'hello2') |
Skip Montanaro | c08fe82 | 2002-03-17 23:03:42 +0000 | [diff] [blame] | 107 | f.close() |
| 108 | |
Brett Cannon | 58425d3 | 2008-11-21 00:17:53 +0000 | [diff] [blame] | 109 | def test_str_read(self): |
| 110 | self.init_db() |
| 111 | f = dumbdbm.open(_fname, 'r') |
| 112 | self.assertEqual(f['\u00fc'], self._dict['\u00fc'.encode('utf-8')]) |
| 113 | |
| 114 | def test_str_write_contains(self): |
| 115 | self.init_db() |
| 116 | f = dumbdbm.open(_fname) |
| 117 | f['\u00fc'] = b'!' |
Brett Cannon | 7317c1e | 2008-11-25 19:19:17 +0000 | [diff] [blame] | 118 | f['1'] = 'a' |
Brett Cannon | 58425d3 | 2008-11-21 00:17:53 +0000 | [diff] [blame] | 119 | f.close() |
| 120 | f = dumbdbm.open(_fname, 'r') |
Benjamin Peterson | 577473f | 2010-01-19 00:09:57 +0000 | [diff] [blame] | 121 | self.assertIn('\u00fc', f) |
Brett Cannon | 58425d3 | 2008-11-21 00:17:53 +0000 | [diff] [blame] | 122 | self.assertEqual(f['\u00fc'.encode('utf-8')], |
| 123 | self._dict['\u00fc'.encode('utf-8')]) |
Brett Cannon | 7317c1e | 2008-11-25 19:19:17 +0000 | [diff] [blame] | 124 | self.assertEqual(f[b'1'], b'a') |
Brett Cannon | 58425d3 | 2008-11-21 00:17:53 +0000 | [diff] [blame] | 125 | |
Andrew M. Kuchling | ecdad85 | 2005-06-07 19:36:10 +0000 | [diff] [blame] | 126 | def test_line_endings(self): |
| 127 | # test for bug #1172763: dumbdbm would die if the line endings |
| 128 | # weren't what was expected. |
| 129 | f = dumbdbm.open(_fname) |
Martin v. Löwis | e656869 | 2007-08-07 05:37:39 +0000 | [diff] [blame] | 130 | f[b'1'] = b'hello' |
| 131 | f[b'2'] = b'hello2' |
Andrew M. Kuchling | ecdad85 | 2005-06-07 19:36:10 +0000 | [diff] [blame] | 132 | f.close() |
| 133 | |
Christian Heimes | 9289bfc | 2007-10-31 19:40:17 +0000 | [diff] [blame] | 134 | # Mangle the file by changing the line separator to Windows or Unix |
Brett Cannon | c0eee31 | 2010-10-29 22:49:14 +0000 | [diff] [blame] | 135 | with io.open(_fname + '.dir', 'rb') as file: |
| 136 | data = file.read() |
Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 137 | if os.linesep == '\n': |
Christian Heimes | 9289bfc | 2007-10-31 19:40:17 +0000 | [diff] [blame] | 138 | data = data.replace(b'\n', b'\r\n') |
| 139 | else: |
| 140 | data = data.replace(b'\r\n', b'\n') |
Brett Cannon | c0eee31 | 2010-10-29 22:49:14 +0000 | [diff] [blame] | 141 | with io.open(_fname + '.dir', 'wb') as file: |
| 142 | file.write(data) |
Anthony Baxter | 95a2a4e | 2005-06-08 04:35:50 +0000 | [diff] [blame] | 143 | |
Andrew M. Kuchling | ecdad85 | 2005-06-07 19:36:10 +0000 | [diff] [blame] | 144 | f = dumbdbm.open(_fname) |
Martin v. Löwis | e656869 | 2007-08-07 05:37:39 +0000 | [diff] [blame] | 145 | self.assertEqual(f[b'1'], b'hello') |
| 146 | self.assertEqual(f[b'2'], b'hello2') |
Anthony Baxter | 95a2a4e | 2005-06-08 04:35:50 +0000 | [diff] [blame] | 147 | |
| 148 | |
Barry Warsaw | 3ca656f | 2001-11-13 20:16:52 +0000 | [diff] [blame] | 149 | def read_helper(self, f): |
| 150 | keys = self.keys_helper(f) |
| 151 | for key in self._dict: |
Brett Cannon | 58425d3 | 2008-11-21 00:17:53 +0000 | [diff] [blame] | 152 | self.assertEqual(self._dict[key], f[key]) |
Tim Peters | 8876848 | 2001-11-13 21:51:26 +0000 | [diff] [blame] | 153 | |
Skip Montanaro | c08fe82 | 2002-03-17 23:03:42 +0000 | [diff] [blame] | 154 | def init_db(self): |
| 155 | f = dumbdbm.open(_fname, 'w') |
| 156 | for k in self._dict: |
Brett Cannon | 58425d3 | 2008-11-21 00:17:53 +0000 | [diff] [blame] | 157 | f[k] = self._dict[k] |
Skip Montanaro | c08fe82 | 2002-03-17 23:03:42 +0000 | [diff] [blame] | 158 | f.close() |
| 159 | |
Barry Warsaw | 3ca656f | 2001-11-13 20:16:52 +0000 | [diff] [blame] | 160 | def keys_helper(self, f): |
Brett Cannon | 58425d3 | 2008-11-21 00:17:53 +0000 | [diff] [blame] | 161 | keys = sorted(f.keys()) |
Brett Cannon | c1b22a0 | 2007-02-22 05:04:32 +0000 | [diff] [blame] | 162 | dkeys = sorted(self._dict.keys()) |
Skip Montanaro | c08fe82 | 2002-03-17 23:03:42 +0000 | [diff] [blame] | 163 | self.assertEqual(keys, dkeys) |
Barry Warsaw | 3ca656f | 2001-11-13 20:16:52 +0000 | [diff] [blame] | 164 | return keys |
| 165 | |
Tim Peters | 663d1b6 | 2003-07-11 04:09:09 +0000 | [diff] [blame] | 166 | # Perform randomized operations. This doesn't make assumptions about |
| 167 | # what *might* fail. |
| 168 | def test_random(self): |
| 169 | import random |
| 170 | d = {} # mirror the database |
| 171 | for dummy in range(5): |
| 172 | f = dumbdbm.open(_fname) |
| 173 | for dummy in range(100): |
| 174 | k = random.choice('abcdefghijklm') |
| 175 | if random.random() < 0.2: |
| 176 | if k in d: |
| 177 | del d[k] |
Brett Cannon | 58425d3 | 2008-11-21 00:17:53 +0000 | [diff] [blame] | 178 | del f[k] |
Tim Peters | 663d1b6 | 2003-07-11 04:09:09 +0000 | [diff] [blame] | 179 | else: |
Guido van Rossum | 6252e10 | 2007-05-23 20:51:02 +0000 | [diff] [blame] | 180 | v = random.choice((b'a', b'b', b'c')) * random.randrange(10000) |
Tim Peters | 663d1b6 | 2003-07-11 04:09:09 +0000 | [diff] [blame] | 181 | d[k] = v |
Brett Cannon | 58425d3 | 2008-11-21 00:17:53 +0000 | [diff] [blame] | 182 | f[k] = v |
| 183 | self.assertEqual(f[k], v) |
Tim Peters | 663d1b6 | 2003-07-11 04:09:09 +0000 | [diff] [blame] | 184 | f.close() |
| 185 | |
| 186 | f = dumbdbm.open(_fname) |
Guido van Rossum | 7b4beea | 2007-08-28 00:09:54 +0000 | [diff] [blame] | 187 | expected = sorted((k.encode("latin-1"), v) for k, v in d.items()) |
Brett Cannon | c1b22a0 | 2007-02-22 05:04:32 +0000 | [diff] [blame] | 188 | got = sorted(f.items()) |
Tim Peters | 663d1b6 | 2003-07-11 04:09:09 +0000 | [diff] [blame] | 189 | self.assertEqual(expected, got) |
| 190 | f.close() |
| 191 | |
Skip Montanaro | c08fe82 | 2002-03-17 23:03:42 +0000 | [diff] [blame] | 192 | def tearDown(self): |
| 193 | _delete_files() |
| 194 | |
| 195 | def setUp(self): |
| 196 | _delete_files() |
| 197 | |
Barry Warsaw | 3ca656f | 2001-11-13 20:16:52 +0000 | [diff] [blame] | 198 | def test_main(): |
Tim Peters | d9fbf35 | 2002-01-30 07:32:53 +0000 | [diff] [blame] | 199 | try: |
Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 200 | support.run_unittest(DumbDBMTestCase) |
Tim Peters | d9fbf35 | 2002-01-30 07:32:53 +0000 | [diff] [blame] | 201 | finally: |
| 202 | _delete_files() |
Barry Warsaw | 3ca656f | 2001-11-13 20:16:52 +0000 | [diff] [blame] | 203 | |
| 204 | if __name__ == "__main__": |
| 205 | test_main() |