/Lib/test/test_dumbdbm.py

http://unladen-swallow.googlecode.com/ · Python · 179 lines · 138 code · 28 blank · 13 comment · 19 complexity · b1956ecc54b839bc457267cba9dbe215 MD5 · raw file

  1. #! /usr/bin/env python
  2. """Test script for the dumbdbm module
  3. Original by Roger E. Masse
  4. """
  5. import os
  6. import unittest
  7. import dumbdbm
  8. from test import test_support
  9. _fname = test_support.TESTFN
  10. def _delete_files():
  11. for ext in [".dir", ".dat", ".bak"]:
  12. try:
  13. os.unlink(_fname + ext)
  14. except OSError:
  15. pass
  16. class DumbDBMTestCase(unittest.TestCase):
  17. _dict = {'0': '',
  18. 'a': 'Python:',
  19. 'b': 'Programming',
  20. 'c': 'the',
  21. 'd': 'way',
  22. 'f': 'Guido',
  23. 'g': 'intended'
  24. }
  25. def __init__(self, *args):
  26. unittest.TestCase.__init__(self, *args)
  27. def test_dumbdbm_creation(self):
  28. f = dumbdbm.open(_fname, 'c')
  29. self.assertEqual(f.keys(), [])
  30. for key in self._dict:
  31. f[key] = self._dict[key]
  32. self.read_helper(f)
  33. f.close()
  34. def test_dumbdbm_creation_mode(self):
  35. # On platforms without chmod, don't do anything.
  36. if not (hasattr(os, 'chmod') and hasattr(os, 'umask')):
  37. return
  38. try:
  39. old_umask = os.umask(0002)
  40. f = dumbdbm.open(_fname, 'c', 0637)
  41. f.close()
  42. finally:
  43. os.umask(old_umask)
  44. expected_mode = 0635
  45. if os.name != 'posix':
  46. # Windows only supports setting the read-only attribute.
  47. # This shouldn't fail, but doesn't work like Unix either.
  48. expected_mode = 0666
  49. import stat
  50. st = os.stat(_fname + '.dat')
  51. self.assertEqual(stat.S_IMODE(st.st_mode), expected_mode)
  52. st = os.stat(_fname + '.dir')
  53. self.assertEqual(stat.S_IMODE(st.st_mode), expected_mode)
  54. def test_close_twice(self):
  55. f = dumbdbm.open(_fname)
  56. f['a'] = 'b'
  57. self.assertEqual(f['a'], 'b')
  58. f.close()
  59. f.close()
  60. def test_dumbdbm_modification(self):
  61. self.init_db()
  62. f = dumbdbm.open(_fname, 'w')
  63. self._dict['g'] = f['g'] = "indented"
  64. self.read_helper(f)
  65. f.close()
  66. def test_dumbdbm_read(self):
  67. self.init_db()
  68. f = dumbdbm.open(_fname, 'r')
  69. self.read_helper(f)
  70. f.close()
  71. def test_dumbdbm_keys(self):
  72. self.init_db()
  73. f = dumbdbm.open(_fname)
  74. keys = self.keys_helper(f)
  75. f.close()
  76. def test_write_write_read(self):
  77. # test for bug #482460
  78. f = dumbdbm.open(_fname)
  79. f['1'] = 'hello'
  80. f['1'] = 'hello2'
  81. f.close()
  82. f = dumbdbm.open(_fname)
  83. self.assertEqual(f['1'], 'hello2')
  84. f.close()
  85. def test_line_endings(self):
  86. # test for bug #1172763: dumbdbm would die if the line endings
  87. # weren't what was expected.
  88. f = dumbdbm.open(_fname)
  89. f['1'] = 'hello'
  90. f['2'] = 'hello2'
  91. f.close()
  92. # Mangle the file by adding \r before each newline
  93. data = open(_fname + '.dir').read()
  94. data = data.replace('\n', '\r\n')
  95. open(_fname + '.dir', 'wb').write(data)
  96. f = dumbdbm.open(_fname)
  97. self.assertEqual(f['1'], 'hello')
  98. self.assertEqual(f['2'], 'hello2')
  99. def read_helper(self, f):
  100. keys = self.keys_helper(f)
  101. for key in self._dict:
  102. self.assertEqual(self._dict[key], f[key])
  103. def init_db(self):
  104. f = dumbdbm.open(_fname, 'w')
  105. for k in self._dict:
  106. f[k] = self._dict[k]
  107. f.close()
  108. def keys_helper(self, f):
  109. keys = f.keys()
  110. keys.sort()
  111. dkeys = self._dict.keys()
  112. dkeys.sort()
  113. self.assertEqual(keys, dkeys)
  114. return keys
  115. # Perform randomized operations. This doesn't make assumptions about
  116. # what *might* fail.
  117. def test_random(self):
  118. import random
  119. d = {} # mirror the database
  120. for dummy in range(5):
  121. f = dumbdbm.open(_fname)
  122. for dummy in range(100):
  123. k = random.choice('abcdefghijklm')
  124. if random.random() < 0.2:
  125. if k in d:
  126. del d[k]
  127. del f[k]
  128. else:
  129. v = random.choice('abc') * random.randrange(10000)
  130. d[k] = v
  131. f[k] = v
  132. self.assertEqual(f[k], v)
  133. f.close()
  134. f = dumbdbm.open(_fname)
  135. expected = d.items()
  136. expected.sort()
  137. got = f.items()
  138. got.sort()
  139. self.assertEqual(expected, got)
  140. f.close()
  141. def tearDown(self):
  142. _delete_files()
  143. def setUp(self):
  144. _delete_files()
  145. def test_main():
  146. try:
  147. test_support.run_unittest(DumbDBMTestCase)
  148. finally:
  149. _delete_files()
  150. if __name__ == "__main__":
  151. test_main()