blob: b6c178f234fc2a8fc774e2362382fcad9df96c17 [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
Paul Vinciguerra025cd9c2019-07-08 14:14:22 -040014import multiprocessing
juraj.linkes184870a2018-07-16 14:22:01 +020015from multiprocessing import Process, Pipe, cpu_count
16from multiprocessing.queues import Queue
17from multiprocessing.managers import BaseManager
Paul Vinciguerra0cbc71d2019-07-03 08:38:38 -040018import framework
juraj.linkes184870a2018-07-16 14:22:01 +020019from framework import VppTestRunner, running_extended_tests, VppTestCase, \
Ole Trøan5ba91592018-11-22 10:01:09 +000020 get_testcase_doc_name, get_test_description, PASS, FAIL, ERROR, SKIP, \
juraj.linkescae64f82018-09-19 15:01:47 +020021 TEST_RUN
Klement Sekera909a6a12017-08-08 04:33:53 +020022from debug import spawn_gdb
juraj.linkes184870a2018-07-16 14:22:01 +020023from log import get_parallel_logger, double_line_delim, RED, YELLOW, GREEN, \
juraj.linkes40dd73b2018-09-21 13:55:16 +020024 colorize, single_line_delim
Klement Sekerafcbf4442017-08-17 07:38:42 +020025from discover_tests import discover_tests
Klement Sekera9b6ece72018-03-23 10:50:11 +010026from subprocess import check_output, CalledProcessError
juraj.linkes40dd73b2018-09-21 13:55:16 +020027from util import check_core_path, get_core_path, is_core_present
Klement Sekera993e0ed2017-03-16 09:14:59 +010028
Klement Sekera05742262018-03-14 18:14:49 +010029# timeout which controls how long the child has to finish after seeing
30# a core dump in test temporary directory. If this is exceeded, parent assumes
31# that child process is stuck (e.g. waiting for shm mutex, which will never
32# get unlocked) and kill the child
33core_timeout = 3
juraj.linkes184870a2018-07-16 14:22:01 +020034min_req_shm = 536870912 # min 512MB shm required
35# 128MB per extra process
36shm_per_process = 134217728
Klement Sekera05742262018-03-14 18:14:49 +010037
Klement Sekera909a6a12017-08-08 04:33:53 +020038
juraj.linkes184870a2018-07-16 14:22:01 +020039class StreamQueue(Queue):
40 def write(self, msg):
41 self.put(msg)
42
43 def flush(self):
44 sys.__stdout__.flush()
45 sys.__stderr__.flush()
46
47 def fileno(self):
48 return self._writer.fileno()
49
50
51class StreamQueueManager(BaseManager):
52 pass
53
54
juraj.linkescae64f82018-09-19 15:01:47 +020055StreamQueueManager.register('StreamQueue', StreamQueue)
juraj.linkes184870a2018-07-16 14:22:01 +020056
57
juraj.linkescae64f82018-09-19 15:01:47 +020058class TestResult(dict):
juraj.linkes40dd73b2018-09-21 13:55:16 +020059 def __init__(self, testcase_suite, testcases_by_id=None):
juraj.linkescae64f82018-09-19 15:01:47 +020060 super(TestResult, self).__init__()
61 self[PASS] = []
62 self[FAIL] = []
63 self[ERROR] = []
64 self[SKIP] = []
65 self[TEST_RUN] = []
juraj.linkes40dd73b2018-09-21 13:55:16 +020066 self.crashed = False
juraj.linkescae64f82018-09-19 15:01:47 +020067 self.testcase_suite = testcase_suite
68 self.testcases = [testcase for testcase in testcase_suite]
juraj.linkes40dd73b2018-09-21 13:55:16 +020069 self.testcases_by_id = testcases_by_id
juraj.linkescae64f82018-09-19 15:01:47 +020070
71 def was_successful(self):
juraj.linkes40dd73b2018-09-21 13:55:16 +020072 return 0 == len(self[FAIL]) == len(self[ERROR]) \
Klement Sekerab8c72a42018-11-08 11:21:39 +010073 and len(self[PASS] + self[SKIP]) \
74 == self.testcase_suite.countTestCases() == len(self[TEST_RUN])
juraj.linkescae64f82018-09-19 15:01:47 +020075
76 def no_tests_run(self):
77 return 0 == len(self[TEST_RUN])
78
79 def process_result(self, test_id, result):
80 self[result].append(test_id)
juraj.linkescae64f82018-09-19 15:01:47 +020081
82 def suite_from_failed(self):
83 rerun_ids = set([])
84 for testcase in self.testcase_suite:
85 tc_id = testcase.id()
86 if tc_id not in self[PASS] and tc_id not in self[SKIP]:
87 rerun_ids.add(tc_id)
Naveen Joy2cbf2fb2019-03-06 10:41:06 -080088 if rerun_ids:
juraj.linkescae64f82018-09-19 15:01:47 +020089 return suite_from_failed(self.testcase_suite, rerun_ids)
90
91 def get_testcase_names(self, test_id):
juraj.linkes2eca70d2018-12-13 11:10:47 +010092 # could be tearDownClass (test_ipsec_esp.TestIpsecEsp1)
93 setup_teardown_match = re.match(
94 r'((tearDownClass)|(setUpClass)) \((.+\..+)\)', test_id)
95 if setup_teardown_match:
96 test_name, _, _, testcase_name = setup_teardown_match.groups()
97 if len(testcase_name.split('.')) == 2:
98 for key in self.testcases_by_id.keys():
99 if key.startswith(testcase_name):
100 testcase_name = key
101 break
102 testcase_name = self._get_testcase_doc_name(testcase_name)
103 else:
Ole Trøan5ba91592018-11-22 10:01:09 +0000104 test_name = self._get_test_description(test_id)
juraj.linkes40dd73b2018-09-21 13:55:16 +0200105 testcase_name = self._get_testcase_doc_name(test_id)
juraj.linkes40dd73b2018-09-21 13:55:16 +0200106
107 return testcase_name, test_name
juraj.linkescae64f82018-09-19 15:01:47 +0200108
Ole Trøan5ba91592018-11-22 10:01:09 +0000109 def _get_test_description(self, test_id):
juraj.linkes2eca70d2018-12-13 11:10:47 +0100110 if test_id in self.testcases_by_id:
111 desc = get_test_description(descriptions,
112 self.testcases_by_id[test_id])
113 else:
114 desc = test_id
115 return desc
Ole Trøan5ba91592018-11-22 10:01:09 +0000116
juraj.linkes40dd73b2018-09-21 13:55:16 +0200117 def _get_testcase_doc_name(self, test_id):
juraj.linkes2eca70d2018-12-13 11:10:47 +0100118 if test_id in self.testcases_by_id:
119 doc_name = get_testcase_doc_name(self.testcases_by_id[test_id])
120 else:
121 doc_name = test_id
122 return doc_name
juraj.linkescae64f82018-09-19 15:01:47 +0200123
124
125def test_runner_wrapper(suite, keep_alive_pipe, stdouterr_queue,
126 finished_pipe, result_pipe, logger):
juraj.linkes184870a2018-07-16 14:22:01 +0200127 sys.stdout = stdouterr_queue
128 sys.stderr = stdouterr_queue
juraj.linkesdfb5f2a2018-11-09 11:58:54 +0100129 VppTestCase.parallel_handler = logger.handlers[0]
juraj.linkes184870a2018-07-16 14:22:01 +0200130 result = VppTestRunner(keep_alive_pipe=keep_alive_pipe,
131 descriptions=descriptions,
132 verbosity=verbose,
juraj.linkescae64f82018-09-19 15:01:47 +0200133 result_pipe=result_pipe,
juraj.linkesabec0122018-11-16 17:28:56 +0100134 failfast=failfast,
135 print_summary=False).run(suite)
juraj.linkescae64f82018-09-19 15:01:47 +0200136 finished_pipe.send(result.wasSuccessful())
137 finished_pipe.close()
Klement Sekera909a6a12017-08-08 04:33:53 +0200138 keep_alive_pipe.close()
139
140
juraj.linkes184870a2018-07-16 14:22:01 +0200141class TestCaseWrapper(object):
142 def __init__(self, testcase_suite, manager):
143 self.keep_alive_parent_end, self.keep_alive_child_end = Pipe(
144 duplex=False)
juraj.linkescae64f82018-09-19 15:01:47 +0200145 self.finished_parent_end, self.finished_child_end = Pipe(duplex=False)
juraj.linkes184870a2018-07-16 14:22:01 +0200146 self.result_parent_end, self.result_child_end = Pipe(duplex=False)
147 self.testcase_suite = testcase_suite
Ole Troan7f991832018-12-06 17:35:12 +0100148 if sys.version[0] == '2':
149 self.stdouterr_queue = manager.StreamQueue()
150 else:
151 from multiprocessing import get_context
152 self.stdouterr_queue = manager.StreamQueue(ctx=get_context())
juraj.linkes184870a2018-07-16 14:22:01 +0200153 self.logger = get_parallel_logger(self.stdouterr_queue)
154 self.child = Process(target=test_runner_wrapper,
juraj.linkescae64f82018-09-19 15:01:47 +0200155 args=(testcase_suite,
156 self.keep_alive_child_end,
157 self.stdouterr_queue,
158 self.finished_child_end,
159 self.result_child_end,
160 self.logger)
juraj.linkes184870a2018-07-16 14:22:01 +0200161 )
162 self.child.start()
juraj.linkes184870a2018-07-16 14:22:01 +0200163 self.last_test_temp_dir = None
164 self.last_test_vpp_binary = None
juraj.linkes40dd73b2018-09-21 13:55:16 +0200165 self._last_test = None
166 self.last_test_id = None
juraj.linkes721872e2018-09-05 18:13:45 +0200167 self.vpp_pid = None
juraj.linkes184870a2018-07-16 14:22:01 +0200168 self.last_heard = time.time()
169 self.core_detected_at = None
juraj.linkes40dd73b2018-09-21 13:55:16 +0200170 self.testcases_by_id = {}
171 self.testclasess_with_core = {}
172 for testcase in self.testcase_suite:
173 self.testcases_by_id[testcase.id()] = testcase
174 self.result = TestResult(testcase_suite, self.testcases_by_id)
175
176 @property
177 def last_test(self):
178 return self._last_test
179
180 @last_test.setter
181 def last_test(self, test_id):
182 self.last_test_id = test_id
183 if test_id in self.testcases_by_id:
184 testcase = self.testcases_by_id[test_id]
185 self._last_test = testcase.shortDescription()
186 if not self._last_test:
187 self._last_test = str(testcase)
188 else:
189 self._last_test = test_id
190
191 def add_testclass_with_core(self):
192 if self.last_test_id in self.testcases_by_id:
193 test = self.testcases_by_id[self.last_test_id]
194 class_name = unittest.util.strclass(test.__class__)
195 test_name = "'{}' ({})".format(get_test_description(descriptions,
196 test),
197 self.last_test_id)
198 else:
199 test_name = self.last_test_id
200 class_name = re.match(r'((tearDownClass)|(setUpClass)) '
201 r'\((.+\..+)\)', test_name).groups()[3]
202 if class_name not in self.testclasess_with_core:
203 self.testclasess_with_core[class_name] = (
204 test_name,
205 self.last_test_vpp_binary,
206 self.last_test_temp_dir)
juraj.linkes184870a2018-07-16 14:22:01 +0200207
208 def close_pipes(self):
209 self.keep_alive_child_end.close()
juraj.linkescae64f82018-09-19 15:01:47 +0200210 self.finished_child_end.close()
juraj.linkes184870a2018-07-16 14:22:01 +0200211 self.result_child_end.close()
212 self.keep_alive_parent_end.close()
juraj.linkescae64f82018-09-19 15:01:47 +0200213 self.finished_parent_end.close()
juraj.linkes184870a2018-07-16 14:22:01 +0200214 self.result_parent_end.close()
215
juraj.linkes40dd73b2018-09-21 13:55:16 +0200216 def was_successful(self):
217 return self.result.was_successful()
218
juraj.linkes184870a2018-07-16 14:22:01 +0200219
220def stdouterr_reader_wrapper(unread_testcases, finished_unread_testcases,
221 read_testcases):
222 read_testcase = None
Naveen Joy2cbf2fb2019-03-06 10:41:06 -0800223 while read_testcases.is_set() or unread_testcases:
224 if finished_unread_testcases:
juraj.linkese6b58cf2018-11-29 09:56:35 +0100225 read_testcase = finished_unread_testcases.pop()
226 unread_testcases.remove(read_testcase)
Naveen Joy2cbf2fb2019-03-06 10:41:06 -0800227 elif unread_testcases:
juraj.linkese6b58cf2018-11-29 09:56:35 +0100228 read_testcase = unread_testcases.pop()
juraj.linkes184870a2018-07-16 14:22:01 +0200229 if read_testcase:
230 data = ''
231 while data is not None:
232 sys.stdout.write(data)
233 data = read_testcase.stdouterr_queue.get()
234
235 read_testcase.stdouterr_queue.close()
236 finished_unread_testcases.discard(read_testcase)
237 read_testcase = None
238
239
juraj.linkes40dd73b2018-09-21 13:55:16 +0200240def handle_failed_suite(logger, last_test_temp_dir, vpp_pid):
241 if last_test_temp_dir:
242 # Need to create link in case of a timeout or core dump without failure
243 lttd = os.path.basename(last_test_temp_dir)
Klement Sekerab8c72a42018-11-08 11:21:39 +0100244 failed_dir = os.getenv('FAILED_DIR')
juraj.linkes40dd73b2018-09-21 13:55:16 +0200245 link_path = '%s%s-FAILED' % (failed_dir, lttd)
246 if not os.path.exists(link_path):
juraj.linkes40dd73b2018-09-21 13:55:16 +0200247 os.symlink(last_test_temp_dir, link_path)
juraj.linkesabec0122018-11-16 17:28:56 +0100248 logger.error("Symlink to failed testcase directory: %s -> %s"
249 % (link_path, lttd))
juraj.linkes40dd73b2018-09-21 13:55:16 +0200250
251 # Report core existence
252 core_path = get_core_path(last_test_temp_dir)
253 if os.path.exists(core_path):
254 logger.error(
255 "Core-file exists in test temporary directory: %s!" %
256 core_path)
257 check_core_path(logger, core_path)
Paul Vinciguerra38a4ec72018-11-28 11:34:21 -0800258 logger.debug("Running 'file %s':" % core_path)
juraj.linkes40dd73b2018-09-21 13:55:16 +0200259 try:
260 info = check_output(["file", core_path])
261 logger.debug(info)
262 except CalledProcessError as e:
Paul Vinciguerra38a4ec72018-11-28 11:34:21 -0800263 logger.error("Subprocess returned with return code "
264 "while running `file' utility on core-file "
265 "returned: "
266 "rc=%s", e.returncode)
267 except OSError as e:
268 logger.error("Subprocess returned with OS error while "
269 "running 'file' utility "
270 "on core-file: "
271 "(%s) %s", e.errno, e.strerror)
272 except Exception as e:
273 logger.exception("Unexpected error running `file' utility "
274 "on core-file")
juraj.linkes40dd73b2018-09-21 13:55:16 +0200275
276 if vpp_pid:
277 # Copy api post mortem
278 api_post_mortem_path = "/tmp/api_post_mortem.%d" % vpp_pid
279 if os.path.isfile(api_post_mortem_path):
280 logger.error("Copying api_post_mortem.%d to %s" %
281 (vpp_pid, last_test_temp_dir))
282 shutil.copy2(api_post_mortem_path, last_test_temp_dir)
283
284
285def check_and_handle_core(vpp_binary, tempdir, core_crash_test):
286 if is_core_present(tempdir):
Klement Sekeraf40ee3a2019-05-06 19:11:25 +0200287 if debug_core:
288 print('VPP core detected in %s. Last test running was %s' %
289 (tempdir, core_crash_test))
290 print(single_line_delim)
291 spawn_gdb(vpp_binary, get_core_path(tempdir))
292 print(single_line_delim)
293 elif compress_core:
294 print("Compressing core-file in test directory `%s'" % tempdir)
295 os.system("gzip %s" % get_core_path(tempdir))
juraj.linkes40dd73b2018-09-21 13:55:16 +0200296
297
298def handle_cores(failed_testcases):
Klement Sekeraf40ee3a2019-05-06 19:11:25 +0200299 for failed_testcase in failed_testcases:
300 tcs_with_core = failed_testcase.testclasess_with_core
301 if tcs_with_core:
302 for test, vpp_binary, tempdir in tcs_with_core.values():
303 check_and_handle_core(vpp_binary, tempdir, test)
juraj.linkes40dd73b2018-09-21 13:55:16 +0200304
305
306def process_finished_testsuite(wrapped_testcase_suite,
307 finished_testcase_suites,
308 failed_wrapped_testcases,
309 results):
310 results.append(wrapped_testcase_suite.result)
311 finished_testcase_suites.add(wrapped_testcase_suite)
312 stop_run = False
313 if failfast and not wrapped_testcase_suite.was_successful():
314 stop_run = True
315
316 if not wrapped_testcase_suite.was_successful():
317 failed_wrapped_testcases.add(wrapped_testcase_suite)
318 handle_failed_suite(wrapped_testcase_suite.logger,
319 wrapped_testcase_suite.last_test_temp_dir,
320 wrapped_testcase_suite.vpp_pid)
321
322 return stop_run
323
324
juraj.linkes721872e2018-09-05 18:13:45 +0200325def run_forked(testcase_suites):
juraj.linkes184870a2018-07-16 14:22:01 +0200326 wrapped_testcase_suites = set()
327
328 # suites are unhashable, need to use list
329 results = []
juraj.linkes184870a2018-07-16 14:22:01 +0200330 unread_testcases = set()
331 finished_unread_testcases = set()
332 manager = StreamQueueManager()
333 manager.start()
334 for i in range(concurrent_tests):
Naveen Joy2cbf2fb2019-03-06 10:41:06 -0800335 if testcase_suites:
juraj.linkes721872e2018-09-05 18:13:45 +0200336 wrapped_testcase_suite = TestCaseWrapper(testcase_suites.pop(0),
337 manager)
juraj.linkes184870a2018-07-16 14:22:01 +0200338 wrapped_testcase_suites.add(wrapped_testcase_suite)
339 unread_testcases.add(wrapped_testcase_suite)
juraj.linkes184870a2018-07-16 14:22:01 +0200340 else:
341 break
342
343 read_from_testcases = threading.Event()
344 read_from_testcases.set()
345 stdouterr_thread = threading.Thread(target=stdouterr_reader_wrapper,
346 args=(unread_testcases,
347 finished_unread_testcases,
348 read_from_testcases))
349 stdouterr_thread.start()
350
juraj.linkes40dd73b2018-09-21 13:55:16 +0200351 failed_wrapped_testcases = set()
352 stop_run = False
juraj.linkese6b58cf2018-11-29 09:56:35 +0100353
354 try:
Naveen Joy2cbf2fb2019-03-06 10:41:06 -0800355 while wrapped_testcase_suites:
juraj.linkese6b58cf2018-11-29 09:56:35 +0100356 finished_testcase_suites = set()
357 for wrapped_testcase_suite in wrapped_testcase_suites:
358 while wrapped_testcase_suite.result_parent_end.poll():
359 wrapped_testcase_suite.result.process_result(
360 *wrapped_testcase_suite.result_parent_end.recv())
361 wrapped_testcase_suite.last_heard = time.time()
362
363 while wrapped_testcase_suite.keep_alive_parent_end.poll():
364 wrapped_testcase_suite.last_test, \
365 wrapped_testcase_suite.last_test_vpp_binary, \
366 wrapped_testcase_suite.last_test_temp_dir, \
367 wrapped_testcase_suite.vpp_pid = \
368 wrapped_testcase_suite.keep_alive_parent_end.recv()
369 wrapped_testcase_suite.last_heard = time.time()
370
371 if wrapped_testcase_suite.finished_parent_end.poll():
372 wrapped_testcase_suite.finished_parent_end.recv()
373 wrapped_testcase_suite.last_heard = time.time()
374 stop_run = process_finished_testsuite(
375 wrapped_testcase_suite,
376 finished_testcase_suites,
377 failed_wrapped_testcases,
378 results) or stop_run
379 continue
380
381 fail = False
382 if wrapped_testcase_suite.last_heard + test_timeout < \
383 time.time():
384 fail = True
385 wrapped_testcase_suite.logger.critical(
386 "Child test runner process timed out "
387 "(last test running was `%s' in `%s')!" %
388 (wrapped_testcase_suite.last_test,
389 wrapped_testcase_suite.last_test_temp_dir))
390 elif not wrapped_testcase_suite.child.is_alive():
391 fail = True
392 wrapped_testcase_suite.logger.critical(
393 "Child test runner process unexpectedly died "
394 "(last test running was `%s' in `%s')!" %
395 (wrapped_testcase_suite.last_test,
396 wrapped_testcase_suite.last_test_temp_dir))
397 elif wrapped_testcase_suite.last_test_temp_dir and \
398 wrapped_testcase_suite.last_test_vpp_binary:
399 if is_core_present(
400 wrapped_testcase_suite.last_test_temp_dir):
401 wrapped_testcase_suite.add_testclass_with_core()
402 if wrapped_testcase_suite.core_detected_at is None:
403 wrapped_testcase_suite.core_detected_at = \
404 time.time()
405 elif wrapped_testcase_suite.core_detected_at + \
406 core_timeout < time.time():
407 wrapped_testcase_suite.logger.critical(
408 "Child test runner process unresponsive and "
409 "core-file exists in test temporary directory "
410 "(last test running was `%s' in `%s')!" %
411 (wrapped_testcase_suite.last_test,
412 wrapped_testcase_suite.last_test_temp_dir))
413 fail = True
414
415 if fail:
416 wrapped_testcase_suite.child.terminate()
417 try:
418 # terminating the child process tends to leave orphan
419 # VPP process around
420 if wrapped_testcase_suite.vpp_pid:
421 os.kill(wrapped_testcase_suite.vpp_pid,
422 signal.SIGTERM)
423 except OSError:
424 # already dead
425 pass
426 wrapped_testcase_suite.result.crashed = True
427 wrapped_testcase_suite.result.process_result(
428 wrapped_testcase_suite.last_test_id, ERROR)
429 stop_run = process_finished_testsuite(
430 wrapped_testcase_suite,
431 finished_testcase_suites,
432 failed_wrapped_testcases,
433 results) or stop_run
434
435 for finished_testcase in finished_testcase_suites:
436 finished_testcase.child.join()
437 finished_testcase.close_pipes()
438 wrapped_testcase_suites.remove(finished_testcase)
439 finished_unread_testcases.add(finished_testcase)
440 finished_testcase.stdouterr_queue.put(None)
441 if stop_run:
Naveen Joy2cbf2fb2019-03-06 10:41:06 -0800442 while testcase_suites:
juraj.linkese6b58cf2018-11-29 09:56:35 +0100443 results.append(TestResult(testcase_suites.pop(0)))
Naveen Joy2cbf2fb2019-03-06 10:41:06 -0800444 elif testcase_suites:
juraj.linkese6b58cf2018-11-29 09:56:35 +0100445 new_testcase = TestCaseWrapper(testcase_suites.pop(0),
446 manager)
447 wrapped_testcase_suites.add(new_testcase)
448 unread_testcases.add(new_testcase)
Paul Vinciguerrac0692a42019-03-15 19:16:50 -0700449 time.sleep(0.1)
juraj.linkese6b58cf2018-11-29 09:56:35 +0100450 except Exception:
juraj.linkes184870a2018-07-16 14:22:01 +0200451 for wrapped_testcase_suite in wrapped_testcase_suites:
juraj.linkese6b58cf2018-11-29 09:56:35 +0100452 wrapped_testcase_suite.child.terminate()
453 wrapped_testcase_suite.stdouterr_queue.put(None)
454 raise
455 finally:
456 read_from_testcases.clear()
457 stdouterr_thread.join(test_timeout)
458 manager.shutdown()
juraj.linkescae64f82018-09-19 15:01:47 +0200459
juraj.linkes40dd73b2018-09-21 13:55:16 +0200460 handle_cores(failed_wrapped_testcases)
juraj.linkes184870a2018-07-16 14:22:01 +0200461 return results
462
463
464class SplitToSuitesCallback:
465 def __init__(self, filter_callback):
466 self.suites = {}
467 self.suite_name = 'default'
468 self.filter_callback = filter_callback
469 self.filtered = unittest.TestSuite()
Klement Sekerafcbf4442017-08-17 07:38:42 +0200470
471 def __call__(self, file_name, cls, method):
juraj.linkes184870a2018-07-16 14:22:01 +0200472 test_method = cls(method)
473 if self.filter_callback(file_name, cls.__name__, method):
474 self.suite_name = file_name + cls.__name__
475 if self.suite_name not in self.suites:
476 self.suites[self.suite_name] = unittest.TestSuite()
477 self.suites[self.suite_name].addTest(test_method)
478
479 else:
480 self.filtered.addTest(test_method)
Klement Sekerafcbf4442017-08-17 07:38:42 +0200481
482
juraj.linkes184870a2018-07-16 14:22:01 +0200483test_option = "TEST"
484
485
486def parse_test_option():
487 f = os.getenv(test_option, None)
488 filter_file_name = None
489 filter_class_name = None
490 filter_func_name = None
491 if f:
492 if '.' in f:
493 parts = f.split('.')
494 if len(parts) > 3:
495 raise Exception("Unrecognized %s option: %s" %
496 (test_option, f))
497 if len(parts) > 2:
498 if parts[2] not in ('*', ''):
499 filter_func_name = parts[2]
500 if parts[1] not in ('*', ''):
501 filter_class_name = parts[1]
502 if parts[0] not in ('*', ''):
503 if parts[0].startswith('test_'):
504 filter_file_name = parts[0]
505 else:
506 filter_file_name = 'test_%s' % parts[0]
507 else:
508 if f.startswith('test_'):
509 filter_file_name = f
510 else:
511 filter_file_name = 'test_%s' % f
512 if filter_file_name:
513 filter_file_name = '%s.py' % filter_file_name
514 return filter_file_name, filter_class_name, filter_func_name
515
516
517def filter_tests(tests, filter_cb):
518 result = unittest.suite.TestSuite()
519 for t in tests:
520 if isinstance(t, unittest.suite.TestSuite):
521 # this is a bunch of tests, recursively filter...
522 x = filter_tests(t, filter_cb)
523 if x.countTestCases() > 0:
524 result.addTest(x)
525 elif isinstance(t, unittest.TestCase):
526 # this is a single test
527 parts = t.id().split('.')
528 # t.id() for common cases like this:
529 # test_classifier.TestClassifier.test_acl_ip
530 # apply filtering only if it is so
531 if len(parts) == 3:
532 if not filter_cb(parts[0], parts[1], parts[2]):
533 continue
534 result.addTest(t)
535 else:
536 # unexpected object, don't touch it
537 result.addTest(t)
538 return result
539
540
541class FilterByTestOption:
542 def __init__(self, filter_file_name, filter_class_name, filter_func_name):
543 self.filter_file_name = filter_file_name
544 self.filter_class_name = filter_class_name
545 self.filter_func_name = filter_func_name
546
547 def __call__(self, file_name, class_name, func_name):
Andrew Yourtchenkod760f792018-10-03 11:38:31 +0200548 if self.filter_file_name:
549 fn_match = fnmatch.fnmatch(file_name, self.filter_file_name)
550 if not fn_match:
551 return False
juraj.linkes184870a2018-07-16 14:22:01 +0200552 if self.filter_class_name and class_name != self.filter_class_name:
553 return False
554 if self.filter_func_name and func_name != self.filter_func_name:
555 return False
556 return True
557
558
559class FilterByClassList:
juraj.linkes721872e2018-09-05 18:13:45 +0200560 def __init__(self, classes_with_filenames):
561 self.classes_with_filenames = classes_with_filenames
Klement Sekeradf2b9802017-10-05 10:26:03 +0200562
563 def __call__(self, file_name, class_name, func_name):
juraj.linkes721872e2018-09-05 18:13:45 +0200564 return '.'.join([file_name, class_name]) in self.classes_with_filenames
Klement Sekeradf2b9802017-10-05 10:26:03 +0200565
566
567def suite_from_failed(suite, failed):
juraj.linkes721872e2018-09-05 18:13:45 +0200568 failed = {x.rsplit('.', 1)[0] for x in failed}
juraj.linkes184870a2018-07-16 14:22:01 +0200569 filter_cb = FilterByClassList(failed)
570 suite = filter_tests(suite, filter_cb)
Klement Sekera4c5422e2018-06-22 13:19:45 +0200571 return suite
Klement Sekeradf2b9802017-10-05 10:26:03 +0200572
573
juraj.linkescae64f82018-09-19 15:01:47 +0200574class AllResults(dict):
juraj.linkes184870a2018-07-16 14:22:01 +0200575 def __init__(self):
juraj.linkescae64f82018-09-19 15:01:47 +0200576 super(AllResults, self).__init__()
juraj.linkes184870a2018-07-16 14:22:01 +0200577 self.all_testcases = 0
juraj.linkescae64f82018-09-19 15:01:47 +0200578 self.results_per_suite = []
579 self[PASS] = 0
580 self[FAIL] = 0
581 self[ERROR] = 0
582 self[SKIP] = 0
583 self[TEST_RUN] = 0
juraj.linkes184870a2018-07-16 14:22:01 +0200584 self.rerun = []
juraj.linkescae64f82018-09-19 15:01:47 +0200585 self.testsuites_no_tests_run = []
Klement Sekera909a6a12017-08-08 04:33:53 +0200586
juraj.linkescae64f82018-09-19 15:01:47 +0200587 def add_results(self, result):
588 self.results_per_suite.append(result)
589 result_types = [PASS, FAIL, ERROR, SKIP, TEST_RUN]
590 for result_type in result_types:
591 self[result_type] += len(result[result_type])
Klement Sekera05742262018-03-14 18:14:49 +0100592
juraj.linkescae64f82018-09-19 15:01:47 +0200593 def add_result(self, result):
juraj.linkes184870a2018-07-16 14:22:01 +0200594 retval = 0
juraj.linkescae64f82018-09-19 15:01:47 +0200595 self.all_testcases += result.testcase_suite.countTestCases()
juraj.linkes40dd73b2018-09-21 13:55:16 +0200596 self.add_results(result)
juraj.linkes184870a2018-07-16 14:22:01 +0200597
juraj.linkes40dd73b2018-09-21 13:55:16 +0200598 if result.no_tests_run():
juraj.linkescae64f82018-09-19 15:01:47 +0200599 self.testsuites_no_tests_run.append(result.testcase_suite)
juraj.linkes40dd73b2018-09-21 13:55:16 +0200600 if result.crashed:
601 retval = -1
602 else:
603 retval = 1
604 elif not result.was_successful():
605 retval = 1
juraj.linkes184870a2018-07-16 14:22:01 +0200606
juraj.linkes184870a2018-07-16 14:22:01 +0200607 if retval != 0:
juraj.linkesabec0122018-11-16 17:28:56 +0100608 self.rerun.append(result.testcase_suite)
juraj.linkes184870a2018-07-16 14:22:01 +0200609
610 return retval
611
612 def print_results(self):
613 print('')
614 print(double_line_delim)
615 print('TEST RESULTS:')
juraj.linkescae64f82018-09-19 15:01:47 +0200616 print(' Scheduled tests: {}'.format(self.all_testcases))
617 print(' Executed tests: {}'.format(self[TEST_RUN]))
618 print(' Passed tests: {}'.format(
619 colorize(str(self[PASS]), GREEN)))
620 if self[SKIP] > 0:
621 print(' Skipped tests: {}'.format(
622 colorize(str(self[SKIP]), YELLOW)))
623 if self.not_executed > 0:
624 print(' Not Executed tests: {}'.format(
625 colorize(str(self.not_executed), RED)))
626 if self[FAIL] > 0:
627 print(' Failures: {}'.format(
628 colorize(str(self[FAIL]), RED)))
629 if self[ERROR] > 0:
630 print(' Errors: {}'.format(
631 colorize(str(self[ERROR]), RED)))
juraj.linkes184870a2018-07-16 14:22:01 +0200632
633 if self.all_failed > 0:
juraj.linkes40dd73b2018-09-21 13:55:16 +0200634 print('FAILURES AND ERRORS IN TESTS:')
juraj.linkescae64f82018-09-19 15:01:47 +0200635 for result in self.results_per_suite:
636 failed_testcase_ids = result[FAIL]
637 errored_testcase_ids = result[ERROR]
638 old_testcase_name = None
Naveen Joy2cbf2fb2019-03-06 10:41:06 -0800639 if failed_testcase_ids or errored_testcase_ids:
juraj.linkescae64f82018-09-19 15:01:47 +0200640 for failed_test_id in failed_testcase_ids:
641 new_testcase_name, test_name = \
642 result.get_testcase_names(failed_test_id)
643 if new_testcase_name != old_testcase_name:
644 print(' Testcase name: {}'.format(
645 colorize(new_testcase_name, RED)))
646 old_testcase_name = new_testcase_name
Klement Sekera33177d62018-11-30 14:17:20 +0100647 print(' FAILURE: {} [{}]'.format(
648 colorize(test_name, RED), failed_test_id))
juraj.linkescae64f82018-09-19 15:01:47 +0200649 for failed_test_id in errored_testcase_ids:
650 new_testcase_name, test_name = \
651 result.get_testcase_names(failed_test_id)
652 if new_testcase_name != old_testcase_name:
653 print(' Testcase name: {}'.format(
654 colorize(new_testcase_name, RED)))
655 old_testcase_name = new_testcase_name
Klement Sekera33177d62018-11-30 14:17:20 +0100656 print(' ERROR: {} [{}]'.format(
657 colorize(test_name, RED), failed_test_id))
Naveen Joy2cbf2fb2019-03-06 10:41:06 -0800658 if self.testsuites_no_tests_run:
juraj.linkescae64f82018-09-19 15:01:47 +0200659 print('TESTCASES WHERE NO TESTS WERE SUCCESSFULLY EXECUTED:')
juraj.linkes40dd73b2018-09-21 13:55:16 +0200660 tc_classes = set()
juraj.linkescae64f82018-09-19 15:01:47 +0200661 for testsuite in self.testsuites_no_tests_run:
662 for testcase in testsuite:
663 tc_classes.add(get_testcase_doc_name(testcase))
664 for tc_class in tc_classes:
665 print(' {}'.format(colorize(tc_class, RED)))
juraj.linkes184870a2018-07-16 14:22:01 +0200666
667 print(double_line_delim)
668 print('')
669
670 @property
juraj.linkescae64f82018-09-19 15:01:47 +0200671 def not_executed(self):
672 return self.all_testcases - self[TEST_RUN]
673
674 @property
juraj.linkes184870a2018-07-16 14:22:01 +0200675 def all_failed(self):
juraj.linkescae64f82018-09-19 15:01:47 +0200676 return self[FAIL] + self[ERROR]
juraj.linkes184870a2018-07-16 14:22:01 +0200677
678
679def parse_results(results):
680 """
juraj.linkescae64f82018-09-19 15:01:47 +0200681 Prints the number of scheduled, executed, not executed, passed, failed,
682 errored and skipped tests and details about failed and errored tests.
juraj.linkes184870a2018-07-16 14:22:01 +0200683
juraj.linkescae64f82018-09-19 15:01:47 +0200684 Also returns all suites where any test failed.
juraj.linkes184870a2018-07-16 14:22:01 +0200685
686 :param results:
687 :return:
688 """
689
juraj.linkescae64f82018-09-19 15:01:47 +0200690 results_per_suite = AllResults()
juraj.linkes184870a2018-07-16 14:22:01 +0200691 crashed = False
692 failed = False
juraj.linkescae64f82018-09-19 15:01:47 +0200693 for result in results:
694 result_code = results_per_suite.add_result(result)
juraj.linkes184870a2018-07-16 14:22:01 +0200695 if result_code == 1:
696 failed = True
697 elif result_code == -1:
698 crashed = True
699
700 results_per_suite.print_results()
701
702 if crashed:
703 return_code = -1
704 elif failed:
705 return_code = 1
706 else:
707 return_code = 0
708 return return_code, results_per_suite.rerun
709
710
711def parse_digit_env(env_var, default):
712 value = os.getenv(env_var, default)
713 if value != default:
714 if value.isdigit():
715 value = int(value)
716 else:
717 print('WARNING: unsupported value "%s" for env var "%s",'
718 'defaulting to %s' % (value, env_var, default))
719 value = default
720 return value
Klement Sekera3f6ff192017-08-11 06:56:05 +0200721
722
723if __name__ == '__main__':
724
juraj.linkes184870a2018-07-16 14:22:01 +0200725 verbose = parse_digit_env("V", 0)
Klement Sekera3f6ff192017-08-11 06:56:05 +0200726
juraj.linkes184870a2018-07-16 14:22:01 +0200727 test_timeout = parse_digit_env("TIMEOUT", 600) # default = 10 minutes
Klement Sekera3f6ff192017-08-11 06:56:05 +0200728
juraj.linkes184870a2018-07-16 14:22:01 +0200729 retries = parse_digit_env("RETRIES", 0)
Klement Sekera3f6ff192017-08-11 06:56:05 +0200730
juraj.linkes184870a2018-07-16 14:22:01 +0200731 debug = os.getenv("DEBUG", "n").lower() in ["gdb", "gdbserver"]
732
juraj.linkes40dd73b2018-09-21 13:55:16 +0200733 debug_core = os.getenv("DEBUG", "").lower() == "core"
Paul Vinciguerra0cbc71d2019-07-03 08:38:38 -0400734 compress_core = framework.BoolEnvironmentVariable("CORE_COMPRESS")
juraj.linkes40dd73b2018-09-21 13:55:16 +0200735
Paul Vinciguerra0cbc71d2019-07-03 08:38:38 -0400736 step = framework.BoolEnvironmentVariable("STEP")
737 force_foreground = framework.BoolEnvironmentVariable("FORCE_FOREGROUND")
juraj.linkesb5ef26d2019-07-03 10:42:40 +0200738
739 run_interactive = debug or step or force_foreground
juraj.linkes184870a2018-07-16 14:22:01 +0200740
Paul Vinciguerra025cd9c2019-07-08 14:14:22 -0400741 try:
742 num_cpus = len(os.sched_getaffinity(0))
743 except AttributeError:
744 num_cpus = multiprocessing.cpu_count()
745 shm_free = psutil.disk_usage('/dev/shm').free
746
747 print('OS reports %s available cpu(s). Free shm: %s' % (
748 num_cpus, "{:,}MB".format(shm_free / (1024 * 1024))))
749
juraj.linkes184870a2018-07-16 14:22:01 +0200750 test_jobs = os.getenv("TEST_JOBS", "1").lower() # default = 1 process
751 if test_jobs == 'auto':
752 if run_interactive:
753 concurrent_tests = 1
754 print('Interactive mode required, running on one core')
755 else:
juraj.linkes184870a2018-07-16 14:22:01 +0200756 shm_max_processes = 1
757 if shm_free < min_req_shm:
758 raise Exception('Not enough free space in /dev/shm. Required '
759 'free space is at least %sM.'
760 % (min_req_shm >> 20))
761 else:
762 extra_shm = shm_free - min_req_shm
Ole Troan1556b3a2019-10-21 19:52:06 +0200763 shm_max_processes += extra_shm // shm_per_process
Klement Sekera6c7bda92018-09-18 14:32:27 +0200764 concurrent_tests = min(cpu_count(), shm_max_processes)
juraj.linkes184870a2018-07-16 14:22:01 +0200765 print('Found enough resources to run tests with %s cores'
766 % concurrent_tests)
767 elif test_jobs.isdigit():
768 concurrent_tests = int(test_jobs)
Paul Vinciguerra025cd9c2019-07-08 14:14:22 -0400769 print("Running on %s core(s) as set by 'TEST_JOBS'." %
770 concurrent_tests)
juraj.linkes184870a2018-07-16 14:22:01 +0200771 else:
772 concurrent_tests = 1
Paul Vinciguerra025cd9c2019-07-08 14:14:22 -0400773 print('Running on one core.')
juraj.linkes184870a2018-07-16 14:22:01 +0200774
775 if run_interactive and concurrent_tests > 1:
776 raise NotImplementedError(
juraj.linkes40dd73b2018-09-21 13:55:16 +0200777 'Running tests interactively (DEBUG is gdb or gdbserver or STEP '
778 'is set) in parallel (TEST_JOBS is more than 1) is not supported')
Klement Sekera13a83ef2018-03-21 12:35:51 +0100779
Klement Sekera3f6ff192017-08-11 06:56:05 +0200780 parser = argparse.ArgumentParser(description="VPP unit tests")
juraj.linkes184870a2018-07-16 14:22:01 +0200781 parser.add_argument("-f", "--failfast", action='store_true',
Klement Sekera3f6ff192017-08-11 06:56:05 +0200782 help="fast failure flag")
783 parser.add_argument("-d", "--dir", action='append', type=str,
784 help="directory containing test files "
785 "(may be specified multiple times)")
786 args = parser.parse_args()
juraj.linkes184870a2018-07-16 14:22:01 +0200787 failfast = args.failfast
788 descriptions = True
Klement Sekera3f6ff192017-08-11 06:56:05 +0200789
juraj.linkes184870a2018-07-16 14:22:01 +0200790 print("Running tests using custom test runner") # debug message
791 filter_file, filter_class, filter_func = parse_test_option()
792
793 print("Active filters: file=%s, class=%s, function=%s" % (
794 filter_file, filter_class, filter_func))
795
796 filter_cb = FilterByTestOption(filter_file, filter_class, filter_func)
797
Klement Sekerab8c72a42018-11-08 11:21:39 +0100798 ignore_path = os.getenv("VENV_PATH", None)
juraj.linkes184870a2018-07-16 14:22:01 +0200799 cb = SplitToSuitesCallback(filter_cb)
Klement Sekera3f6ff192017-08-11 06:56:05 +0200800 for d in args.dir:
Klement Sekeradf2b9802017-10-05 10:26:03 +0200801 print("Adding tests from directory tree %s" % d)
Klement Sekerab8c72a42018-11-08 11:21:39 +0100802 discover_tests(d, cb, ignore_path)
Klement Sekera3f6ff192017-08-11 06:56:05 +0200803
juraj.linkes184870a2018-07-16 14:22:01 +0200804 # suites are not hashable, need to use list
805 suites = []
806 tests_amount = 0
807 for testcase_suite in cb.suites.values():
808 tests_amount += testcase_suite.countTestCases()
809 suites.append(testcase_suite)
Klement Sekerabbfa5fd2018-06-27 13:54:32 +0200810
juraj.linkes184870a2018-07-16 14:22:01 +0200811 print("%s out of %s tests match specified filters" % (
812 tests_amount, tests_amount + cb.filtered.countTestCases()))
813
Paul Vinciguerradefde0f2018-12-06 07:46:13 -0800814 if not running_extended_tests:
juraj.linkes184870a2018-07-16 14:22:01 +0200815 print("Not running extended tests (some tests will be skipped)")
816
Klement Sekeradf2b9802017-10-05 10:26:03 +0200817 attempts = retries + 1
818 if attempts > 1:
819 print("Perform %s attempts to pass the suite..." % attempts)
juraj.linkes184870a2018-07-16 14:22:01 +0200820
Naveen Joy2cbf2fb2019-03-06 10:41:06 -0800821 if run_interactive and suites:
juraj.linkes184870a2018-07-16 14:22:01 +0200822 # don't fork if requiring interactive terminal
juraj.linkesb5ef26d2019-07-03 10:42:40 +0200823 print('Running tests in foreground in the current process')
juraj.linkes46e8e912019-01-10 12:13:07 +0100824 full_suite = unittest.TestSuite()
825 map(full_suite.addTests, suites)
juraj.linkesabec0122018-11-16 17:28:56 +0100826 result = VppTestRunner(verbosity=verbose,
827 failfast=failfast,
juraj.linkes46e8e912019-01-10 12:13:07 +0100828 print_summary=True).run(full_suite)
juraj.linkes40dd73b2018-09-21 13:55:16 +0200829 was_successful = result.wasSuccessful()
830 if not was_successful:
831 for test_case_info in result.failed_test_cases_info:
832 handle_failed_suite(test_case_info.logger,
833 test_case_info.tempdir,
834 test_case_info.vpp_pid)
Klement Sekeraf40ee3a2019-05-06 19:11:25 +0200835 if test_case_info in result.core_crash_test_cases_info:
juraj.linkes40dd73b2018-09-21 13:55:16 +0200836 check_and_handle_core(test_case_info.vpp_bin_path,
837 test_case_info.tempdir,
838 test_case_info.core_crash_test)
839
840 sys.exit(not was_successful)
Klement Sekera13a83ef2018-03-21 12:35:51 +0100841 else:
juraj.linkesb5ef26d2019-07-03 10:42:40 +0200842 print('Running each VPPTestCase in a separate background process'
843 ' with {} parallel process(es)'.format(concurrent_tests))
juraj.linkes184870a2018-07-16 14:22:01 +0200844 exit_code = 0
Naveen Joy2cbf2fb2019-03-06 10:41:06 -0800845 while suites and attempts > 0:
juraj.linkes184870a2018-07-16 14:22:01 +0200846 results = run_forked(suites)
847 exit_code, suites = parse_results(results)
848 attempts -= 1
849 if exit_code == 0:
850 print('Test run was successful')
851 else:
852 print('%s attempt(s) left.' % attempts)
853 sys.exit(exit_code)