mirror of https://github.com/progval/irctest.git
50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
import sys
|
|
import unittest
|
|
import argparse
|
|
import unittest
|
|
import functools
|
|
import importlib
|
|
from .cases import _IrcTestCase
|
|
from .optional_extensions import OptionalityReportingTextTestRunner
|
|
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):
|
|
import irctest.client_tests as module
|
|
elif issubclass(controller_class, BaseServerController):
|
|
import irctest.server_tests as module
|
|
_IrcTestCase.server_start_delay = args.server_start_delay
|
|
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
|
|
ts = module.discover()
|
|
testRunner = OptionalityReportingTextTestRunner()
|
|
testLoader = unittest.loader.defaultTestLoader
|
|
testRunner.run(ts)
|
|
|
|
|
|
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.')
|
|
parser.add_argument('--server-start-delay', type=float, default=None,
|
|
help='Number of seconds to wait before querying a server.')
|
|
|
|
|
|
args = parser.parse_args()
|
|
main(args)
|