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