10 from multiprocessing import Process, Pipe
11 from framework import VppTestRunner
12 from debug import spawn_gdb
13 from log import global_logger
14 from discover_tests import discover_tests
17 def test_runner_wrapper(suite, keep_alive_pipe, result_pipe, failed_pipe):
18 result = not VppTestRunner(
19 keep_alive_pipe=keep_alive_pipe,
20 failed_pipe=failed_pipe,
22 failfast=failfast).run(suite).wasSuccessful()
23 result_pipe.send(result)
25 keep_alive_pipe.close()
29 class add_to_suite_callback:
30 def __init__(self, suite):
33 def __call__(self, file_name, cls, method):
34 suite.addTest(cls(method))
37 class Filter_by_class_list:
38 def __init__(self, class_list):
39 self.class_list = class_list
41 def __call__(self, file_name, class_name, func_name):
42 return class_name in self.class_list
45 def suite_from_failed(suite, failed):
46 filter_cb = Filter_by_class_list(failed)
47 return VppTestRunner.filter_tests(suite, filter_cb)
50 def run_forked(suite):
51 keep_alive_parent_end, keep_alive_child_end = Pipe(duplex=False)
52 result_parent_end, result_child_end = Pipe(duplex=False)
53 failed_parent_end, failed_child_end = Pipe(duplex=False)
55 child = Process(target=test_runner_wrapper,
56 args=(suite, keep_alive_child_end, result_child_end,
59 last_test_temp_dir = None
60 last_test_vpp_binary = None
64 last_heard = time.time()
66 readable = select.select([keep_alive_parent_end.fileno(),
67 result_parent_end.fileno(),
68 failed_parent_end.fileno(),
71 if result_parent_end.fileno() in readable:
72 result = result_parent_end.recv()
74 if keep_alive_parent_end.fileno() in readable:
75 while keep_alive_parent_end.poll():
76 last_test, last_test_vpp_binary,\
77 last_test_temp_dir, vpp_pid = keep_alive_parent_end.recv()
78 last_heard = time.time()
79 if failed_parent_end.fileno() in readable:
80 while failed_parent_end.poll():
81 failed_test = failed_parent_end.recv()
82 failed.add(failed_test.__name__)
83 last_heard = time.time()
85 if last_heard + test_timeout < time.time():
87 global_logger.critical("Timeout while waiting for child test "
88 "runner process (last test running was "
90 (last_test, last_test_temp_dir))
91 elif not child.is_alive():
93 global_logger.critical("Child process unexpectedly died (last "
94 "test running was `%s' in `%s')!" %
95 (last_test, last_test_temp_dir))
97 failed_dir = os.getenv('VPP_TEST_FAILED_DIR')
98 lttd = last_test_temp_dir.split("/")[-1]
99 link_path = '%s%s-FAILED' % (failed_dir, lttd)
100 global_logger.error("Creating a link to the failed " +
101 "test: %s -> %s" % (link_path, lttd))
102 os.symlink(last_test_temp_dir, link_path)
103 api_post_mortem_path = "/tmp/api_post_mortem.%d" % vpp_pid
104 if os.path.isfile(api_post_mortem_path):
105 global_logger.error("Copying api_post_mortem.%d to %s" %
106 (vpp_pid, last_test_temp_dir))
107 shutil.copy2(api_post_mortem_path, last_test_temp_dir)
108 if last_test_temp_dir and last_test_vpp_binary:
109 core_path = "%s/core" % last_test_temp_dir
110 if os.path.isfile(core_path):
111 global_logger.error("Core-file exists in test temporary "
112 "directory: %s!" % core_path)
113 if d and d.lower() == "core":
114 spawn_gdb(last_test_vpp_binary, core_path,
119 keep_alive_parent_end.close()
120 result_parent_end.close()
121 failed_parent_end.close()
122 return result, failed
125 if __name__ == '__main__':
128 verbose = int(os.getenv("V", 0))
132 default_test_timeout = 600 # 10 minutes
134 test_timeout = int(os.getenv("TIMEOUT", default_test_timeout))
136 test_timeout = default_test_timeout
139 debug = os.getenv("DEBUG")
143 parser = argparse.ArgumentParser(description="VPP unit tests")
144 parser.add_argument("-f", "--failfast", action='count',
145 help="fast failure flag")
146 parser.add_argument("-d", "--dir", action='append', type=str,
147 help="directory containing test files "
148 "(may be specified multiple times)")
149 args = parser.parse_args()
150 failfast = True if args.failfast == 1 else False
152 suite = unittest.TestSuite()
153 cb = add_to_suite_callback(suite)
155 print("Adding tests from directory tree %s" % d)
156 discover_tests(d, cb)
159 retries = int(os.getenv("RETRIES"))
164 attempts = retries + 1
166 print("Perform %s attempts to pass the suite..." % attempts)
167 if debug is None or debug.lower() not in ["gdb", "gdbserver"]:
169 result, failed = run_forked(suite)
170 attempts = attempts - 1
171 print("%s test(s) failed, %s attempt(s) left" %
172 (len(failed), attempts))
173 if len(failed) > 0 and attempts > 0:
174 suite = suite_from_failed(suite, failed)
178 # don't fork if debugging..
179 sys.exit(not VppTestRunner(verbosity=verbose,
180 failfast=failfast).run(suite).wasSuccessful())