Lib/test/test_whichdb.py

   1 #! /usr/bin/env python
   2 """Test script for the whichdb module
   3    based on test_anydbm.py
   4 """
   5
   6 import os
   7 import test.test_support
   8 import unittest
   9 import whichdb
  10 import anydbm
  11 import tempfile
  12 import glob
  13
  14 _fname = test.test_support.TESTFN
  15
  16 def _delete_files():
  17     # we don't know the precise name the underlying database uses
  18     # so we use glob to locate all names
  19     for f in glob.glob(_fname + "*"):
  20         try:
  21             os.unlink(f)
  22         except OSError:
  23             pass
  24
  25 class WhichDBTestCase(unittest.TestCase):
  26     # Actual test methods are added to namespace
  27     # after class definition.
  28     def __init__(self, *args):
  29         unittest.TestCase.__init__(self, *args)
  30
  31     def tearDown(self):
  32         _delete_files()
  33
  34     def setUp(self):
  35         _delete_files()
  36
  37 for name in anydbm._names:
  38     # we define a new test method for each
  39     # candidate database module.
  40     try:
  41         mod = __import__(name)
  42     except ImportError:
  43         continue
  44
  45     def test_whichdb_name(self, name=name, mod=mod):
  46         # Check whether whichdb correctly guesses module name
  47         # for databases opened with module mod.
  48         # Try with empty files first
  49         f = mod.open(_fname, 'c')
  50         f.close()
  51         self.assertEqual(name, whichdb.whichdb(_fname))
  52         # Now add a key
  53         f = mod.open(_fname, 'w')
  54         f["1"] = "1"
  55         f.close()
  56         self.assertEqual(name, whichdb.whichdb(_fname))
  57     setattr(WhichDBTestCase,"test_whichdb_%s" % name, test_whichdb_name)
  58
  59 def test_main():
  60     try:
  61         test.test_support.run_unittest(WhichDBTestCase)
  62     finally:
  63         _delete_files()
  64
  65 if __name__ == "__main__":
  66     test_main()