blob: 0c3b3f11299fabfb4da407efec716b110c73b1dc [file] [log] [blame]
Damjan Marionf56b77a2016-10-03 19:44:57 +02001#!/usr/bin/env python
2
Klement Sekera993e0ed2017-03-16 09:14:59 +01003import sys
Dave Wallacee2efd122017-09-30 22:04:21 -04004import shutil
Damjan Marionf56b77a2016-10-03 19:44:57 +02005import os
Andrew Yourtchenkod760f792018-10-03 11:38:31 +02006import fnmatch
Damjan Marionf56b77a2016-10-03 19:44:57 +02007import unittest
Klement Sekera993e0ed2017-03-16 09:14:59 +01008import argparse
Klement Sekera545be522018-02-16 19:25:06 +01009import time
juraj.linkes184870a2018-07-16 14:22:01 +020010import threading
11import signal
12import psutil
juraj.linkes40dd73b2018-09-21 13:55:16 +020013import re
juraj.linkes184870a2018-07-16 14:22:01 +020014from multiprocessing import Process, Pipe, cpu_count
15from multiprocessing.queues import Queue
16from multiprocessing.managers import BaseManager
17from framework import VppTestRunner, running_extended_tests, VppTestCase, \
Ole Trøan5ba91592018-11-22 10:01:09 +000018 get_testcase_doc_name, get_test_description, PASS, FAIL, ERROR, SKIP, \
juraj.linkescae64f82018-09-19 15:01:47 +020019 TEST_RUN
Klement Sekera909a6a12017-08-08 04:33:53 +020020from debug import spawn_gdb
juraj.linkes184870a2018-07-16 14:22:01 +020021from log import get_parallel_logger, double_line_delim, RED, YELLOW, GREEN, \
juraj.linkes40dd73b2018-09-21 13:55:16 +020022 colorize, single_line_delim
Klement Sekerafcbf4442017-08-17 07:38:42 +020023from discover_tests import discover_tests
Klement Sekera9b6ece72018-03-23 10:50:11 +010024from subprocess import check_output, CalledProcessError
juraj.linkes40dd73b2018-09-21 13:55:16 +020025from util import check_core_path, get_core_path, is_core_present
Klement Sekera993e0ed2017-03-16 09:14:59 +010026
Klement Sekera05742262018-03-14 18:14:49 +010027# timeout which controls how long the child has to finish after seeing
28# a core dump in test temporary directory. If this is exceeded, parent assumes
29# that child process is stuck (e.g. waiting for shm mutex, which will never
30# get unlocked) and kill the child
31core_timeout = 3
juraj.linkes184870a2018-07-16 14:22:01 +020032min_req_shm = 536870912 # min 512MB shm required
33# 128MB per extra process
34shm_per_process = 134217728
Klement Sekera05742262018-03-14 18:14:49 +010035
Klement Sekera909a6a12017-08-08 04:33:53 +020036
juraj.linkes184870a2018-07-16 14:22:01 +020037class StreamQueue(Queue):
38 def write(self, msg):
39 self.put(msg)
40
41 def flush(self):
42 sys.__stdout__.flush()
43 sys.__stderr__.flush()
44
45 def fileno(self):
46 return self._writer.fileno()
47
48
49class StreamQueueManager(BaseManager):
50 pass
51
52
juraj.linkescae64f82018-09-19 15:01:47 +020053StreamQueueManager.register('StreamQueue', StreamQueue)
juraj.linkes184870a2018-07-16 14:22:01 +020054
55
juraj.linkescae64f82018-09-19 15:01:47 +020056class TestResult(dict):
juraj.linkes40dd73b2018-09-21 13:55:16 +020057 def __init__(self, testcase_suite, testcases_by_id=None):
juraj.linkescae64f82018-09-19 15:01:47 +020058 super(TestResult, self).__init__()
59 self[PASS] = []
60 self[FAIL] = []
61 self[ERROR] = []
62 self[SKIP] = []
63 self[TEST_RUN] = []
juraj.linkes40dd73b2018-09-21 13:55:16 +020064 self.crashed = False
juraj.linkescae64f82018-09-19 15:01:47 +020065 self.testcase_suite = testcase_suite
66 self.testcases = [testcase for testcase in testcase_suite]
juraj.linkes40dd73b2018-09-21 13:55:16 +020067 self.testcases_by_id = testcases_by_id
juraj.linkescae64f82018-09-19 15:01:47 +020068
69 def was_successful(self):
juraj.linkes40dd73b2018-09-21 13:55:16 +020070 return 0 == len(self[FAIL]) == len(self[ERROR]) \
Klement Sekerab8c72a42018-11-08 11:21:39 +010071 and len(self[PASS] + self[SKIP]) \
72 == self.testcase_suite.countTestCases() == len(self[TEST_RUN])
juraj.linkescae64f82018-09-19 15:01:47 +020073
74 def no_tests_run(self):
75 return 0 == len(self[TEST_RUN])
76
77 def process_result(self, test_id, result):
78 self[result].append(test_id)
juraj.linkescae64f82018-09-19 15:01:47 +020079
80 def suite_from_failed(self):
81 rerun_ids = set([])
82 for testcase in self.testcase_suite:
83 tc_id = testcase.id()
84 if tc_id not in self[PASS] and tc_id not in self[SKIP]:
85 rerun_ids.add(tc_id)
Naveen Joy2cbf2fb2019-03-06 10:41:06 -080086 if rerun_ids:
juraj.linkescae64f82018-09-19 15:01:47 +020087 return suite_from_failed(self.testcase_suite, rerun_ids)
88
89 def get_testcase_names(self, test_id):
juraj.linkes2eca70d2018-12-13 11:10:47 +010090 # could be tearDownClass (test_ipsec_esp.TestIpsecEsp1)
91 setup_teardown_match = re.match(
92 r'((tearDownClass)|(setUpClass)) \((.+\..+)\)', test_id)
93 if setup_teardown_match:
94 test_name, _, _, testcase_name = setup_teardown_match.groups()
95 if len(testcase_name.split('.')) == 2:
96 for key in self.testcases_by_id.keys():
97 if key.startswith(testcase_name):
98 testcase_name = key
99 break
100 testcase_name = self._get_testcase_doc_name(testcase_name)
101 else:
Ole Trøan5ba91592018-11-22 10:01:09 +0000102 test_name = self._get_test_description(test_id)
juraj.linkes40dd73b2018-09-21 13:55:16 +0200103 testcase_name = self._get_testcase_doc_name(test_id)
juraj.linkes40dd73b2018-09-21 13:55:16 +0200104
105 return testcase_name, test_name
juraj.linkescae64f82018-09-19 15:01:47 +0200106
Ole Trøan5ba91592018-11-22 10:01:09 +0000107 def _get_test_description(self, test_id):
juraj.linkes2eca70d2018-12-13 11:10:47 +0100108 if test_id in self.testcases_by_id:
109 desc = get_test_description(descriptions,
110 self.testcases_by_id[test_id])
111 else:
112 desc = test_id
113 return desc
Ole Trøan5ba91592018-11-22 10:01:09 +0000114
juraj.linkes40dd73b2018-09-21 13:55:16 +0200115 def _get_testcase_doc_name(self, test_id):
juraj.linkes2eca70d2018-12-13 11:10:47 +0100116 if test_id in self.testcases_by_id:
117 doc_name = get_testcase_doc_name(self.testcases_by_id[test_id])
118 else:
119 doc_name = test_id
120 return doc_name
juraj.linkescae64f82018-09-19 15:01:47 +0200121
122
123def test_runner_wrapper(suite, keep_alive_pipe, stdouterr_queue,
124 finished_pipe, result_pipe, logger):
juraj.linkes184870a2018-07-16 14:22:01 +0200125 sys.stdout = stdouterr_queue
126 sys.stderr = stdouterr_queue
juraj.linkesdfb5f2a2018-11-09 11:58:54 +0100127 VppTestCase.parallel_handler = logger.handlers[0]
juraj.linkes184870a2018-07-16 14:22:01 +0200128 result = VppTestRunner(keep_alive_pipe=keep_alive_pipe,
129 descriptions=descriptions,
130 verbosity=verbose,
juraj.linkescae64f82018-09-19 15:01:47 +0200131 result_pipe=result_pipe,
juraj.linkesabec0122018-11-16 17:28:56 +0100132 failfast=failfast,
133 print_summary=False).run(suite)
juraj.linkescae64f82018-09-19 15:01:47 +0200134 finished_pipe.send(result.wasSuccessful())
135 finished_pipe.close()
Klement Sekera909a6a12017-08-08 04:33:53 +0200136 keep_alive_pipe.close()
137
138
juraj.linkes184870a2018-07-16 14:22:01 +0200139class TestCaseWrapper(object):
140 def __init__(self, testcase_suite, manager):
141 self.keep_alive_parent_end, self.keep_alive_child_end = Pipe(
142 duplex=False)
juraj.linkescae64f82018-09-19 15:01:47 +0200143 self.finished_parent_end, self.finished_child_end = Pipe(duplex=False)
juraj.linkes184870a2018-07-16 14:22:01 +0200144 self.result_parent_end, self.result_child_end = Pipe(duplex=False)
145 self.testcase_suite = testcase_suite
Ole Troan7f991832018-12-06 17:35:12 +0100146 if sys.version[0] == '2':
147 self.stdouterr_queue = manager.StreamQueue()
148 else:
149 from multiprocessing import get_context
150 self.stdouterr_queue = manager.StreamQueue(ctx=get_context())
juraj.linkes184870a2018-07-16 14:22:01 +0200151 self.logger = get_parallel_logger(self.stdouterr_queue)
152 self.child = Process(target=test_runner_wrapper,
juraj.linkescae64f82018-09-19 15:01:47 +0200153 args=(testcase_suite,
154 self.keep_alive_child_end,
155 self.stdouterr_queue,
156 self.finished_child_end,
157 self.result_child_end,
158 self.logger)
juraj.linkes184870a2018-07-16 14:22:01 +0200159 )
160 self.child.start()
juraj.linkes184870a2018-07-16 14:22:01 +0200161 self.last_test_temp_dir = None
162 self.last_test_vpp_binary = None
juraj.linkes40dd73b2018-09-21 13:55:16 +0200163 self._last_test = None
164 self.last_test_id = None
juraj.linkes721872e2018-09-05 18:13:45 +0200165 self.vpp_pid = None
juraj.linkes184870a2018-07-16 14:22:01 +0200166 self.last_heard = time.time()
167 self.core_detected_at = None
juraj.linkes40dd73b2018-09-21 13:55:16 +0200168 self.testcases_by_id = {}
169 self.testclasess_with_core = {}
170 for testcase in self.testcase_suite:
171 self.testcases_by_id[testcase.id()] = testcase
172 self.result = TestResult(testcase_suite, self.testcases_by_id)
173
174 @property
175 def last_test(self):
176 return self._last_test
177
178 @last_test.setter
179 def last_test(self, test_id):
180 self.last_test_id = test_id
181 if test_id in self.testcases_by_id:
182 testcase = self.testcases_by_id[test_id]
183 self._last_test = testcase.shortDescription()
184 if not self._last_test:
185 self._last_test = str(testcase)
186 else:
187 self._last_test = test_id
188
189 def add_testclass_with_core(self):
190 if self.last_test_id in self.testcases_by_id:
191 test = self.testcases_by_id[self.last_test_id]
192 class_name = unittest.util.strclass(test.__class__)
193 test_name = "'{}' ({})".format(get_test_description(descriptions,
194 test),
195 self.last_test_id)
196 else:
197 test_name = self.last_test_id
198 class_name = re.match(r'((tearDownClass)|(setUpClass)) '
199 r'\((.+\..+)\)', test_name).groups()[3]
200 if class_name not in self.testclasess_with_core:
201 self.testclasess_with_core[class_name] = (
202 test_name,
203 self.last_test_vpp_binary,
204 self.last_test_temp_dir)
juraj.linkes184870a2018-07-16 14:22:01 +0200205
206 def close_pipes(self):
207 self.keep_alive_child_end.close()
juraj.linkescae64f82018-09-19 15:01:47 +0200208 self.finished_child_end.close()
juraj.linkes184870a2018-07-16 14:22:01 +0200209 self.result_child_end.close()
210 self.keep_alive_parent_end.close()
juraj.linkescae64f82018-09-19 15:01:47 +0200211 self.finished_parent_end.close()
juraj.linkes184870a2018-07-16 14:22:01 +0200212 self.result_parent_end.close()
213
juraj.linkes40dd73b2018-09-21 13:55:16 +0200214 def was_successful(self):
215 return self.result.was_successful()
216
juraj.linkes184870a2018-07-16 14:22:01 +0200217
218def stdouterr_reader_wrapper(unread_testcases, finished_unread_testcases,
219 read_testcases):
220 read_testcase = None
Naveen Joy2cbf2fb2019-03-06 10:41:06 -0800221 while read_testcases.is_set() or unread_testcases:
222 if finished_unread_testcases:
juraj.linkese6b58cf2018-11-29 09:56:35 +0100223 read_testcase = finished_unread_testcases.pop()
224 unread_testcases.remove(read_testcase)
Naveen Joy2cbf2fb2019-03-06 10:41:06 -0800225 elif unread_testcases:
juraj.linkese6b58cf2018-11-29 09:56:35 +0100226 read_testcase = unread_testcases.pop()
juraj.linkes184870a2018-07-16 14:22:01 +0200227 if read_testcase:
228 data = ''
229 while data is not None:
230 sys.stdout.write(data)
231 data = read_testcase.stdouterr_queue.get()
232
233 read_testcase.stdouterr_queue.close()
234 finished_unread_testcases.discard(read_testcase)
235 read_testcase = None
236
237
juraj.linkes40dd73b2018-09-21 13:55:16 +0200238def handle_failed_suite(logger, last_test_temp_dir, vpp_pid):
239 if last_test_temp_dir:
240 # Need to create link in case of a timeout or core dump without failure
241 lttd = os.path.basename(last_test_temp_dir)
Klement Sekerab8c72a42018-11-08 11:21:39 +0100242 failed_dir = os.getenv('FAILED_DIR')
juraj.linkes40dd73b2018-09-21 13:55:16 +0200243 link_path = '%s%s-FAILED' % (failed_dir, lttd)
244 if not os.path.exists(link_path):
juraj.linkes40dd73b2018-09-21 13:55:16 +0200245 os.symlink(last_test_temp_dir, link_path)
juraj.linkesabec0122018-11-16 17:28:56 +0100246 logger.error("Symlink to failed testcase directory: %s -> %s"
247 % (link_path, lttd))
juraj.linkes40dd73b2018-09-21 13:55:16 +0200248
249 # Report core existence
250 core_path = get_core_path(last_test_temp_dir)
251 if os.path.exists(core_path):
252 logger.error(
253 "Core-file exists in test temporary directory: %s!" %
254 core_path)
255 check_core_path(logger, core_path)
Paul Vinciguerra38a4ec72018-11-28 11:34:21 -0800256 logger.debug("Running 'file %s':" % core_path)
juraj.linkes40dd73b2018-09-21 13:55:16 +0200257 try:
258 info = check_output(["file", core_path])
259 logger.debug(info)
260 except CalledProcessError as e:
Paul Vinciguerra38a4ec72018-11-28 11:34:21 -0800261 logger.error("Subprocess returned with return code "
262 "while running `file' utility on core-file "
263 "returned: "
264 "rc=%s", e.returncode)
265 except OSError as e:
266 logger.error("Subprocess returned with OS error while "
267 "running 'file' utility "
268 "on core-file: "
269 "(%s) %s", e.errno, e.strerror)
270 except Exception as e:
271 logger.exception("Unexpected error running `file' utility "
272 "on core-file")
juraj.linkes40dd73b2018-09-21 13:55:16 +0200273
274 if vpp_pid:
275 # Copy api post mortem
276 api_post_mortem_path = "/tmp/api_post_mortem.%d" % vpp_pid
277 if os.path.isfile(api_post_mortem_path):
278 logger.error("Copying api_post_mortem.%d to %s" %
279 (vpp_pid, last_test_temp_dir))
280 shutil.copy2(api_post_mortem_path, last_test_temp_dir)
281
282
283def check_and_handle_core(vpp_binary, tempdir, core_crash_test):
284 if is_core_present(tempdir):
285 print('VPP core detected in %s. Last test running was %s' %
286 (tempdir, core_crash_test))
287 print(single_line_delim)
288 spawn_gdb(vpp_binary, get_core_path(tempdir))
289 print(single_line_delim)
290
291
292def handle_cores(failed_testcases):
293 if debug_core:
294 for failed_testcase in failed_testcases:
295 tcs_with_core = failed_testcase.testclasess_with_core
Naveen Joy2cbf2fb2019-03-06 10:41:06 -0800296 if tcs_with_core:
juraj.linkes40dd73b2018-09-21 13:55:16 +0200297 for test, vpp_binary, tempdir in tcs_with_core.values():
298 check_and_handle_core(vpp_binary, tempdir, test)
299
300
301def process_finished_testsuite(wrapped_testcase_suite,
302 finished_testcase_suites,
303 failed_wrapped_testcases,
304 results):
305 results.append(wrapped_testcase_suite.result)
306 finished_testcase_suites.add(wrapped_testcase_suite)
307 stop_run = False
308 if failfast and not wrapped_testcase_suite.was_successful():
309 stop_run = True
310
311 if not wrapped_testcase_suite.was_successful():
312 failed_wrapped_testcases.add(wrapped_testcase_suite)
313 handle_failed_suite(wrapped_testcase_suite.logger,
314 wrapped_testcase_suite.last_test_temp_dir,
315 wrapped_testcase_suite.vpp_pid)
316
317 return stop_run
318
319
juraj.linkes721872e2018-09-05 18:13:45 +0200320def run_forked(testcase_suites):
juraj.linkes184870a2018-07-16 14:22:01 +0200321 wrapped_testcase_suites = set()
322
323 # suites are unhashable, need to use list
324 results = []
juraj.linkes184870a2018-07-16 14:22:01 +0200325 unread_testcases = set()
326 finished_unread_testcases = set()
327 manager = StreamQueueManager()
328 manager.start()
329 for i in range(concurrent_tests):
Naveen Joy2cbf2fb2019-03-06 10:41:06 -0800330 if testcase_suites:
juraj.linkes721872e2018-09-05 18:13:45 +0200331 wrapped_testcase_suite = TestCaseWrapper(testcase_suites.pop(0),
332 manager)
juraj.linkes184870a2018-07-16 14:22:01 +0200333 wrapped_testcase_suites.add(wrapped_testcase_suite)
334 unread_testcases.add(wrapped_testcase_suite)
juraj.linkes184870a2018-07-16 14:22:01 +0200335 else:
336 break
337
338 read_from_testcases = threading.Event()
339 read_from_testcases.set()
340 stdouterr_thread = threading.Thread(target=stdouterr_reader_wrapper,
341 args=(unread_testcases,
342 finished_unread_testcases,
343 read_from_testcases))
344 stdouterr_thread.start()
345
juraj.linkes40dd73b2018-09-21 13:55:16 +0200346 failed_wrapped_testcases = set()
347 stop_run = False
juraj.linkese6b58cf2018-11-29 09:56:35 +0100348
349 try:
Naveen Joy2cbf2fb2019-03-06 10:41:06 -0800350 while wrapped_testcase_suites:
juraj.linkese6b58cf2018-11-29 09:56:35 +0100351 finished_testcase_suites = set()
352 for wrapped_testcase_suite in wrapped_testcase_suites:
353 while wrapped_testcase_suite.result_parent_end.poll():
354 wrapped_testcase_suite.result.process_result(
355 *wrapped_testcase_suite.result_parent_end.recv())
356 wrapped_testcase_suite.last_heard = time.time()
357
358 while wrapped_testcase_suite.keep_alive_parent_end.poll():
359 wrapped_testcase_suite.last_test, \
360 wrapped_testcase_suite.last_test_vpp_binary, \
361 wrapped_testcase_suite.last_test_temp_dir, \
362 wrapped_testcase_suite.vpp_pid = \
363 wrapped_testcase_suite.keep_alive_parent_end.recv()
364 wrapped_testcase_suite.last_heard = time.time()
365
366 if wrapped_testcase_suite.finished_parent_end.poll():
367 wrapped_testcase_suite.finished_parent_end.recv()
368 wrapped_testcase_suite.last_heard = time.time()
369 stop_run = process_finished_testsuite(
370 wrapped_testcase_suite,
371 finished_testcase_suites,
372 failed_wrapped_testcases,
373 results) or stop_run
374 continue
375
376 fail = False
377 if wrapped_testcase_suite.last_heard + test_timeout < \
378 time.time():
379 fail = True
380 wrapped_testcase_suite.logger.critical(
381 "Child test runner process timed out "
382 "(last test running was `%s' in `%s')!" %
383 (wrapped_testcase_suite.last_test,
384 wrapped_testcase_suite.last_test_temp_dir))
385 elif not wrapped_testcase_suite.child.is_alive():
386 fail = True
387 wrapped_testcase_suite.logger.critical(
388 "Child test runner process unexpectedly died "
389 "(last test running was `%s' in `%s')!" %
390 (wrapped_testcase_suite.last_test,
391 wrapped_testcase_suite.last_test_temp_dir))
392 elif wrapped_testcase_suite.last_test_temp_dir and \
393 wrapped_testcase_suite.last_test_vpp_binary:
394 if is_core_present(
395 wrapped_testcase_suite.last_test_temp_dir):
396 wrapped_testcase_suite.add_testclass_with_core()
397 if wrapped_testcase_suite.core_detected_at is None:
398 wrapped_testcase_suite.core_detected_at = \
399 time.time()
400 elif wrapped_testcase_suite.core_detected_at + \
401 core_timeout < time.time():
402 wrapped_testcase_suite.logger.critical(
403 "Child test runner process unresponsive and "
404 "core-file exists in test temporary directory "
405 "(last test running was `%s' in `%s')!" %
406 (wrapped_testcase_suite.last_test,
407 wrapped_testcase_suite.last_test_temp_dir))
408 fail = True
409
410 if fail:
411 wrapped_testcase_suite.child.terminate()
412 try:
413 # terminating the child process tends to leave orphan
414 # VPP process around
415 if wrapped_testcase_suite.vpp_pid:
416 os.kill(wrapped_testcase_suite.vpp_pid,
417 signal.SIGTERM)
418 except OSError:
419 # already dead
420 pass
421 wrapped_testcase_suite.result.crashed = True
422 wrapped_testcase_suite.result.process_result(
423 wrapped_testcase_suite.last_test_id, ERROR)
424 stop_run = process_finished_testsuite(
425 wrapped_testcase_suite,
426 finished_testcase_suites,
427 failed_wrapped_testcases,
428 results) or stop_run
429
430 for finished_testcase in finished_testcase_suites:
431 finished_testcase.child.join()
432 finished_testcase.close_pipes()
433 wrapped_testcase_suites.remove(finished_testcase)
434 finished_unread_testcases.add(finished_testcase)
435 finished_testcase.stdouterr_queue.put(None)
436 if stop_run:
Naveen Joy2cbf2fb2019-03-06 10:41:06 -0800437 while testcase_suites:
juraj.linkese6b58cf2018-11-29 09:56:35 +0100438 results.append(TestResult(testcase_suites.pop(0)))
Naveen Joy2cbf2fb2019-03-06 10:41:06 -0800439 elif testcase_suites:
juraj.linkese6b58cf2018-11-29 09:56:35 +0100440 new_testcase = TestCaseWrapper(testcase_suites.pop(0),
441 manager)
442 wrapped_testcase_suites.add(new_testcase)
443 unread_testcases.add(new_testcase)
Paul Vinciguerrac0692a42019-03-15 19:16:50 -0700444 time.sleep(0.1)
juraj.linkese6b58cf2018-11-29 09:56:35 +0100445 except Exception:
juraj.linkes184870a2018-07-16 14:22:01 +0200446 for wrapped_testcase_suite in wrapped_testcase_suites:
juraj.linkese6b58cf2018-11-29 09:56:35 +0100447 wrapped_testcase_suite.child.terminate()
448 wrapped_testcase_suite.stdouterr_queue.put(None)
449 raise
450 finally:
451 read_from_testcases.clear()
452 stdouterr_thread.join(test_timeout)
453 manager.shutdown()
juraj.linkescae64f82018-09-19 15:01:47 +0200454
juraj.linkes40dd73b2018-09-21 13:55:16 +0200455 handle_cores(failed_wrapped_testcases)
juraj.linkes184870a2018-07-16 14:22:01 +0200456 return results
457
458
459class SplitToSuitesCallback:
460 def __init__(self, filter_callback):
461 self.suites = {}
462 self.suite_name = 'default'
463 self.filter_callback = filter_callback
464 self.filtered = unittest.TestSuite()
Klement Sekerafcbf4442017-08-17 07:38:42 +0200465
466 def __call__(self, file_name, cls, method):
juraj.linkes184870a2018-07-16 14:22:01 +0200467 test_method = cls(method)
468 if self.filter_callback(file_name, cls.__name__, method):
469 self.suite_name = file_name + cls.__name__
470 if self.suite_name not in self.suites:
471 self.suites[self.suite_name] = unittest.TestSuite()
472 self.suites[self.suite_name].addTest(test_method)
473
474 else:
475 self.filtered.addTest(test_method)
Klement Sekerafcbf4442017-08-17 07:38:42 +0200476
477
juraj.linkes184870a2018-07-16 14:22:01 +0200478test_option = "TEST"
479
480
481def parse_test_option():
482 f = os.getenv(test_option, None)
483 filter_file_name = None
484 filter_class_name = None
485 filter_func_name = None
486 if f:
487 if '.' in f:
488 parts = f.split('.')
489 if len(parts) > 3:
490 raise Exception("Unrecognized %s option: %s" %
491 (test_option, f))
492 if len(parts) > 2:
493 if parts[2] not in ('*', ''):
494 filter_func_name = parts[2]
495 if parts[1] not in ('*', ''):
496 filter_class_name = parts[1]
497 if parts[0] not in ('*', ''):
498 if parts[0].startswith('test_'):
499 filter_file_name = parts[0]
500 else:
501 filter_file_name = 'test_%s' % parts[0]
502 else:
503 if f.startswith('test_'):
504 filter_file_name = f
505 else:
506 filter_file_name = 'test_%s' % f
507 if filter_file_name:
508 filter_file_name = '%s.py' % filter_file_name
509 return filter_file_name, filter_class_name, filter_func_name
510
511
512def filter_tests(tests, filter_cb):
513 result = unittest.suite.TestSuite()
514 for t in tests:
515 if isinstance(t, unittest.suite.TestSuite):
516 # this is a bunch of tests, recursively filter...
517 x = filter_tests(t, filter_cb)
518 if x.countTestCases() > 0:
519 result.addTest(x)
520 elif isinstance(t, unittest.TestCase):
521 # this is a single test
522 parts = t.id().split('.')
523 # t.id() for common cases like this:
524 # test_classifier.TestClassifier.test_acl_ip
525 # apply filtering only if it is so
526 if len(parts) == 3:
527 if not filter_cb(parts[0], parts[1], parts[2]):
528 continue
529 result.addTest(t)
530 else:
531 # unexpected object, don't touch it
532 result.addTest(t)
533 return result
534
535
536class FilterByTestOption:
537 def __init__(self, filter_file_name, filter_class_name, filter_func_name):
538 self.filter_file_name = filter_file_name
539 self.filter_class_name = filter_class_name
540 self.filter_func_name = filter_func_name
541
542 def __call__(self, file_name, class_name, func_name):
Andrew Yourtchenkod760f792018-10-03 11:38:31 +0200543 if self.filter_file_name:
544 fn_match = fnmatch.fnmatch(file_name, self.filter_file_name)
545 if not fn_match:
546 return False
juraj.linkes184870a2018-07-16 14:22:01 +0200547 if self.filter_class_name and class_name != self.filter_class_name:
548 return False
549 if self.filter_func_name and func_name != self.filter_func_name:
550 return False
551 return True
552
553
554class FilterByClassList:
juraj.linkes721872e2018-09-05 18:13:45 +0200555 def __init__(self, classes_with_filenames):
556 self.classes_with_filenames = classes_with_filenames
Klement Sekeradf2b9802017-10-05 10:26:03 +0200557
558 def __call__(self, file_name, class_name, func_name):
juraj.linkes721872e2018-09-05 18:13:45 +0200559 return '.'.join([file_name, class_name]) in self.classes_with_filenames
Klement Sekeradf2b9802017-10-05 10:26:03 +0200560
561
562def suite_from_failed(suite, failed):
juraj.linkes721872e2018-09-05 18:13:45 +0200563 failed = {x.rsplit('.', 1)[0] for x in failed}
juraj.linkes184870a2018-07-16 14:22:01 +0200564 filter_cb = FilterByClassList(failed)
565 suite = filter_tests(suite, filter_cb)
Klement Sekera4c5422e2018-06-22 13:19:45 +0200566 return suite
Klement Sekeradf2b9802017-10-05 10:26:03 +0200567
568
juraj.linkescae64f82018-09-19 15:01:47 +0200569class AllResults(dict):
juraj.linkes184870a2018-07-16 14:22:01 +0200570 def __init__(self):
juraj.linkescae64f82018-09-19 15:01:47 +0200571 super(AllResults, self).__init__()
juraj.linkes184870a2018-07-16 14:22:01 +0200572 self.all_testcases = 0
juraj.linkescae64f82018-09-19 15:01:47 +0200573 self.results_per_suite = []
574 self[PASS] = 0
575 self[FAIL] = 0
576 self[ERROR] = 0
577 self[SKIP] = 0
578 self[TEST_RUN] = 0
juraj.linkes184870a2018-07-16 14:22:01 +0200579 self.rerun = []
juraj.linkescae64f82018-09-19 15:01:47 +0200580 self.testsuites_no_tests_run = []
Klement Sekera909a6a12017-08-08 04:33:53 +0200581
juraj.linkescae64f82018-09-19 15:01:47 +0200582 def add_results(self, result):
583 self.results_per_suite.append(result)
584 result_types = [PASS, FAIL, ERROR, SKIP, TEST_RUN]
585 for result_type in result_types:
586 self[result_type] += len(result[result_type])
Klement Sekera05742262018-03-14 18:14:49 +0100587
juraj.linkescae64f82018-09-19 15:01:47 +0200588 def add_result(self, result):
juraj.linkes184870a2018-07-16 14:22:01 +0200589 retval = 0
juraj.linkescae64f82018-09-19 15:01:47 +0200590 self.all_testcases += result.testcase_suite.countTestCases()
juraj.linkes40dd73b2018-09-21 13:55:16 +0200591 self.add_results(result)
juraj.linkes184870a2018-07-16 14:22:01 +0200592
juraj.linkes40dd73b2018-09-21 13:55:16 +0200593 if result.no_tests_run():
juraj.linkescae64f82018-09-19 15:01:47 +0200594 self.testsuites_no_tests_run.append(result.testcase_suite)
juraj.linkes40dd73b2018-09-21 13:55:16 +0200595 if result.crashed:
596 retval = -1
597 else:
598 retval = 1
599 elif not result.was_successful():
600 retval = 1
juraj.linkes184870a2018-07-16 14:22:01 +0200601
juraj.linkes184870a2018-07-16 14:22:01 +0200602 if retval != 0:
juraj.linkesabec0122018-11-16 17:28:56 +0100603 self.rerun.append(result.testcase_suite)
juraj.linkes184870a2018-07-16 14:22:01 +0200604
605 return retval
606
607 def print_results(self):
608 print('')
609 print(double_line_delim)
610 print('TEST RESULTS:')
juraj.linkescae64f82018-09-19 15:01:47 +0200611 print(' Scheduled tests: {}'.format(self.all_testcases))
612 print(' Executed tests: {}'.format(self[TEST_RUN]))
613 print(' Passed tests: {}'.format(
614 colorize(str(self[PASS]), GREEN)))
615 if self[SKIP] > 0:
616 print(' Skipped tests: {}'.format(
617 colorize(str(self[SKIP]), YELLOW)))
618 if self.not_executed > 0:
619 print(' Not Executed tests: {}'.format(
620 colorize(str(self.not_executed), RED)))
621 if self[FAIL] > 0:
622 print(' Failures: {}'.format(
623 colorize(str(self[FAIL]), RED)))
624 if self[ERROR] > 0:
625 print(' Errors: {}'.format(
626 colorize(str(self[ERROR]), RED)))
juraj.linkes184870a2018-07-16 14:22:01 +0200627
628 if self.all_failed > 0:
juraj.linkes40dd73b2018-09-21 13:55:16 +0200629 print('FAILURES AND ERRORS IN TESTS:')
juraj.linkescae64f82018-09-19 15:01:47 +0200630 for result in self.results_per_suite:
631 failed_testcase_ids = result[FAIL]
632 errored_testcase_ids = result[ERROR]
633 old_testcase_name = None
Naveen Joy2cbf2fb2019-03-06 10:41:06 -0800634 if failed_testcase_ids or errored_testcase_ids:
juraj.linkescae64f82018-09-19 15:01:47 +0200635 for failed_test_id in failed_testcase_ids:
636 new_testcase_name, test_name = \
637 result.get_testcase_names(failed_test_id)
638 if new_testcase_name != old_testcase_name:
639 print(' Testcase name: {}'.format(
640 colorize(new_testcase_name, RED)))
641 old_testcase_name = new_testcase_name
Klement Sekera33177d62018-11-30 14:17:20 +0100642 print(' FAILURE: {} [{}]'.format(
643 colorize(test_name, RED), failed_test_id))
juraj.linkescae64f82018-09-19 15:01:47 +0200644 for failed_test_id in errored_testcase_ids:
645 new_testcase_name, test_name = \
646 result.get_testcase_names(failed_test_id)
647 if new_testcase_name != old_testcase_name:
648 print(' Testcase name: {}'.format(
649 colorize(new_testcase_name, RED)))
650 old_testcase_name = new_testcase_name
Klement Sekera33177d62018-11-30 14:17:20 +0100651 print(' ERROR: {} [{}]'.format(
652 colorize(test_name, RED), failed_test_id))
Naveen Joy2cbf2fb2019-03-06 10:41:06 -0800653 if self.testsuites_no_tests_run:
juraj.linkescae64f82018-09-19 15:01:47 +0200654 print('TESTCASES WHERE NO TESTS WERE SUCCESSFULLY EXECUTED:')
juraj.linkes40dd73b2018-09-21 13:55:16 +0200655 tc_classes = set()
juraj.linkescae64f82018-09-19 15:01:47 +0200656 for testsuite in self.testsuites_no_tests_run:
657 for testcase in testsuite:
658 tc_classes.add(get_testcase_doc_name(testcase))
659 for tc_class in tc_classes:
660 print(' {}'.format(colorize(tc_class, RED)))
juraj.linkes184870a2018-07-16 14:22:01 +0200661
662 print(double_line_delim)
663 print('')
664
665 @property
juraj.linkescae64f82018-09-19 15:01:47 +0200666 def not_executed(self):
667 return self.all_testcases - self[TEST_RUN]
668
669 @property
juraj.linkes184870a2018-07-16 14:22:01 +0200670 def all_failed(self):
juraj.linkescae64f82018-09-19 15:01:47 +0200671 return self[FAIL] + self[ERROR]
juraj.linkes184870a2018-07-16 14:22:01 +0200672
673
674def parse_results(results):
675 """
juraj.linkescae64f82018-09-19 15:01:47 +0200676 Prints the number of scheduled, executed, not executed, passed, failed,
677 errored and skipped tests and details about failed and errored tests.
juraj.linkes184870a2018-07-16 14:22:01 +0200678
juraj.linkescae64f82018-09-19 15:01:47 +0200679 Also returns all suites where any test failed.
juraj.linkes184870a2018-07-16 14:22:01 +0200680
681 :param results:
682 :return:
683 """
684
juraj.linkescae64f82018-09-19 15:01:47 +0200685 results_per_suite = AllResults()
juraj.linkes184870a2018-07-16 14:22:01 +0200686 crashed = False
687 failed = False
juraj.linkescae64f82018-09-19 15:01:47 +0200688 for result in results:
689 result_code = results_per_suite.add_result(result)
juraj.linkes184870a2018-07-16 14:22:01 +0200690 if result_code == 1:
691 failed = True
692 elif result_code == -1:
693 crashed = True
694
695 results_per_suite.print_results()
696
697 if crashed:
698 return_code = -1
699 elif failed:
700 return_code = 1
701 else:
702 return_code = 0
703 return return_code, results_per_suite.rerun
704
705
706def parse_digit_env(env_var, default):
707 value = os.getenv(env_var, default)
708 if value != default:
709 if value.isdigit():
710 value = int(value)
711 else:
712 print('WARNING: unsupported value "%s" for env var "%s",'
713 'defaulting to %s' % (value, env_var, default))
714 value = default
715 return value
Klement Sekera3f6ff192017-08-11 06:56:05 +0200716
717
718if __name__ == '__main__':
719
juraj.linkes184870a2018-07-16 14:22:01 +0200720 verbose = parse_digit_env("V", 0)
Klement Sekera3f6ff192017-08-11 06:56:05 +0200721
juraj.linkes184870a2018-07-16 14:22:01 +0200722 test_timeout = parse_digit_env("TIMEOUT", 600) # default = 10 minutes
Klement Sekera3f6ff192017-08-11 06:56:05 +0200723
juraj.linkes184870a2018-07-16 14:22:01 +0200724 retries = parse_digit_env("RETRIES", 0)
Klement Sekera3f6ff192017-08-11 06:56:05 +0200725
juraj.linkes184870a2018-07-16 14:22:01 +0200726 debug = os.getenv("DEBUG", "n").lower() in ["gdb", "gdbserver"]
727
juraj.linkes40dd73b2018-09-21 13:55:16 +0200728 debug_core = os.getenv("DEBUG", "").lower() == "core"
729
juraj.linkes184870a2018-07-16 14:22:01 +0200730 step = os.getenv("STEP", "n").lower() in ("y", "yes", "1")
731
juraj.linkes40dd73b2018-09-21 13:55:16 +0200732 run_interactive = debug or step
juraj.linkes184870a2018-07-16 14:22:01 +0200733
734 test_jobs = os.getenv("TEST_JOBS", "1").lower() # default = 1 process
735 if test_jobs == 'auto':
736 if run_interactive:
737 concurrent_tests = 1
738 print('Interactive mode required, running on one core')
739 else:
740 shm_free = psutil.disk_usage('/dev/shm').free
741 shm_max_processes = 1
742 if shm_free < min_req_shm:
743 raise Exception('Not enough free space in /dev/shm. Required '
744 'free space is at least %sM.'
745 % (min_req_shm >> 20))
746 else:
747 extra_shm = shm_free - min_req_shm
748 shm_max_processes += extra_shm / shm_per_process
Klement Sekera6c7bda92018-09-18 14:32:27 +0200749 concurrent_tests = min(cpu_count(), shm_max_processes)
juraj.linkes184870a2018-07-16 14:22:01 +0200750 print('Found enough resources to run tests with %s cores'
751 % concurrent_tests)
752 elif test_jobs.isdigit():
753 concurrent_tests = int(test_jobs)
754 else:
755 concurrent_tests = 1
756
757 if run_interactive and concurrent_tests > 1:
758 raise NotImplementedError(
juraj.linkes40dd73b2018-09-21 13:55:16 +0200759 'Running tests interactively (DEBUG is gdb or gdbserver or STEP '
760 'is set) in parallel (TEST_JOBS is more than 1) is not supported')
Klement Sekera13a83ef2018-03-21 12:35:51 +0100761
Klement Sekera3f6ff192017-08-11 06:56:05 +0200762 parser = argparse.ArgumentParser(description="VPP unit tests")
juraj.linkes184870a2018-07-16 14:22:01 +0200763 parser.add_argument("-f", "--failfast", action='store_true',
Klement Sekera3f6ff192017-08-11 06:56:05 +0200764 help="fast failure flag")
765 parser.add_argument("-d", "--dir", action='append', type=str,
766 help="directory containing test files "
767 "(may be specified multiple times)")
768 args = parser.parse_args()
juraj.linkes184870a2018-07-16 14:22:01 +0200769 failfast = args.failfast
770 descriptions = True
Klement Sekera3f6ff192017-08-11 06:56:05 +0200771
juraj.linkes184870a2018-07-16 14:22:01 +0200772 print("Running tests using custom test runner") # debug message
773 filter_file, filter_class, filter_func = parse_test_option()
774
775 print("Active filters: file=%s, class=%s, function=%s" % (
776 filter_file, filter_class, filter_func))
777
778 filter_cb = FilterByTestOption(filter_file, filter_class, filter_func)
779
Klement Sekerab8c72a42018-11-08 11:21:39 +0100780 ignore_path = os.getenv("VENV_PATH", None)
juraj.linkes184870a2018-07-16 14:22:01 +0200781 cb = SplitToSuitesCallback(filter_cb)
Klement Sekera3f6ff192017-08-11 06:56:05 +0200782 for d in args.dir:
Klement Sekeradf2b9802017-10-05 10:26:03 +0200783 print("Adding tests from directory tree %s" % d)
Klement Sekerab8c72a42018-11-08 11:21:39 +0100784 discover_tests(d, cb, ignore_path)
Klement Sekera3f6ff192017-08-11 06:56:05 +0200785
juraj.linkes184870a2018-07-16 14:22:01 +0200786 # suites are not hashable, need to use list
787 suites = []
788 tests_amount = 0
789 for testcase_suite in cb.suites.values():
790 tests_amount += testcase_suite.countTestCases()
791 suites.append(testcase_suite)
Klement Sekerabbfa5fd2018-06-27 13:54:32 +0200792
juraj.linkes184870a2018-07-16 14:22:01 +0200793 print("%s out of %s tests match specified filters" % (
794 tests_amount, tests_amount + cb.filtered.countTestCases()))
795
Paul Vinciguerradefde0f2018-12-06 07:46:13 -0800796 if not running_extended_tests:
juraj.linkes184870a2018-07-16 14:22:01 +0200797 print("Not running extended tests (some tests will be skipped)")
798
Klement Sekeradf2b9802017-10-05 10:26:03 +0200799 attempts = retries + 1
800 if attempts > 1:
801 print("Perform %s attempts to pass the suite..." % attempts)
juraj.linkes184870a2018-07-16 14:22:01 +0200802
Naveen Joy2cbf2fb2019-03-06 10:41:06 -0800803 if run_interactive and suites:
juraj.linkes184870a2018-07-16 14:22:01 +0200804 # don't fork if requiring interactive terminal
juraj.linkes46e8e912019-01-10 12:13:07 +0100805 full_suite = unittest.TestSuite()
806 map(full_suite.addTests, suites)
juraj.linkesabec0122018-11-16 17:28:56 +0100807 result = VppTestRunner(verbosity=verbose,
808 failfast=failfast,
juraj.linkes46e8e912019-01-10 12:13:07 +0100809 print_summary=True).run(full_suite)
juraj.linkes40dd73b2018-09-21 13:55:16 +0200810 was_successful = result.wasSuccessful()
811 if not was_successful:
812 for test_case_info in result.failed_test_cases_info:
813 handle_failed_suite(test_case_info.logger,
814 test_case_info.tempdir,
815 test_case_info.vpp_pid)
816 if debug_core and \
817 test_case_info in result.core_crash_test_cases_info:
818 check_and_handle_core(test_case_info.vpp_bin_path,
819 test_case_info.tempdir,
820 test_case_info.core_crash_test)
821
822 sys.exit(not was_successful)
Klement Sekera13a83ef2018-03-21 12:35:51 +0100823 else:
juraj.linkes184870a2018-07-16 14:22:01 +0200824 exit_code = 0
Naveen Joy2cbf2fb2019-03-06 10:41:06 -0800825 while suites and attempts > 0:
juraj.linkes184870a2018-07-16 14:22:01 +0200826 results = run_forked(suites)
827 exit_code, suites = parse_results(results)
828 attempts -= 1
829 if exit_code == 0:
830 print('Test run was successful')
831 else:
832 print('%s attempt(s) left.' % attempts)
833 sys.exit(exit_code)