1# Test driver for bsddb package.
2"""
3Run all test cases.
4"""
5import os
6import sys
7import tempfile
8import time
9import unittest
10from test.test_support import requires, run_unittest, import_module
11
12# Skip test if _bsddb module was not built.
13import_module('_bsddb')
14# Silence Py3k warning
15import_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.
19if __name__ != '__main__':
20    requires('bsddb')
21
22verbose = False
23if 'verbose' in sys.argv:
24    verbose = True
25    sys.argv.remove('verbose')
26
27if 'silent' in sys.argv:  # take care of old flag, just in case
28    verbose = False
29    sys.argv.remove('silent')
30
31
32class 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
57def 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
80if __name__ == '__main__':
81    test_main()
82