zulip/contrib_bots/test-bots

#!/usr/bin/env python

from __future__ import absolute_import
from __future__ import print_function

import os
import sys
import unittest
from unittest import TestCase

if __name__ == '__main__':

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

    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)

    # mypy doesn't recognize the TestLoader attribute, even though the code
    # is executable
    loader = unittest.TestLoader() # type: ignore
    suite = loader.discover(start_dir=bots_test_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!')