zulip/api/bots_api/test-bots

#!/usr/bin/env python

from __future__ import absolute_import
from __future__ import print_function

import argparse
import os
import sys
import unittest
import logging
from unittest import TestCase

def dir_join(dir1, dir2):
    # type: (str, str) -> str
    return os.path.abspath(os.path.join(dir1, dir2))


if __name__ == '__main__':

    description = 'Script to run test_<bot>.py files in bots/<bot> directories'
    parser = argparse.ArgumentParser(description=description)
    parser.add_argument('bot_to_test', metavar='bot', nargs='*',
                        help='specific bots to test (default is all)')
    args = parser.parse_args()

    bots_dir = os.path.dirname(os.path.abspath(__file__))
    root_dir = dir_join(bots_dir, '..')
    bots_test_dir = dir_join(bots_dir, '../bots')

    sys.path.insert(0, root_dir)
    sys.path.insert(0, bots_test_dir)

    def run_tests_in(start_dir):
        # type: (str) -> None
        # mypy doesn't recognize the TestLoader attribute, even though the code
        # is executable
        loader = unittest.TestLoader()  # type: ignore
        suite = loader.discover(start_dir=start_dir, top_level_dir=root_dir)
        runner = unittest.TextTestRunner(verbosity=2)
        # same issue as for TestLoader
        result = runner.run(suite)  # type: ignore
        if result.errors or result.failures:
            raise Exception('Test failed!')

    if len(args.bot_to_test) > 0:
        btd = bots_test_dir
        available_bots = [b for b in os.listdir(btd) if os.path.isdir(dir_join(btd, b))]
        tests_incomplete = False
        for n in args.bot_to_test:
            if n not in available_bots:
                logging.warning("Bot with name '%s' is unavailable for testing." % (n))
                tests_incomplete = True
            else:
                run_tests_in(dir_join(bots_test_dir, n))
        sys.exit(tests_incomplete)
    else:
        run_tests_in(bots_test_dir)