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
|
from server_lib.test_handler import handle_input_and_run_tests_for_package
|
||||||
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'.")
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
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