xenofem/python-zulip-api
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

test-bots: Add support for testing a single bot.

Browse source
This commit is contained in:
Theodore Chen 2017-05-25 20:52:01 +00:00 committed by Tim Abbott
parent 8cbda46a61
commit ab2dcb869d
1 changed files with 16 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
contrib_bots/test-bots
View file

@ -3,17 +3,27 @@
from __future__ import absolute_import from __future__ import absolute_import
from __future__ import print_function from __future__ import print_function
import argparse
import os import os
import sys import sys
import unittest import unittest
from unittest import TestCase from unittest import TestCase
if __name__ == '__main__':
def dir_join(dir1, dir2): def dir_join(dir1, dir2):
# type: (str, str) -> str # type: (str, str) -> str
return os.path.abspath(os.path.join(dir1, dir2)) 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',
nargs=1,
type=str,
action='store',
help='test specified single bot')
args = parser.parse_args()
bots_dir = os.path.dirname(os.path.abspath(__file__)) bots_dir = os.path.dirname(os.path.abspath(__file__))
root_dir = dir_join(bots_dir, '..') root_dir = dir_join(bots_dir, '..')
bots_test_dir = dir_join(bots_dir, 'bots') bots_test_dir = dir_join(bots_dir, 'bots')
@ -23,6 +33,8 @@ if __name__ == '__main__':
# mypy doesn't recognize the TestLoader attribute, even though the code # mypy doesn't recognize the TestLoader attribute, even though the code
# is executable # is executable
loader = unittest.TestLoader() # type: ignore loader = unittest.TestLoader() # type: ignore
if args.bot is not None:
bots_test_dir = dir_join(bots_test_dir, args.bot[0])
suite = loader.discover(start_dir=bots_test_dir, top_level_dir=root_dir) suite = loader.discover(start_dir=bots_test_dir, top_level_dir=root_dir)
runner = unittest.TextTestRunner(verbosity=2) runner = unittest.TextTestRunner(verbosity=2)
# same issue as for TestLoader # same issue as for TestLoader