Add test runner script
Change-Id: I8230f34e4e5599a2e61ad6353efd0f18c3d8d24c
This commit is contained in:
parent
debe895033
commit
971d53aea4
|
@ -32,7 +32,7 @@ class RunConfigFixture(tobiko.SharedFixture):
|
||||||
package_file = os.path.realpath(os.path.realpath(tobiko.__file__))
|
package_file = os.path.realpath(os.path.realpath(tobiko.__file__))
|
||||||
package_dir = os.path.dirname(package_file)
|
package_dir = os.path.dirname(package_file)
|
||||||
tobiko_dir = os.path.dirname(package_dir)
|
tobiko_dir = os.path.dirname(package_dir)
|
||||||
self.test_path = [os.path.join(tobiko_dir, 'tobiko', 'tests')]
|
self.test_path = [os.path.join(tobiko_dir, 'tobiko', 'tests', 'unit')]
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def forked(self) -> bool:
|
def forked(self) -> bool:
|
||||||
|
|
|
@ -0,0 +1,108 @@
|
||||||
|
# Copyright (c) 2021 Red Hat, Inc.
|
||||||
|
#
|
||||||
|
# All Rights Reserved.
|
||||||
|
#
|
||||||
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
||||||
|
# not use this file except in compliance with the License. You may obtain
|
||||||
|
# a copy of the License at
|
||||||
|
#
|
||||||
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
#
|
||||||
|
# Unless required by applicable law or agreed to in writing, software
|
||||||
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||||||
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||||
|
# License for the specific language governing permissions and limitations
|
||||||
|
# under the License.
|
||||||
|
from __future__ import absolute_import
|
||||||
|
|
||||||
|
import collections
|
||||||
|
import sys
|
||||||
|
import typing
|
||||||
|
import unittest
|
||||||
|
|
||||||
|
from oslo_log import log
|
||||||
|
|
||||||
|
import tobiko
|
||||||
|
from tobiko.run import _config
|
||||||
|
from tobiko.run import _discover
|
||||||
|
from tobiko.run import _worker
|
||||||
|
|
||||||
|
|
||||||
|
LOG = log.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
def run_tests(test_path: typing.Iterable[str],
|
||||||
|
test_filename: str = None,
|
||||||
|
python_path: typing.Iterable[str] = None,
|
||||||
|
forked: bool = None,
|
||||||
|
config: _config.RunConfigFixture = None):
|
||||||
|
test_ids = _discover.find_test_ids(test_path=test_path,
|
||||||
|
test_filename=test_filename,
|
||||||
|
python_path=python_path,
|
||||||
|
forked=forked,
|
||||||
|
config=config)
|
||||||
|
if forked:
|
||||||
|
forked_run_test_ids(test_ids=test_ids)
|
||||||
|
else:
|
||||||
|
run_test_ids(test_ids=test_ids)
|
||||||
|
|
||||||
|
|
||||||
|
def run_test_ids(test_ids: typing.Iterable[str]) -> int:
|
||||||
|
test_classes: typing.Dict[str, typing.List[str]] = \
|
||||||
|
collections.defaultdict(list)
|
||||||
|
test_ids = list(test_ids)
|
||||||
|
LOG.info(f'Run {len(test_ids)} test(s)')
|
||||||
|
for test_id in test_ids:
|
||||||
|
test_class_id, test_name = test_id.rsplit('.', 1)
|
||||||
|
test_classes[test_class_id].append(test_name)
|
||||||
|
result = unittest.TestResult()
|
||||||
|
for test_class_id, test_names in sorted(test_classes.items()):
|
||||||
|
LOG.info(f'Enter test class {test_class_id}')
|
||||||
|
test_class = tobiko.load_object(test_class_id)
|
||||||
|
for test_name in test_names:
|
||||||
|
LOG.info(f'Enter test case {test_class_id}.{test_name}')
|
||||||
|
test_case = test_class(test_name)
|
||||||
|
test_case.run(result)
|
||||||
|
LOG.info(f'Exit test case {test_class_id}.{test_name}')
|
||||||
|
LOG.info(f'Exit test class {test_class_id}')
|
||||||
|
LOG.info(f'{result.testsRun} test(s) run')
|
||||||
|
return result.testsRun
|
||||||
|
|
||||||
|
|
||||||
|
def forked_run_test_ids(test_ids: typing.Iterable[str]) -> int:
|
||||||
|
test_classes: typing.Dict[str, typing.List[str]] = \
|
||||||
|
collections.defaultdict(list)
|
||||||
|
test_ids = list(test_ids)
|
||||||
|
LOG.info(f'Run {len(test_ids)} test(s)')
|
||||||
|
for test_id in test_ids:
|
||||||
|
test_class_id, _ = test_id.rsplit('.', 1)
|
||||||
|
test_classes[test_class_id].append(test_id)
|
||||||
|
results = [_worker.call_async(run_test_ids, test_ids=grouped_ids)
|
||||||
|
for _, grouped_ids in sorted(test_classes.items())]
|
||||||
|
count = 0
|
||||||
|
for result in results:
|
||||||
|
count += result.get()
|
||||||
|
LOG.info(f'{count} test(s) run')
|
||||||
|
return count
|
||||||
|
|
||||||
|
|
||||||
|
def main(test_path: typing.Iterable[str] = None,
|
||||||
|
test_filename: str = None,
|
||||||
|
forked: bool = True,
|
||||||
|
python_path: typing.Iterable[str] = None):
|
||||||
|
if test_path is None:
|
||||||
|
test_path = sys.argv[1:]
|
||||||
|
try:
|
||||||
|
run_tests(test_path=test_path,
|
||||||
|
test_filename=test_filename,
|
||||||
|
forked=forked,
|
||||||
|
python_path=python_path)
|
||||||
|
except Exception as ex:
|
||||||
|
sys.stderr.write(f'{ex}\n')
|
||||||
|
sys.exit(1)
|
||||||
|
else:
|
||||||
|
sys.exit(0)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
main()
|
Loading…
Reference in New Issue