Lib/test/test_anydbm.py

   1 #! /usr/bin/env python
   2 """Test script for the anydbm module
   3    based on testdumbdbm.py
   4 """
   5
   6 import os
   7 import unittest
   8 import anydbm
   9 import glob
  10 from test import test_support
  11
  12 _fname = test_support.TESTFN
  13
  14 def _delete_files():
  15     # we don't know the precise name the underlying database uses
  16     # so we use glob to locate all names
  17     for f in glob.glob(_fname + "*"):
  18         try:
  19             os.unlink(f)
  20         except OSError:
  21             pass
  22
  23 class AnyDBMTestCase(unittest.TestCase):
  24     _dict = {'0': '',
  25              'a': 'Python:',
  26              'b': 'Programming',
  27              'c': 'the',
  28              'd': 'way',
  29              'f': 'Guido',
  30              'g': 'intended'
  31              }
  32
  33     def __init__(self, *args):
  34         unittest.TestCase.__init__(self, *args)
  35
  36     def test_anydbm_creation(self):
  37         f = anydbm.open(_fname, 'c')
  38         self.assertEqual(f.keys(), [])
  39         for key in self._dict:
  40             f[key] = self._dict[key]
  41         self.read_helper(f)
  42         f.close()
  43
  44     def test_anydbm_modification(self):
  45         self.init_db()
  46         f = anydbm.open(_fname, 'c')
  47         self._dict['g'] = f['g'] = "indented"
  48         self.read_helper(f)
  49         f.close()
  50
  51     def test_anydbm_read(self):
  52         self.init_db()
  53         f = anydbm.open(_fname, 'r')
  54         self.read_helper(f)
  55         f.close()
  56
  57     def test_anydbm_keys(self):
  58         self.init_db()
  59         f = anydbm.open(_fname, 'r')
  60         keys = self.keys_helper(f)
  61         f.close()
  62
  63     def read_helper(self, f):
  64         keys = self.keys_helper(f)
  65         for key in self._dict:
  66             self.assertEqual(self._dict[key], f[key])
  67
  68     def init_db(self):
  69         f = anydbm.open(_fname, 'n')
  70         for k in self._dict:
  71             f[k] = self._dict[k]
  72         f.close()
  73
  74     def keys_helper(self, f):
  75         keys = f.keys()
  76         keys.sort()
  77         dkeys = self._dict.keys()
  78         dkeys.sort()
  79         self.assertEqual(keys, dkeys)
  80         return keys
  81
  82     def tearDown(self):
  83         _delete_files()
  84
  85     def setUp(self):
  86         _delete_files()
  87
  88 def test_main():
  89     try:
  90         test_support.run_unittest(AnyDBMTestCase)
  91     finally:
  92         _delete_files()
  93
  94 if __name__ == "__main__":
  95     test_main()