940a70fff4
Type: test Change-Id: I01500466f3d15c79e38028677ce7e5c75d427fdc Signed-off-by: Dave Wallace <dwallacelf@gmail.com>
1095 lines
38 KiB
Python
1095 lines
38 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import sys
|
|
import shutil
|
|
import os
|
|
import fnmatch
|
|
import unittest
|
|
import time
|
|
import threading
|
|
import traceback
|
|
import signal
|
|
import re
|
|
from multiprocessing import Process, Pipe, get_context
|
|
from multiprocessing.queues import Queue
|
|
from multiprocessing.managers import BaseManager
|
|
from config import config, num_cpus, available_cpus, max_vpp_cpus
|
|
from vpp_papi import VPPApiJSONFiles
|
|
from asfframework import (
|
|
VppTestRunner,
|
|
get_testcase_doc_name,
|
|
get_test_description,
|
|
get_failed_testcase_linkname,
|
|
get_testcase_dirname,
|
|
)
|
|
from framework import VppTestCase
|
|
from test_result_code import TestResultCode
|
|
from debug import spawn_gdb
|
|
from log import (
|
|
get_parallel_logger,
|
|
double_line_delim,
|
|
RED,
|
|
YELLOW,
|
|
GREEN,
|
|
colorize,
|
|
single_line_delim,
|
|
)
|
|
from discover_tests import discover_tests
|
|
import sanity_run_vpp
|
|
from subprocess import check_output, CalledProcessError
|
|
from util import check_core_path, get_core_path, is_core_present
|
|
|
|
# timeout which controls how long the child has to finish after seeing
|
|
# a core dump in test temporary directory. If this is exceeded, parent assumes
|
|
# that child process is stuck (e.g. waiting for event from vpp) and kill
|
|
# the child
|
|
core_timeout = 3
|
|
|
|
|
|
class StreamQueue(Queue):
|
|
def write(self, msg):
|
|
self.put(msg)
|
|
|
|
def flush(self):
|
|
sys.__stdout__.flush()
|
|
sys.__stderr__.flush()
|
|
|
|
def fileno(self):
|
|
return self._writer.fileno()
|
|
|
|
|
|
class StreamQueueManager(BaseManager):
|
|
pass
|
|
|
|
|
|
StreamQueueManager.register("StreamQueue", StreamQueue)
|
|
|
|
|
|
class TestResult(dict):
|
|
def __init__(self, testcase_suite, testcases_by_id=None):
|
|
super(TestResult, self).__init__()
|
|
for trc in list(TestResultCode):
|
|
self[trc] = []
|
|
self.crashed = False
|
|
self.testcase_suite = testcase_suite
|
|
self.testcases = [testcase for testcase in testcase_suite]
|
|
self.testcases_by_id = testcases_by_id
|
|
|
|
def was_successful(self):
|
|
return (
|
|
0
|
|
== len(self[TestResultCode.FAIL])
|
|
== len(self[TestResultCode.ERROR])
|
|
== len(self[TestResultCode.UNEXPECTED_PASS])
|
|
and len(self[TestResultCode.PASS])
|
|
+ len(self[TestResultCode.SKIP])
|
|
+ len(self[TestResultCode.SKIP_CPU_SHORTAGE])
|
|
+ len(self[TestResultCode.EXPECTED_FAIL])
|
|
== self.testcase_suite.countTestCases()
|
|
)
|
|
|
|
def no_tests_run(self):
|
|
return 0 == len(self[TestResultCode.TEST_RUN])
|
|
|
|
def process_result(self, test_id, result):
|
|
self[result].append(test_id)
|
|
|
|
def suite_from_failed(self):
|
|
rerun_ids = set([])
|
|
for testcase in self.testcase_suite:
|
|
tc_id = testcase.id()
|
|
if (
|
|
tc_id
|
|
not in self[TestResultCode.PASS]
|
|
+ self[TestResultCode.SKIP]
|
|
+ self[TestResultCode.SKIP_CPU_SHORTAGE]
|
|
+ self[TestResultCode.EXPECTED_FAIL]
|
|
):
|
|
rerun_ids.add(tc_id)
|
|
if rerun_ids:
|
|
return suite_from_failed(self.testcase_suite, rerun_ids)
|
|
|
|
def get_testcase_names(self, test_id):
|
|
# could be tearDownClass (test_ipsec_esp.TestIpsecEsp1)
|
|
setup_teardown_match = re.match(
|
|
r"((tearDownClass)|(setUpClass)) \((.+\..+)\)", test_id
|
|
)
|
|
if setup_teardown_match:
|
|
test_name, _, _, testcase_name = setup_teardown_match.groups()
|
|
if len(testcase_name.split(".")) == 2:
|
|
for key in self.testcases_by_id.keys():
|
|
if key.startswith(testcase_name):
|
|
testcase_name = key
|
|
break
|
|
testcase_name = self._get_testcase_doc_name(testcase_name)
|
|
else:
|
|
test_name = self._get_test_description(test_id)
|
|
testcase_name = self._get_testcase_doc_name(test_id)
|
|
|
|
return testcase_name, test_name
|
|
|
|
def _get_test_description(self, test_id):
|
|
if test_id in self.testcases_by_id:
|
|
desc = get_test_description(descriptions, self.testcases_by_id[test_id])
|
|
else:
|
|
desc = test_id
|
|
return desc
|
|
|
|
def _get_testcase_doc_name(self, test_id):
|
|
if test_id in self.testcases_by_id:
|
|
doc_name = get_testcase_doc_name(self.testcases_by_id[test_id])
|
|
else:
|
|
doc_name = test_id
|
|
return doc_name
|
|
|
|
|
|
def test_runner_wrapper(
|
|
suite, keep_alive_pipe, stdouterr_queue, finished_pipe, result_pipe, logger
|
|
):
|
|
sys.stdout = stdouterr_queue
|
|
sys.stderr = stdouterr_queue
|
|
VppTestCase.parallel_handler = logger.handlers[0]
|
|
result = VppTestRunner(
|
|
keep_alive_pipe=keep_alive_pipe,
|
|
descriptions=descriptions,
|
|
verbosity=config.verbose,
|
|
result_pipe=result_pipe,
|
|
failfast=config.failfast,
|
|
print_summary=False,
|
|
).run(suite)
|
|
finished_pipe.send(result.wasSuccessful())
|
|
finished_pipe.close()
|
|
keep_alive_pipe.close()
|
|
|
|
|
|
class TestCaseWrapper(object):
|
|
def __init__(self, testcase_suite, manager):
|
|
self.keep_alive_parent_end, self.keep_alive_child_end = Pipe(duplex=False)
|
|
self.finished_parent_end, self.finished_child_end = Pipe(duplex=False)
|
|
self.result_parent_end, self.result_child_end = Pipe(duplex=False)
|
|
self.testcase_suite = testcase_suite
|
|
self.stdouterr_queue = manager.StreamQueue(ctx=get_context())
|
|
self.logger = get_parallel_logger(self.stdouterr_queue)
|
|
self.child = Process(
|
|
target=test_runner_wrapper,
|
|
args=(
|
|
testcase_suite,
|
|
self.keep_alive_child_end,
|
|
self.stdouterr_queue,
|
|
self.finished_child_end,
|
|
self.result_child_end,
|
|
self.logger,
|
|
),
|
|
)
|
|
self.child.start()
|
|
self.last_test_temp_dir = None
|
|
self.last_test_vpp_binary = None
|
|
self._last_test = None
|
|
self.last_test_id = None
|
|
self.vpp_pid = None
|
|
self.last_heard = time.time()
|
|
self.core_detected_at = None
|
|
self.testcases_by_id = {}
|
|
self.testclasess_with_core = {}
|
|
for testcase in self.testcase_suite:
|
|
self.testcases_by_id[testcase.id()] = testcase
|
|
self.result = TestResult(testcase_suite, self.testcases_by_id)
|
|
|
|
@property
|
|
def last_test(self):
|
|
return self._last_test
|
|
|
|
@last_test.setter
|
|
def last_test(self, test_id):
|
|
self.last_test_id = test_id
|
|
if test_id in self.testcases_by_id:
|
|
testcase = self.testcases_by_id[test_id]
|
|
self._last_test = testcase.shortDescription()
|
|
if not self._last_test:
|
|
self._last_test = str(testcase)
|
|
else:
|
|
self._last_test = test_id
|
|
|
|
def add_testclass_with_core(self):
|
|
if self.last_test_id in self.testcases_by_id:
|
|
test = self.testcases_by_id[self.last_test_id]
|
|
class_name = unittest.util.strclass(test.__class__)
|
|
test_name = "'{}' ({})".format(
|
|
get_test_description(descriptions, test), self.last_test_id
|
|
)
|
|
else:
|
|
test_name = self.last_test_id
|
|
class_name = re.match(
|
|
r"((tearDownClass)|(setUpClass)) " r"\((.+\..+)\)", test_name
|
|
).groups()[3]
|
|
if class_name not in self.testclasess_with_core:
|
|
self.testclasess_with_core[class_name] = (
|
|
test_name,
|
|
self.last_test_vpp_binary,
|
|
self.last_test_temp_dir,
|
|
)
|
|
|
|
def close_pipes(self):
|
|
self.keep_alive_child_end.close()
|
|
self.finished_child_end.close()
|
|
self.result_child_end.close()
|
|
self.keep_alive_parent_end.close()
|
|
self.finished_parent_end.close()
|
|
self.result_parent_end.close()
|
|
|
|
def was_successful(self):
|
|
return self.result.was_successful()
|
|
|
|
@property
|
|
def cpus_used(self):
|
|
return self.testcase_suite.cpus_used
|
|
|
|
def get_assigned_cpus(self):
|
|
return self.testcase_suite.get_assigned_cpus()
|
|
|
|
|
|
def stdouterr_reader_wrapper(
|
|
unread_testcases, finished_unread_testcases, read_testcases
|
|
):
|
|
read_testcase = None
|
|
while read_testcases.is_set() or unread_testcases:
|
|
if finished_unread_testcases:
|
|
read_testcase = finished_unread_testcases.pop()
|
|
unread_testcases.remove(read_testcase)
|
|
elif unread_testcases:
|
|
read_testcase = unread_testcases.pop()
|
|
if read_testcase:
|
|
data = ""
|
|
while data is not None:
|
|
sys.stdout.write(data)
|
|
data = read_testcase.stdouterr_queue.get()
|
|
|
|
read_testcase.stdouterr_queue.close()
|
|
finished_unread_testcases.discard(read_testcase)
|
|
read_testcase = None
|
|
|
|
|
|
def handle_failed_suite(logger, last_test_temp_dir, vpp_pid, vpp_binary):
|
|
if last_test_temp_dir:
|
|
# Need to create link in case of a timeout or core dump without failure
|
|
lttd = os.path.basename(last_test_temp_dir)
|
|
link_path = os.path.join(config.failed_dir, f"{lttd}-FAILED")
|
|
if not os.path.exists(link_path):
|
|
os.symlink(last_test_temp_dir, link_path)
|
|
logger.error(
|
|
"Symlink to failed testcase directory: %s -> %s" % (link_path, lttd)
|
|
)
|
|
|
|
# Report core existence
|
|
core_path = get_core_path(last_test_temp_dir)
|
|
if os.path.exists(core_path):
|
|
logger.error(
|
|
"Core-file exists in test temporary directory: %s!" % core_path
|
|
)
|
|
check_core_path(logger, core_path)
|
|
logger.debug("Running 'file %s':" % core_path)
|
|
try:
|
|
info = check_output(["file", core_path])
|
|
logger.debug(info)
|
|
except CalledProcessError as e:
|
|
logger.error(
|
|
"Subprocess returned with return code "
|
|
"while running `file' utility on core-file "
|
|
"returned: "
|
|
"rc=%s",
|
|
e.returncode,
|
|
)
|
|
except OSError as e:
|
|
logger.error(
|
|
"Subprocess returned with OS error while "
|
|
"running 'file' utility "
|
|
"on core-file: "
|
|
"(%s) %s",
|
|
e.errno,
|
|
e.strerror,
|
|
)
|
|
except Exception as e:
|
|
logger.exception("Unexpected error running `file' utility on core-file")
|
|
logger.error(f"gdb {vpp_binary} {core_path}")
|
|
|
|
if vpp_pid:
|
|
# Copy api post mortem
|
|
api_post_mortem_path = "/tmp/api_post_mortem.%d" % vpp_pid
|
|
if os.path.isfile(api_post_mortem_path):
|
|
logger.error(
|
|
"Copying api_post_mortem.%d to %s" % (vpp_pid, last_test_temp_dir)
|
|
)
|
|
shutil.copy2(api_post_mortem_path, last_test_temp_dir)
|
|
|
|
|
|
def check_and_handle_core(vpp_binary, tempdir, core_crash_test):
|
|
if is_core_present(tempdir):
|
|
if debug_core:
|
|
print(
|
|
"VPP core detected in %s. Last test running was %s"
|
|
% (tempdir, core_crash_test)
|
|
)
|
|
print(single_line_delim)
|
|
spawn_gdb(vpp_binary, get_core_path(tempdir))
|
|
print(single_line_delim)
|
|
elif config.compress_core:
|
|
print("Compressing core-file in test directory `%s'" % tempdir)
|
|
os.system("gzip %s" % get_core_path(tempdir))
|
|
|
|
|
|
def handle_cores(failed_testcases):
|
|
for failed_testcase in failed_testcases:
|
|
tcs_with_core = failed_testcase.testclasess_with_core
|
|
if tcs_with_core:
|
|
for test, vpp_binary, tempdir in tcs_with_core.values():
|
|
check_and_handle_core(vpp_binary, tempdir, test)
|
|
|
|
|
|
def process_finished_testsuite(
|
|
wrapped_testcase_suite, finished_testcase_suites, failed_wrapped_testcases, results
|
|
):
|
|
results.append(wrapped_testcase_suite.result)
|
|
finished_testcase_suites.add(wrapped_testcase_suite)
|
|
stop_run = False
|
|
if config.failfast and not wrapped_testcase_suite.was_successful():
|
|
stop_run = True
|
|
|
|
if not wrapped_testcase_suite.was_successful():
|
|
failed_wrapped_testcases.add(wrapped_testcase_suite)
|
|
handle_failed_suite(
|
|
wrapped_testcase_suite.logger,
|
|
wrapped_testcase_suite.last_test_temp_dir,
|
|
wrapped_testcase_suite.vpp_pid,
|
|
wrapped_testcase_suite.last_test_vpp_binary,
|
|
)
|
|
|
|
return stop_run
|
|
|
|
|
|
def run_forked(testcase_suites):
|
|
wrapped_testcase_suites = set()
|
|
solo_testcase_suites = []
|
|
|
|
# suites are unhashable, need to use list
|
|
results = []
|
|
unread_testcases = set()
|
|
finished_unread_testcases = set()
|
|
manager = StreamQueueManager()
|
|
manager.start()
|
|
tests_running = 0
|
|
free_cpus = list(available_cpus)
|
|
|
|
def on_suite_start(tc):
|
|
nonlocal tests_running
|
|
nonlocal free_cpus
|
|
tests_running = tests_running + 1
|
|
|
|
def on_suite_finish(tc):
|
|
nonlocal tests_running
|
|
nonlocal free_cpus
|
|
tests_running = tests_running - 1
|
|
assert tests_running >= 0
|
|
free_cpus.extend(tc.get_assigned_cpus())
|
|
|
|
def run_suite(suite):
|
|
nonlocal manager
|
|
nonlocal wrapped_testcase_suites
|
|
nonlocal unread_testcases
|
|
nonlocal free_cpus
|
|
suite.assign_cpus(free_cpus[: suite.cpus_used])
|
|
free_cpus = free_cpus[suite.cpus_used :]
|
|
wrapper = TestCaseWrapper(suite, manager)
|
|
wrapped_testcase_suites.add(wrapper)
|
|
unread_testcases.add(wrapper)
|
|
on_suite_start(suite)
|
|
|
|
def can_run_suite(suite):
|
|
return tests_running < max_concurrent_tests and (
|
|
suite.cpus_used <= len(free_cpus) or suite.cpus_used > max_vpp_cpus
|
|
)
|
|
|
|
while free_cpus and testcase_suites:
|
|
a_suite = testcase_suites[0]
|
|
if a_suite.is_tagged_run_solo:
|
|
a_suite = testcase_suites.pop(0)
|
|
solo_testcase_suites.append(a_suite)
|
|
continue
|
|
if can_run_suite(a_suite):
|
|
a_suite = testcase_suites.pop(0)
|
|
run_suite(a_suite)
|
|
else:
|
|
break
|
|
|
|
if tests_running == 0 and solo_testcase_suites:
|
|
a_suite = solo_testcase_suites.pop(0)
|
|
run_suite(a_suite)
|
|
|
|
read_from_testcases = threading.Event()
|
|
read_from_testcases.set()
|
|
stdouterr_thread = threading.Thread(
|
|
target=stdouterr_reader_wrapper,
|
|
args=(unread_testcases, finished_unread_testcases, read_from_testcases),
|
|
)
|
|
stdouterr_thread.start()
|
|
|
|
failed_wrapped_testcases = set()
|
|
stop_run = False
|
|
|
|
try:
|
|
while wrapped_testcase_suites or testcase_suites:
|
|
finished_testcase_suites = set()
|
|
for wrapped_testcase_suite in wrapped_testcase_suites:
|
|
while wrapped_testcase_suite.result_parent_end.poll():
|
|
wrapped_testcase_suite.result.process_result(
|
|
*wrapped_testcase_suite.result_parent_end.recv()
|
|
)
|
|
wrapped_testcase_suite.last_heard = time.time()
|
|
|
|
while wrapped_testcase_suite.keep_alive_parent_end.poll():
|
|
(
|
|
wrapped_testcase_suite.last_test,
|
|
wrapped_testcase_suite.last_test_vpp_binary,
|
|
wrapped_testcase_suite.last_test_temp_dir,
|
|
wrapped_testcase_suite.vpp_pid,
|
|
) = wrapped_testcase_suite.keep_alive_parent_end.recv()
|
|
wrapped_testcase_suite.last_heard = time.time()
|
|
|
|
if wrapped_testcase_suite.finished_parent_end.poll():
|
|
wrapped_testcase_suite.finished_parent_end.recv()
|
|
wrapped_testcase_suite.last_heard = time.time()
|
|
stop_run = (
|
|
process_finished_testsuite(
|
|
wrapped_testcase_suite,
|
|
finished_testcase_suites,
|
|
failed_wrapped_testcases,
|
|
results,
|
|
)
|
|
or stop_run
|
|
)
|
|
continue
|
|
|
|
fail = False
|
|
if wrapped_testcase_suite.last_heard + config.timeout < time.time():
|
|
fail = True
|
|
wrapped_testcase_suite.logger.critical(
|
|
"Child test runner process timed out "
|
|
"(last test running was `%s' in `%s')!"
|
|
% (
|
|
wrapped_testcase_suite.last_test,
|
|
wrapped_testcase_suite.last_test_temp_dir,
|
|
)
|
|
)
|
|
elif not wrapped_testcase_suite.child.is_alive():
|
|
fail = True
|
|
wrapped_testcase_suite.logger.critical(
|
|
"Child test runner process unexpectedly died "
|
|
"(last test running was `%s' in `%s')!"
|
|
% (
|
|
wrapped_testcase_suite.last_test,
|
|
wrapped_testcase_suite.last_test_temp_dir,
|
|
)
|
|
)
|
|
elif (
|
|
wrapped_testcase_suite.last_test_temp_dir
|
|
and wrapped_testcase_suite.last_test_vpp_binary
|
|
):
|
|
if is_core_present(wrapped_testcase_suite.last_test_temp_dir):
|
|
wrapped_testcase_suite.add_testclass_with_core()
|
|
if wrapped_testcase_suite.core_detected_at is None:
|
|
wrapped_testcase_suite.core_detected_at = time.time()
|
|
elif (
|
|
wrapped_testcase_suite.core_detected_at + core_timeout
|
|
< time.time()
|
|
):
|
|
wrapped_testcase_suite.logger.critical(
|
|
"Child test runner process unresponsive and "
|
|
"core-file exists in test temporary directory "
|
|
"(last test running was `%s' in `%s')!"
|
|
% (
|
|
wrapped_testcase_suite.last_test,
|
|
wrapped_testcase_suite.last_test_temp_dir,
|
|
)
|
|
)
|
|
fail = True
|
|
|
|
if fail:
|
|
wrapped_testcase_suite.child.terminate()
|
|
try:
|
|
# terminating the child process tends to leave orphan
|
|
# VPP process around
|
|
if wrapped_testcase_suite.vpp_pid:
|
|
os.kill(wrapped_testcase_suite.vpp_pid, signal.SIGTERM)
|
|
except OSError:
|
|
# already dead
|
|
pass
|
|
wrapped_testcase_suite.result.crashed = True
|
|
wrapped_testcase_suite.result.process_result(
|
|
wrapped_testcase_suite.last_test_id, TestResultCode.ERROR
|
|
)
|
|
stop_run = (
|
|
process_finished_testsuite(
|
|
wrapped_testcase_suite,
|
|
finished_testcase_suites,
|
|
failed_wrapped_testcases,
|
|
results,
|
|
)
|
|
or stop_run
|
|
)
|
|
|
|
for finished_testcase in finished_testcase_suites:
|
|
# Somewhat surprisingly, the join below may
|
|
# timeout, even if client signaled that
|
|
# it finished - so we note it just in case.
|
|
join_start = time.time()
|
|
finished_testcase.child.join(test_finished_join_timeout)
|
|
join_end = time.time()
|
|
if join_end - join_start >= test_finished_join_timeout:
|
|
finished_testcase.logger.error(
|
|
"Timeout joining finished test: %s (pid %d)"
|
|
% (finished_testcase.last_test, finished_testcase.child.pid)
|
|
)
|
|
finished_testcase.close_pipes()
|
|
wrapped_testcase_suites.remove(finished_testcase)
|
|
finished_unread_testcases.add(finished_testcase)
|
|
finished_testcase.stdouterr_queue.put(None)
|
|
on_suite_finish(finished_testcase)
|
|
if stop_run:
|
|
while testcase_suites:
|
|
results.append(TestResult(testcase_suites.pop(0)))
|
|
elif testcase_suites:
|
|
a_suite = testcase_suites[0]
|
|
while a_suite and a_suite.is_tagged_run_solo:
|
|
testcase_suites.pop(0)
|
|
solo_testcase_suites.append(a_suite)
|
|
if testcase_suites:
|
|
a_suite = testcase_suites[0]
|
|
else:
|
|
a_suite = None
|
|
if a_suite and can_run_suite(a_suite):
|
|
testcase_suites.pop(0)
|
|
run_suite(a_suite)
|
|
if solo_testcase_suites and tests_running == 0:
|
|
a_suite = solo_testcase_suites.pop(0)
|
|
run_suite(a_suite)
|
|
time.sleep(0.1)
|
|
except Exception:
|
|
for wrapped_testcase_suite in wrapped_testcase_suites:
|
|
wrapped_testcase_suite.child.terminate()
|
|
wrapped_testcase_suite.stdouterr_queue.put(None)
|
|
raise
|
|
finally:
|
|
read_from_testcases.clear()
|
|
stdouterr_thread.join(config.timeout)
|
|
manager.shutdown()
|
|
|
|
handle_cores(failed_wrapped_testcases)
|
|
return results
|
|
|
|
|
|
class TestSuiteWrapper(unittest.TestSuite):
|
|
cpus_used = 0
|
|
|
|
def __init__(self):
|
|
return super().__init__()
|
|
|
|
def addTest(self, test):
|
|
self.cpus_used = max(self.cpus_used, test.get_cpus_required())
|
|
super().addTest(test)
|
|
|
|
def assign_cpus(self, cpus):
|
|
self.cpus = cpus
|
|
|
|
def _handleClassSetUp(self, test, result):
|
|
if not test.__class__.skipped_due_to_cpu_lack:
|
|
test.assign_cpus(self.cpus)
|
|
super()._handleClassSetUp(test, result)
|
|
|
|
def get_assigned_cpus(self):
|
|
return self.cpus
|
|
|
|
|
|
class SplitToSuitesCallback:
|
|
def __init__(self, filter_callback):
|
|
self.suites = {}
|
|
self.suite_name = "default"
|
|
self.filter_callback = filter_callback
|
|
self.filtered = TestSuiteWrapper()
|
|
|
|
def __call__(self, file_name, cls, method):
|
|
test_method = cls(method)
|
|
if self.filter_callback(file_name, cls.__name__, method):
|
|
self.suite_name = file_name + cls.__name__
|
|
if self.suite_name not in self.suites:
|
|
self.suites[self.suite_name] = TestSuiteWrapper()
|
|
self.suites[self.suite_name].is_tagged_run_solo = False
|
|
self.suites[self.suite_name].addTest(test_method)
|
|
if test_method.is_tagged_run_solo():
|
|
self.suites[self.suite_name].is_tagged_run_solo = True
|
|
|
|
else:
|
|
self.filtered.addTest(test_method)
|
|
|
|
|
|
def parse_test_filter(test_filter):
|
|
f = test_filter
|
|
filter_file_name = None
|
|
filter_class_name = None
|
|
filter_func_name = None
|
|
if f:
|
|
if "." in f:
|
|
parts = f.split(".")
|
|
if len(parts) > 3:
|
|
raise Exception(f"Invalid test filter: {test_filter}")
|
|
if len(parts) > 2:
|
|
if parts[2] not in ("*", ""):
|
|
filter_func_name = parts[2]
|
|
if parts[1] not in ("*", ""):
|
|
filter_class_name = parts[1]
|
|
if parts[0] not in ("*", ""):
|
|
if parts[0].startswith("test_"):
|
|
filter_file_name = parts[0]
|
|
else:
|
|
filter_file_name = "test_%s" % parts[0]
|
|
else:
|
|
if f.startswith("test_"):
|
|
filter_file_name = f
|
|
else:
|
|
filter_file_name = "test_%s" % f
|
|
if filter_file_name:
|
|
filter_file_name = "%s.py" % filter_file_name
|
|
return filter_file_name, filter_class_name, filter_func_name
|
|
|
|
|
|
def filter_tests(tests, filter_cb):
|
|
result = TestSuiteWrapper()
|
|
for t in tests:
|
|
if isinstance(t, unittest.suite.TestSuite):
|
|
# this is a bunch of tests, recursively filter...
|
|
x = filter_tests(t, filter_cb)
|
|
if x.countTestCases() > 0:
|
|
result.addTest(x)
|
|
elif isinstance(t, unittest.TestCase):
|
|
# this is a single test
|
|
parts = t.id().split(".")
|
|
# t.id() for common cases like this:
|
|
# test_classifier.TestClassifier.test_acl_ip
|
|
# apply filtering only if it is so
|
|
if len(parts) == 3:
|
|
if not filter_cb(parts[0], parts[1], parts[2]):
|
|
continue
|
|
result.addTest(t)
|
|
else:
|
|
# unexpected object, don't touch it
|
|
result.addTest(t)
|
|
return result
|
|
|
|
|
|
class FilterByTestOption:
|
|
def __init__(self, filters):
|
|
self.filters = filters
|
|
|
|
def __call__(self, file_name, class_name, func_name):
|
|
def test_one(
|
|
filter_file_name,
|
|
filter_class_name,
|
|
filter_func_name,
|
|
file_name,
|
|
class_name,
|
|
func_name,
|
|
):
|
|
if filter_file_name:
|
|
fn_match = fnmatch.fnmatch(file_name, filter_file_name)
|
|
if not fn_match:
|
|
return False
|
|
if filter_class_name and class_name != filter_class_name:
|
|
return False
|
|
if filter_func_name and func_name != filter_func_name:
|
|
return False
|
|
return True
|
|
|
|
for filter_file_name, filter_class_name, filter_func_name in self.filters:
|
|
if test_one(
|
|
filter_file_name,
|
|
filter_class_name,
|
|
filter_func_name,
|
|
file_name,
|
|
class_name,
|
|
func_name,
|
|
):
|
|
return True
|
|
|
|
return False
|
|
|
|
|
|
class FilterByClassList:
|
|
def __init__(self, classes_with_filenames):
|
|
self.classes_with_filenames = classes_with_filenames
|
|
|
|
def __call__(self, file_name, class_name, func_name):
|
|
return ".".join([file_name, class_name]) in self.classes_with_filenames
|
|
|
|
|
|
def suite_from_failed(suite, failed):
|
|
failed = {x.rsplit(".", 1)[0] for x in failed}
|
|
filter_cb = FilterByClassList(failed)
|
|
suite = filter_tests(suite, filter_cb)
|
|
return suite
|
|
|
|
|
|
class AllResults(dict):
|
|
def __init__(self):
|
|
super(AllResults, self).__init__()
|
|
self.all_testcases = 0
|
|
self.results_per_suite = []
|
|
for trc in list(TestResultCode):
|
|
self[trc] = 0
|
|
self.rerun = []
|
|
self.testsuites_no_tests_run = []
|
|
|
|
def add_results(self, result):
|
|
self.results_per_suite.append(result)
|
|
for trc in list(TestResultCode):
|
|
self[trc] += len(result[trc])
|
|
|
|
def add_result(self, result):
|
|
retval = 0
|
|
self.all_testcases += result.testcase_suite.countTestCases()
|
|
self.add_results(result)
|
|
|
|
if result.no_tests_run():
|
|
self.testsuites_no_tests_run.append(result.testcase_suite)
|
|
if result.crashed:
|
|
retval = -1
|
|
else:
|
|
retval = 1
|
|
elif not result.was_successful():
|
|
retval = 1
|
|
|
|
if retval != 0:
|
|
self.rerun.append(result.testcase_suite)
|
|
|
|
return retval
|
|
|
|
def print_results(self):
|
|
print("")
|
|
print(double_line_delim)
|
|
print("TEST RESULTS:")
|
|
|
|
def indent_results(lines):
|
|
lines = list(filter(None, lines))
|
|
maximum = max(lines, key=lambda x: x.index(":"))
|
|
maximum = 4 + maximum.index(":")
|
|
for l in lines:
|
|
padding = " " * (maximum - l.index(":"))
|
|
print(f"{padding}{l}")
|
|
|
|
indent_results(
|
|
[
|
|
f"Scheduled tests: {self.all_testcases}",
|
|
f"Executed tests: {self[TestResultCode.TEST_RUN]}",
|
|
f"Passed tests: {colorize(self[TestResultCode.PASS], GREEN)}",
|
|
(
|
|
f"Expected failures: {colorize(self[TestResultCode.EXPECTED_FAIL], GREEN)}"
|
|
if self[TestResultCode.EXPECTED_FAIL]
|
|
else None
|
|
),
|
|
(
|
|
f"Skipped tests: {colorize(self[TestResultCode.SKIP], YELLOW)}"
|
|
if self[TestResultCode.SKIP]
|
|
else None
|
|
),
|
|
(
|
|
f"Not Executed tests: {colorize(self.not_executed, RED)}"
|
|
if self.not_executed
|
|
else None
|
|
),
|
|
(
|
|
f"Failures: {colorize(self[TestResultCode.FAIL], RED)}"
|
|
if self[TestResultCode.FAIL]
|
|
else None
|
|
),
|
|
(
|
|
f"Unexpected passes: {colorize(self[TestResultCode.UNEXPECTED_PASS], RED)}"
|
|
if self[TestResultCode.UNEXPECTED_PASS]
|
|
else None
|
|
),
|
|
(
|
|
f"Errors: {colorize(self[TestResultCode.ERROR], RED)}"
|
|
if self[TestResultCode.ERROR]
|
|
else None
|
|
),
|
|
(
|
|
"Tests skipped due to lack of CPUS: "
|
|
f"{colorize(self[TestResultCode.SKIP_CPU_SHORTAGE], YELLOW)}"
|
|
if self[TestResultCode.SKIP_CPU_SHORTAGE]
|
|
else None
|
|
),
|
|
]
|
|
)
|
|
|
|
if self.all_failed > 0:
|
|
print("FAILURES AND ERRORS IN TESTS:")
|
|
for result in self.results_per_suite:
|
|
old_testcase_name = None
|
|
for tr_code, headline in (
|
|
(TestResultCode.FAIL, "FAILURE"),
|
|
(TestResultCode.ERROR, "ERROR"),
|
|
(TestResultCode.UNEXPECTED_PASS, "UNEXPECTED PASS"),
|
|
):
|
|
if not result[tr_code]:
|
|
continue
|
|
|
|
for failed_test_id in result[tr_code]:
|
|
new_testcase_name, test_name = result.get_testcase_names(
|
|
failed_test_id
|
|
)
|
|
if new_testcase_name != old_testcase_name:
|
|
print(
|
|
f" Testcase name: {colorize(new_testcase_name, RED)}"
|
|
)
|
|
old_testcase_name = new_testcase_name
|
|
print(
|
|
f" {headline}: {colorize(test_name, RED)} [{failed_test_id}]"
|
|
)
|
|
|
|
if self.testsuites_no_tests_run:
|
|
print("TESTCASES WHERE NO TESTS WERE SUCCESSFULLY EXECUTED:")
|
|
tc_classes = set()
|
|
for testsuite in self.testsuites_no_tests_run:
|
|
for testcase in testsuite:
|
|
tc_classes.add(get_testcase_doc_name(testcase))
|
|
for tc_class in tc_classes:
|
|
print(" {}".format(colorize(tc_class, RED)))
|
|
|
|
if self[TestResultCode.SKIP_CPU_SHORTAGE]:
|
|
print()
|
|
print(
|
|
colorize(
|
|
" SOME TESTS WERE SKIPPED BECAUSE THERE ARE NOT"
|
|
" ENOUGH CPUS AVAILABLE",
|
|
YELLOW,
|
|
)
|
|
)
|
|
print(double_line_delim)
|
|
print("")
|
|
|
|
@property
|
|
def not_executed(self):
|
|
return self.all_testcases - self[TestResultCode.TEST_RUN]
|
|
|
|
@property
|
|
def all_failed(self):
|
|
return (
|
|
self[TestResultCode.FAIL]
|
|
+ self[TestResultCode.ERROR]
|
|
+ self[TestResultCode.UNEXPECTED_PASS]
|
|
)
|
|
|
|
|
|
def parse_results(results):
|
|
"""
|
|
Prints the number of scheduled, executed, not executed, passed, failed,
|
|
errored and skipped tests and details about failed and errored tests.
|
|
|
|
Also returns all suites where any test failed.
|
|
|
|
:param results:
|
|
:return:
|
|
"""
|
|
|
|
results_per_suite = AllResults()
|
|
crashed = False
|
|
failed = False
|
|
for result in results:
|
|
result_code = results_per_suite.add_result(result)
|
|
if result_code == 1:
|
|
failed = True
|
|
elif result_code == -1:
|
|
crashed = True
|
|
|
|
results_per_suite.print_results()
|
|
|
|
if crashed:
|
|
return_code = -1
|
|
elif failed:
|
|
return_code = 1
|
|
else:
|
|
return_code = 0
|
|
return return_code, results_per_suite.rerun
|
|
|
|
|
|
if __name__ == "__main__":
|
|
print(f"Config is: {config}")
|
|
|
|
if config.api_preload:
|
|
VPPApiJSONFiles.load_api(apidir=config.extern_apidir + [config.vpp_install_dir])
|
|
|
|
if config.sanity:
|
|
print("Running sanity test case.")
|
|
try:
|
|
rc = sanity_run_vpp.main()
|
|
if rc != 0:
|
|
sys.exit(rc)
|
|
except Exception as e:
|
|
print(traceback.format_exc())
|
|
print("Couldn't run sanity test case.")
|
|
sys.exit(-1)
|
|
|
|
test_finished_join_timeout = 15
|
|
|
|
debug_gdb = config.debug in ["gdb", "gdbserver", "attach"]
|
|
debug_core = config.debug == "core"
|
|
|
|
run_interactive = debug_gdb or config.step or config.force_foreground
|
|
|
|
max_concurrent_tests = 0
|
|
print(f"OS reports {num_cpus} available cpu(s).")
|
|
|
|
test_jobs = config.jobs
|
|
if test_jobs == "auto":
|
|
if run_interactive:
|
|
max_concurrent_tests = 1
|
|
print("Interactive mode required, running tests consecutively.")
|
|
else:
|
|
max_concurrent_tests = num_cpus
|
|
print(
|
|
f"Running at most {max_concurrent_tests} python test "
|
|
"processes concurrently."
|
|
)
|
|
else:
|
|
max_concurrent_tests = test_jobs
|
|
print(
|
|
f"Running at most {max_concurrent_tests} python test processes "
|
|
"concurrently as set by 'TEST_JOBS'."
|
|
)
|
|
|
|
print(f"Using at most {max_vpp_cpus} cpus for VPP threads.")
|
|
|
|
if run_interactive and max_concurrent_tests > 1:
|
|
raise NotImplementedError(
|
|
"Running tests interactively (DEBUG is gdb[server] or ATTACH or "
|
|
"STEP is set) in parallel (TEST_JOBS is more than 1) is not "
|
|
"supported"
|
|
)
|
|
|
|
descriptions = True
|
|
|
|
print("Running tests using custom test runner.")
|
|
filters = [(parse_test_filter(f)) for f in config.filter.split(",")]
|
|
|
|
print(
|
|
"Selected filters: ",
|
|
"|".join(
|
|
f"file={filter_file}, class={filter_class}, function={filter_func}"
|
|
for filter_file, filter_class, filter_func in filters
|
|
),
|
|
)
|
|
|
|
filter_cb = FilterByTestOption(filters)
|
|
|
|
cb = SplitToSuitesCallback(filter_cb)
|
|
for d in config.test_src_dir:
|
|
print("Adding tests from directory tree %s" % d)
|
|
discover_tests(d, cb)
|
|
|
|
# suites are not hashable, need to use list
|
|
suites = []
|
|
tests_amount = 0
|
|
for testcase_suite in cb.suites.values():
|
|
tests_amount += testcase_suite.countTestCases()
|
|
if testcase_suite.cpus_used > max_vpp_cpus:
|
|
# here we replace test functions with lambdas to just skip them
|
|
# but we also replace setUp/tearDown functions to do nothing
|
|
# so that the test can be "started" and "stopped", so that we can
|
|
# still keep those prints (test description - SKIP), which are done
|
|
# in stopTest() (for that to trigger, test function must run)
|
|
for t in testcase_suite:
|
|
for m in dir(t):
|
|
if m.startswith("test_"):
|
|
setattr(t, m, lambda: t.skipTest("not enough cpus"))
|
|
setattr(t.__class__, "setUpClass", lambda: None)
|
|
setattr(t.__class__, "tearDownClass", lambda: None)
|
|
setattr(t, "setUp", lambda: None)
|
|
setattr(t, "tearDown", lambda: None)
|
|
t.__class__.skipped_due_to_cpu_lack = True
|
|
suites.append(testcase_suite)
|
|
|
|
print(
|
|
"%s out of %s tests match specified filters"
|
|
% (tests_amount, tests_amount + cb.filtered.countTestCases())
|
|
)
|
|
|
|
if not config.extended:
|
|
print("Not running extended tests (some tests will be skipped)")
|
|
|
|
attempts = config.retries + 1
|
|
if attempts > 1:
|
|
print("Perform %s attempts to pass the suite..." % attempts)
|
|
|
|
if run_interactive and suites:
|
|
# don't fork if requiring interactive terminal
|
|
print("Running tests in foreground in the current process")
|
|
full_suite = unittest.TestSuite()
|
|
free_cpus = list(available_cpus)
|
|
cpu_shortage = False
|
|
for suite in suites:
|
|
if suite.cpus_used <= max_vpp_cpus:
|
|
suite.assign_cpus(free_cpus[: suite.cpus_used])
|
|
else:
|
|
suite.assign_cpus([])
|
|
cpu_shortage = True
|
|
full_suite.addTests(suites)
|
|
result = VppTestRunner(
|
|
verbosity=config.verbose, failfast=config.failfast, print_summary=True
|
|
).run(full_suite)
|
|
was_successful = result.wasSuccessful()
|
|
if not was_successful:
|
|
for test_case_info in result.failed_test_cases_info:
|
|
handle_failed_suite(
|
|
test_case_info.logger,
|
|
test_case_info.tempdir,
|
|
test_case_info.vpp_pid,
|
|
config.vpp,
|
|
)
|
|
if test_case_info in result.core_crash_test_cases_info:
|
|
check_and_handle_core(
|
|
test_case_info.vpp_bin_path,
|
|
test_case_info.tempdir,
|
|
test_case_info.core_crash_test,
|
|
)
|
|
|
|
if cpu_shortage:
|
|
print()
|
|
print(
|
|
colorize(
|
|
"SOME TESTS WERE SKIPPED BECAUSE THERE ARE NOT"
|
|
" ENOUGH CPUS AVAILABLE",
|
|
YELLOW,
|
|
)
|
|
)
|
|
print()
|
|
sys.exit(not was_successful)
|
|
else:
|
|
print(
|
|
"Running each VPPTestCase in a separate background process"
|
|
f" with at most {max_concurrent_tests} parallel python test "
|
|
"process(es)"
|
|
)
|
|
exit_code = 0
|
|
while suites and attempts > 0:
|
|
for suite in suites:
|
|
failed_link = get_failed_testcase_linkname(
|
|
config.failed_dir,
|
|
f"{get_testcase_dirname(suite._tests[0].__class__.__name__)}",
|
|
)
|
|
if os.path.islink(failed_link):
|
|
os.unlink(failed_link)
|
|
results = run_forked(suites)
|
|
exit_code, suites = parse_results(results)
|
|
attempts -= 1
|
|
if exit_code == 0:
|
|
print("Test run was successful")
|
|
else:
|
|
print("%s attempt(s) left." % attempts)
|
|
sys.exit(exit_code)
|