mirror of https://github.com/zulip/zulip.git
48 lines
1.5 KiB
Python
Executable File
48 lines
1.5 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
from __future__ import absolute_import
|
|
from __future__ import print_function
|
|
|
|
import argparse
|
|
import os
|
|
import sys
|
|
import unittest
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--coverage', dest='coverage',
|
|
action="store_true",
|
|
default=False, help='compute test coverage')
|
|
args = parser.parse_args()
|
|
|
|
def dir_join(dir1, dir2):
|
|
# type: (str, str) -> str
|
|
return os.path.abspath(os.path.join(dir1, dir2))
|
|
|
|
tools_dir = os.path.dirname(os.path.abspath(__file__))
|
|
root_dir = dir_join(tools_dir, '..')
|
|
tools_test_dir = dir_join(tools_dir, 'tests')
|
|
|
|
sys.path.insert(0, root_dir)
|
|
|
|
loader = unittest.TestLoader() # type: ignore # https://github.com/python/typeshed/issues/372
|
|
|
|
if args.coverage:
|
|
import coverage
|
|
cov = coverage.Coverage(branch=True, omit=[ "*/zulip-venv-cache/*", dir_join(tools_test_dir, "*") ])
|
|
cov.start()
|
|
|
|
suite = loader.discover(start_dir=tools_test_dir, top_level_dir=root_dir)
|
|
runner = unittest.TextTestRunner(verbosity=2)
|
|
result = runner.run(suite) # type: ignore # https://github.com/python/typeshed/issues/372
|
|
if result.errors or result.failures:
|
|
raise Exception('Test failed!')
|
|
|
|
if args.coverage:
|
|
cov.stop()
|
|
cov.save()
|
|
cov.html_report(directory='var/tools_coverage')
|
|
print("HTML report saved to var/tools_coverage")
|
|
|
|
print('SUCCESS')
|