Home | History | Annotate | Download | only in test 
      1 # Test driver for bsddb package.
      2 """
      3 Run all test cases.
      4 """
      5 import os
      6 import sys
      7 import tempfile
      8 import time
      9 import unittest
     10 from test.test_support import requires, run_unittest, import_module
     11 
     12 # Skip test if _bsddb module was not built.
     13 import_module('_bsddb')
     14 # Silence Py3k warning
     15 import_module('bsddb', deprecated=True)
     16 
     17 # When running as a script instead of within the regrtest framework, skip the
     18 # requires test, since it's obvious we want to run them.
     19 if __name__ != '__main__':
     20     requires('bsddb')
     21 
     22 verbose = False
     23 if 'verbose' in sys.argv:
     24     verbose = True
     25     sys.argv.remove('verbose')
     26 
     27 if 'silent' in sys.argv:  # take care of old flag, just in case
     28     verbose = False
     29     sys.argv.remove('silent')
     30 
     31 
     32 class TimingCheck(unittest.TestCase):
     33 
     34     """This class is not a real test.  Its purpose is to print a message
     35     periodically when the test runs slowly.  This will prevent the buildbots
     36     from timing out on slow machines."""
     37 
     38     # How much time in seconds before printing a 'Still working' message.
     39     # Since this is run at most once between each test module, use a smaller
     40     # interval than other tests.
     41     _PRINT_WORKING_MSG_INTERVAL = 4 * 60
     42 
     43     # next_time is used as a global variable that survives each instance.
     44     # This is necessary since a new instance will be created for each test.
     45     next_time = time.time() + _PRINT_WORKING_MSG_INTERVAL
     46 
     47     def testCheckElapsedTime(self):
     48         # Print still working message since these tests can be really slow.
     49         now = time.time()
     50         if self.next_time <= now:
     51             TimingCheck.next_time = now + self._PRINT_WORKING_MSG_INTERVAL
     52             sys.__stdout__.write('  test_bsddb3 still working, be patient...\n')
     53             sys.__stdout__.flush()
     54 
     55 
     56 # For invocation through regrtest
     57 def test_main():
     58     from bsddb import db
     59     from bsddb.test import test_all
     60     test_all.set_test_path_prefix(os.path.join(tempfile.gettempdir(),
     61                                  'z-test_bsddb3-%s' %
     62                                  os.getpid()))
     63     # Please leave this print in, having this show up in the buildbots
     64     # makes diagnosing problems a lot easier.
     65     print >>sys.stderr, db.DB_VERSION_STRING
     66     print >>sys.stderr, 'Test path prefix: ', test_all.get_test_path_prefix()
     67     try:
     68         run_unittest(test_all.suite(module_prefix='bsddb.test.',
     69                                     timing_check=TimingCheck))
     70     finally:
     71         # The only reason to remove db_home is in case if there is an old
     72         # one lying around.  This might be by a different user, so just
     73         # ignore errors.  We should always make a unique name now.
     74         try:
     75             test_all.remove_test_path_directory()
     76         except:
     77             pass
     78 
     79 
     80 if __name__ == '__main__':
     81     test_main()
     82