test-bots: Add support for testing a single bot.
This commit is contained in:
parent
8cbda46a61
commit
ab2dcb869d
|
@ -3,17 +3,27 @@
|
|||
from __future__ import absolute_import
|
||||
from __future__ import print_function
|
||||
|
||||
import argparse
|
||||
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))
|
||||
|
||||
|
||||
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__))
|
||||
root_dir = dir_join(bots_dir, '..')
|
||||
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
|
||||
# is executable
|
||||
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)
|
||||
runner = unittest.TextTestRunner(verbosity=2)
|
||||
# same issue as for TestLoader
|
||||
|
|
Loading…
Reference in a new issue