mirror of https://github.com/zulip/zulip.git
49 lines
1.3 KiB
Python
Executable File
49 lines
1.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import argparse
|
|
import os
|
|
import sys
|
|
import unittest
|
|
|
|
# check for the venv
|
|
from lib import sanity_check
|
|
|
|
sanity_check.check_venv(__file__)
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--coverage", action="store_true", help="compute test coverage")
|
|
args = parser.parse_args()
|
|
|
|
def dir_join(dir1: str, dir2: 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()
|
|
|
|
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)
|
|
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")
|