Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-06-23 15:30:45 -04:00
Code Issues Packages Projects Releases Wiki Activity

Allow choosing individual tests to run

Browse Source
This commit is contained in:
Kovid Goyal 2016-06-21 23:34:20 +05:30
parent f89a794396
commit 18792fb11a
1 changed files with 24 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
setup/test.py
View File

@ -60,10 +60,33 @@ def find_tests(which_tests=None):
tests = unittest.TestSuite(ans)
return tests
def filter_tests(suite, *names):
names = {x if x.startswith('test_') else 'test_' + x for x in names}
tests = []
for test in suite._tests:
if test.__class__.__name__ == 'ModuleImportFailure':
raise Exception('Failed to import a test module: %s' % test)
for s in test:
if s._testMethodName in names:
tests.append(s)
return unittest.TestSuite(tests)
class Test(Command):
def add_options(self, parser):
parser.add_option('--test-module', default=[], action='append', type='choice', choices=list(TEST_MODULES),
help='The test module to run (can be specified more than once for multiple modules). Choices: %s' % ', '.join(sorted(TEST_MODULES)))
parser.add_option('--test-verbosity', type=int, default=4, help='Test verbosity (0-4)')
parser.add_option('--test-name', default=[], action='append',
help='The name of an individual test to run. Can be specified more than once for multiple tests. The name of the'
' test is the name of the test function without the leading test_. For example, the function test_something()'
' can be run by specifying the name "something".')
def run(self, opts):
from calibre.gui2 import ensure_app, load_builtin_fonts
ensure_app(), load_builtin_fonts()
r = unittest.TextTestRunner
r(verbosity=2).run(find_tests())
tests = find_tests(which_tests=frozenset(opts.test_module))
if opts.test_name:
tests = filter_tests(tests, *opts.test_name)
r(verbosity=opts.test_verbosity).run(tests)