#!/usr/bin/env python

from importlib import import_module

import os
import sys
import argparse
import unittest

def main():
    TOOLS_DIR = os.path.dirname(os.path.abspath(__file__))
    os.chdir(os.path.dirname(TOOLS_DIR))

    parser = argparse.ArgumentParser(description="Run tests for the zulip_botserver package.")
    parser.add_argument('--coverage',
                        action='store_true',
                        help='compute test coverage')
    options = parser.parse_args()

    if options.coverage:
        import coverage
        cov = coverage.Coverage(config_file="tools/.coveragerc")
        cov.start()

    test_suites = unittest.defaultTestLoader.discover('zulip_botserver')
    suite = unittest.TestSuite(test_suites)
    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.save()
        cov.combine()
        cov.data_suffix = False  # Disable suffix so that filename is .coverage
        cov.save()
        cov.html_report()
        print("HTML report saved in directory 'htmlcov'.")


if __name__ == '__main__':
    main()