test-lib: Drastically simplify by using test_handler function.
This automatically enables use of the pytest runner using common code.
This commit is contained in:
parent
52ab8cc44e
commit
2fd9162030
|
@ -1,51 +1,6 @@
|
|||
#!/usr/bin/env python3
|
||||
#!/usr/bin/env python
|
||||
|
||||
import os
|
||||
import sys
|
||||
import unittest
|
||||
import argparse
|
||||
|
||||
from importlib import import_module
|
||||
|
||||
def parse_args():
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument('--coverage',
|
||||
nargs='?',
|
||||
const=True,
|
||||
default=False,
|
||||
help='compute test coverage ("--coverage combine" to combine with previous reports)')
|
||||
return parser.parse_args()
|
||||
|
||||
def run_all():
|
||||
TOOLS_DIR = os.path.dirname(os.path.abspath(__file__))
|
||||
ROOT_DIR = os.path.abspath(os.path.join(TOOLS_DIR, '..'))
|
||||
BOTS_DIR = os.path.join(ROOT_DIR, 'zulib_bots')
|
||||
sys.path.insert(0, BOTS_DIR)
|
||||
|
||||
options = parse_args()
|
||||
|
||||
if options.coverage:
|
||||
import coverage
|
||||
cov = coverage.Coverage(config_file="tools/.coveragerc")
|
||||
if options.coverage == 'combine':
|
||||
cov.load()
|
||||
cov.start()
|
||||
|
||||
module = 'zulip_bots/zulip_bots/tests/'
|
||||
suite = unittest.defaultTestLoader.discover(module)
|
||||
|
||||
suite = unittest.TestSuite([suite])
|
||||
runner = unittest.TextTestRunner(verbosity=2)
|
||||
result = runner.run(suite)
|
||||
if result.failures or result.errors:
|
||||
sys.exit(1)
|
||||
|
||||
if not result.failures and options.coverage:
|
||||
cov.stop()
|
||||
cov.data_suffix = False # Disable suffix so that filename is .coverage
|
||||
cov.save()
|
||||
cov.html_report()
|
||||
print("HTML report saved under directory 'htmlcov'.")
|
||||
from server_lib.test_handler import handle_input_and_run_tests_for_package
|
||||
|
||||
if __name__ == '__main__':
|
||||
run_all()
|
||||
handle_input_and_run_tests_for_package('Bot library', ['zulip_bots', 'zulip_bots', 'tests'])
|
||||
|
|
Loading…
Reference in a new issue