1 """Test script for the whichdb module 2 based on test_anydbm.py 3 """ 4 5 import os 6 import test.test_support 7 import unittest 8 import whichdb 9 import glob 10 11 _fname = test.test_support.TESTFN 12 13 # Silence Py3k warning 14 anydbm = test.test_support.import_module('anydbm', deprecated=True) 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 # Silence Py3k warning 42 mod = test.test_support.import_module(name, deprecated=True) 43 except unittest.SkipTest: 44 continue 45 46 def test_whichdb_name(self, name=name, mod=mod): 47 # Check whether whichdb correctly guesses module name 48 # for databases opened with module mod. 49 # Try with empty files first 50 f = mod.open(_fname, 'c') 51 f.close() 52 self.assertEqual(name, whichdb.whichdb(_fname)) 53 # Now add a key 54 f = mod.open(_fname, 'w') 55 f["1"] = "1" 56 f.close() 57 self.assertEqual(name, whichdb.whichdb(_fname)) 58 setattr(WhichDBTestCase,"test_whichdb_%s" % name, test_whichdb_name) 59 60 def test_main(): 61 try: 62 test.test_support.run_unittest(WhichDBTestCase) 63 finally: 64 _delete_files() 65 66 if __name__ == "__main__": 67 test_main() 68
