#!/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')