mirror of https://github.com/progval/irctest.git
42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
import sys
|
|
import unittest
|
|
import argparse
|
|
import unittest
|
|
import functools
|
|
import importlib
|
|
from .cases import _IrcTestCase
|
|
from .basecontrollers import BaseClientController, BaseServerController
|
|
|
|
def main(args):
|
|
try:
|
|
module = importlib.import_module(args.module)
|
|
except ImportError:
|
|
print('Cannot import module {}'.format(args.module), file=sys.stderr)
|
|
exit(1)
|
|
|
|
controller_class = module.get_irctest_controller_class()
|
|
if issubclass(controller_class, BaseClientController):
|
|
module = 'irctest.client_tests'
|
|
elif issubclass(controller_class, BaseServerController):
|
|
module = 'irctest.server_tests'
|
|
else:
|
|
print(r'{}.Controller should be a subclass of '
|
|
r'irctest.basecontroller.Base{{Client,Server}}Controller'
|
|
.format(args.module),
|
|
file=sys.stderr)
|
|
exit(1)
|
|
_IrcTestCase.controllerClass = controller_class
|
|
_IrcTestCase.show_io = args.show_io
|
|
unittest.main(module=module, argv=[sys.argv[0], 'discover'])
|
|
|
|
|
|
parser = argparse.ArgumentParser(
|
|
description='A script to test interoperability of IRC software.')
|
|
parser.add_argument('module', type=str,
|
|
help='The module used to run the tested program.')
|
|
parser.add_argument('--show-io', action='store_true',
|
|
help='Show input/outputs with the tested program.')
|
|
|
|
args = parser.parse_args()
|
|
main(args)
|