Lib/test/test_dumbdbm.py

   1 #! /usr/bin/env python
   2 """Test script for the dumbdbm module
   3    Original by Roger E. Masse
   4 """
   5
   6 import os
   7 import unittest
   8 import dumbdbm
   9 from test import test_support
  10
  11 _fname = test_support.TESTFN
  12
  13 def _delete_files():
  14     for ext in [".dir", ".dat", ".bak"]:
  15         try:
  16             os.unlink(_fname + ext)
  17         except OSError:
  18             pass
  19
  20 class DumbDBMTestCase(unittest.TestCase):
  21     _dict = {'0': '',
  22              'a': 'Python:',
  23              'b': 'Programming',
  24              'c': 'the',
  25              'd': 'way',
  26              'f': 'Guido',
  27              'g': 'intended'
  28              }
  29
  30     def __init__(self, *args):
  31         unittest.TestCase.__init__(self, *args)
  32
  33     def test_dumbdbm_creation(self):
  34         f = dumbdbm.open(_fname, 'c')
  35         self.assertEqual(f.keys(), [])
  36         for key in self._dict:
  37             f[key] = self._dict[key]
  38         self.read_helper(f)
  39         f.close()
  40
  41     def test_dumbdbm_modification(self):
  42         self.init_db()
  43         f = dumbdbm.open(_fname, 'w')
  44         self._dict['g'] = f['g'] = "indented"
  45         self.read_helper(f)
  46         f.close()
  47
  48     def test_dumbdbm_read(self):
  49         self.init_db()
  50         f = dumbdbm.open(_fname, 'r')
  51         self.read_helper(f)
  52         f.close()
  53
  54     def test_dumbdbm_keys(self):
  55         self.init_db()
  56         f = dumbdbm.open(_fname)
  57         keys = self.keys_helper(f)
  58         f.close()
  59
  60     def test_write_write_read(self):
  61         # test for bug #482460
  62         f = dumbdbm.open(_fname)
  63         f['1'] = 'hello'
  64         f['1'] = 'hello2'
  65         f.close()
  66         f = dumbdbm.open(_fname)
  67         self.assertEqual(f['1'], 'hello2')
  68         f.close()
  69
  70     def read_helper(self, f):
  71         keys = self.keys_helper(f)
  72         for key in self._dict:
  73             self.assertEqual(self._dict[key], f[key])
  74
  75     def init_db(self):
  76         f = dumbdbm.open(_fname, 'w')
  77         for k in self._dict:
  78             f[k] = self._dict[k]
  79         f.close()
  80
  81     def keys_helper(self, f):
  82         keys = f.keys()
  83         keys.sort()
  84         dkeys = self._dict.keys()
  85         dkeys.sort()
  86         self.assertEqual(keys, dkeys)
  87         return keys
  88
  89     def tearDown(self):
  90         _delete_files()
  91
  92     def setUp(self):
  93         _delete_files()
  94
  95 def test_main():
  96     try:
  97         test_support.run_unittest(DumbDBMTestCase)
  98     finally:
  99         _delete_files()
 100
 101 if __name__ == "__main__":
 102     test_main()