1d201506cSStephen Warren# Copyright (c) 2015 Stephen Warren 2d201506cSStephen Warren# Copyright (c) 2015-2016, NVIDIA CORPORATION. All rights reserved. 3d201506cSStephen Warren# 4d201506cSStephen Warren# SPDX-License-Identifier: GPL-2.0 5d201506cSStephen Warren 6d201506cSStephen Warren# Implementation of pytest run-time hook functions. These are invoked by 7d201506cSStephen Warren# pytest at certain points during operation, e.g. startup, for each executed 8d201506cSStephen Warren# test, at shutdown etc. These hooks perform functions such as: 9d201506cSStephen Warren# - Parsing custom command-line options. 10d201506cSStephen Warren# - Pullilng in user-specified board configuration. 11d201506cSStephen Warren# - Creating the U-Boot console test fixture. 12d201506cSStephen Warren# - Creating the HTML log file. 13d201506cSStephen Warren# - Monitoring each test's results. 14d201506cSStephen Warren# - Implementing custom pytest markers. 15d201506cSStephen Warren 16d201506cSStephen Warrenimport atexit 17d201506cSStephen Warrenimport errno 18d201506cSStephen Warrenimport os 19d201506cSStephen Warrenimport os.path 20d201506cSStephen Warrenimport pexpect 21d201506cSStephen Warrenimport pytest 22d201506cSStephen Warrenfrom _pytest.runner import runtestprotocol 23d201506cSStephen Warrenimport ConfigParser 24d201506cSStephen Warrenimport StringIO 25d201506cSStephen Warrenimport sys 26d201506cSStephen Warren 27d201506cSStephen Warren# Globals: The HTML log file, and the connection to the U-Boot console. 28d201506cSStephen Warrenlog = None 29d201506cSStephen Warrenconsole = None 30d201506cSStephen Warren 31d201506cSStephen Warrendef mkdir_p(path): 32e8debf39SStephen Warren """Create a directory path. 33d201506cSStephen Warren 34d201506cSStephen Warren This includes creating any intermediate/parent directories. Any errors 35d201506cSStephen Warren caused due to already extant directories are ignored. 36d201506cSStephen Warren 37d201506cSStephen Warren Args: 38d201506cSStephen Warren path: The directory path to create. 39d201506cSStephen Warren 40d201506cSStephen Warren Returns: 41d201506cSStephen Warren Nothing. 42e8debf39SStephen Warren """ 43d201506cSStephen Warren 44d201506cSStephen Warren try: 45d201506cSStephen Warren os.makedirs(path) 46d201506cSStephen Warren except OSError as exc: 47d201506cSStephen Warren if exc.errno == errno.EEXIST and os.path.isdir(path): 48d201506cSStephen Warren pass 49d201506cSStephen Warren else: 50d201506cSStephen Warren raise 51d201506cSStephen Warren 52d201506cSStephen Warrendef pytest_addoption(parser): 53e8debf39SStephen Warren """pytest hook: Add custom command-line options to the cmdline parser. 54d201506cSStephen Warren 55d201506cSStephen Warren Args: 56d201506cSStephen Warren parser: The pytest command-line parser. 57d201506cSStephen Warren 58d201506cSStephen Warren Returns: 59d201506cSStephen Warren Nothing. 60e8debf39SStephen Warren """ 61d201506cSStephen Warren 62d201506cSStephen Warren parser.addoption('--build-dir', default=None, 63d201506cSStephen Warren help='U-Boot build directory (O=)') 64d201506cSStephen Warren parser.addoption('--result-dir', default=None, 65d201506cSStephen Warren help='U-Boot test result/tmp directory') 66d201506cSStephen Warren parser.addoption('--persistent-data-dir', default=None, 67d201506cSStephen Warren help='U-Boot test persistent generated data directory') 68d201506cSStephen Warren parser.addoption('--board-type', '--bd', '-B', default='sandbox', 69d201506cSStephen Warren help='U-Boot board type') 70d201506cSStephen Warren parser.addoption('--board-identity', '--id', default='na', 71d201506cSStephen Warren help='U-Boot board identity/instance') 72d201506cSStephen Warren parser.addoption('--build', default=False, action='store_true', 73d201506cSStephen Warren help='Compile U-Boot before running tests') 7489ab8410SStephen Warren parser.addoption('--gdbserver', default=None, 7589ab8410SStephen Warren help='Run sandbox under gdbserver. The argument is the channel '+ 7689ab8410SStephen Warren 'over which gdbserver should communicate, e.g. localhost:1234') 77d201506cSStephen Warren 78d201506cSStephen Warrendef pytest_configure(config): 79e8debf39SStephen Warren """pytest hook: Perform custom initialization at startup time. 80d201506cSStephen Warren 81d201506cSStephen Warren Args: 82d201506cSStephen Warren config: The pytest configuration. 83d201506cSStephen Warren 84d201506cSStephen Warren Returns: 85d201506cSStephen Warren Nothing. 86e8debf39SStephen Warren """ 87d201506cSStephen Warren 88d201506cSStephen Warren global log 89d201506cSStephen Warren global console 90d201506cSStephen Warren global ubconfig 91d201506cSStephen Warren 92d201506cSStephen Warren test_py_dir = os.path.dirname(os.path.abspath(__file__)) 93d201506cSStephen Warren source_dir = os.path.dirname(os.path.dirname(test_py_dir)) 94d201506cSStephen Warren 95d201506cSStephen Warren board_type = config.getoption('board_type') 96d201506cSStephen Warren board_type_filename = board_type.replace('-', '_') 97d201506cSStephen Warren 98d201506cSStephen Warren board_identity = config.getoption('board_identity') 99d201506cSStephen Warren board_identity_filename = board_identity.replace('-', '_') 100d201506cSStephen Warren 101d201506cSStephen Warren build_dir = config.getoption('build_dir') 102d201506cSStephen Warren if not build_dir: 103d201506cSStephen Warren build_dir = source_dir + '/build-' + board_type 104d201506cSStephen Warren mkdir_p(build_dir) 105d201506cSStephen Warren 106d201506cSStephen Warren result_dir = config.getoption('result_dir') 107d201506cSStephen Warren if not result_dir: 108d201506cSStephen Warren result_dir = build_dir 109d201506cSStephen Warren mkdir_p(result_dir) 110d201506cSStephen Warren 111d201506cSStephen Warren persistent_data_dir = config.getoption('persistent_data_dir') 112d201506cSStephen Warren if not persistent_data_dir: 113d201506cSStephen Warren persistent_data_dir = build_dir + '/persistent-data' 114d201506cSStephen Warren mkdir_p(persistent_data_dir) 115d201506cSStephen Warren 11689ab8410SStephen Warren gdbserver = config.getoption('gdbserver') 11789ab8410SStephen Warren if gdbserver and board_type != 'sandbox': 11889ab8410SStephen Warren raise Exception('--gdbserver only supported with sandbox') 11989ab8410SStephen Warren 120d201506cSStephen Warren import multiplexed_log 121d201506cSStephen Warren log = multiplexed_log.Logfile(result_dir + '/test-log.html') 122d201506cSStephen Warren 123d201506cSStephen Warren if config.getoption('build'): 124d201506cSStephen Warren if build_dir != source_dir: 125d201506cSStephen Warren o_opt = 'O=%s' % build_dir 126d201506cSStephen Warren else: 127d201506cSStephen Warren o_opt = '' 128d201506cSStephen Warren cmds = ( 129d201506cSStephen Warren ['make', o_opt, '-s', board_type + '_defconfig'], 130d201506cSStephen Warren ['make', o_opt, '-s', '-j8'], 131d201506cSStephen Warren ) 132*83357fd5SStephen Warren with log.section('make'): 133d201506cSStephen Warren runner = log.get_runner('make', sys.stdout) 134d201506cSStephen Warren for cmd in cmds: 135d201506cSStephen Warren runner.run(cmd, cwd=source_dir) 136d201506cSStephen Warren runner.close() 137*83357fd5SStephen Warren log.status_pass('OK') 138d201506cSStephen Warren 139d201506cSStephen Warren class ArbitraryAttributeContainer(object): 140d201506cSStephen Warren pass 141d201506cSStephen Warren 142d201506cSStephen Warren ubconfig = ArbitraryAttributeContainer() 143d201506cSStephen Warren ubconfig.brd = dict() 144d201506cSStephen Warren ubconfig.env = dict() 145d201506cSStephen Warren 146d201506cSStephen Warren modules = [ 147d201506cSStephen Warren (ubconfig.brd, 'u_boot_board_' + board_type_filename), 148d201506cSStephen Warren (ubconfig.env, 'u_boot_boardenv_' + board_type_filename), 149d201506cSStephen Warren (ubconfig.env, 'u_boot_boardenv_' + board_type_filename + '_' + 150d201506cSStephen Warren board_identity_filename), 151d201506cSStephen Warren ] 152d201506cSStephen Warren for (dict_to_fill, module_name) in modules: 153d201506cSStephen Warren try: 154d201506cSStephen Warren module = __import__(module_name) 155d201506cSStephen Warren except ImportError: 156d201506cSStephen Warren continue 157d201506cSStephen Warren dict_to_fill.update(module.__dict__) 158d201506cSStephen Warren 159d201506cSStephen Warren ubconfig.buildconfig = dict() 160d201506cSStephen Warren 161d201506cSStephen Warren for conf_file in ('.config', 'include/autoconf.mk'): 162d201506cSStephen Warren dot_config = build_dir + '/' + conf_file 163d201506cSStephen Warren if not os.path.exists(dot_config): 164d201506cSStephen Warren raise Exception(conf_file + ' does not exist; ' + 165d201506cSStephen Warren 'try passing --build option?') 166d201506cSStephen Warren 167d201506cSStephen Warren with open(dot_config, 'rt') as f: 168d201506cSStephen Warren ini_str = '[root]\n' + f.read() 169d201506cSStephen Warren ini_sio = StringIO.StringIO(ini_str) 170d201506cSStephen Warren parser = ConfigParser.RawConfigParser() 171d201506cSStephen Warren parser.readfp(ini_sio) 172d201506cSStephen Warren ubconfig.buildconfig.update(parser.items('root')) 173d201506cSStephen Warren 174d201506cSStephen Warren ubconfig.test_py_dir = test_py_dir 175d201506cSStephen Warren ubconfig.source_dir = source_dir 176d201506cSStephen Warren ubconfig.build_dir = build_dir 177d201506cSStephen Warren ubconfig.result_dir = result_dir 178d201506cSStephen Warren ubconfig.persistent_data_dir = persistent_data_dir 179d201506cSStephen Warren ubconfig.board_type = board_type 180d201506cSStephen Warren ubconfig.board_identity = board_identity 18189ab8410SStephen Warren ubconfig.gdbserver = gdbserver 182d201506cSStephen Warren 183d201506cSStephen Warren env_vars = ( 184d201506cSStephen Warren 'board_type', 185d201506cSStephen Warren 'board_identity', 186d201506cSStephen Warren 'source_dir', 187d201506cSStephen Warren 'test_py_dir', 188d201506cSStephen Warren 'build_dir', 189d201506cSStephen Warren 'result_dir', 190d201506cSStephen Warren 'persistent_data_dir', 191d201506cSStephen Warren ) 192d201506cSStephen Warren for v in env_vars: 193d201506cSStephen Warren os.environ['U_BOOT_' + v.upper()] = getattr(ubconfig, v) 194d201506cSStephen Warren 195d201506cSStephen Warren if board_type == 'sandbox': 196d201506cSStephen Warren import u_boot_console_sandbox 197d201506cSStephen Warren console = u_boot_console_sandbox.ConsoleSandbox(log, ubconfig) 198d201506cSStephen Warren else: 199d201506cSStephen Warren import u_boot_console_exec_attach 200d201506cSStephen Warren console = u_boot_console_exec_attach.ConsoleExecAttach(log, ubconfig) 201d201506cSStephen Warren 202d201506cSStephen Warrendef pytest_generate_tests(metafunc): 203e8debf39SStephen Warren """pytest hook: parameterize test functions based on custom rules. 204d201506cSStephen Warren 205d201506cSStephen Warren If a test function takes parameter(s) (fixture names) of the form brd__xxx 206d201506cSStephen Warren or env__xxx, the brd and env configuration dictionaries are consulted to 207d201506cSStephen Warren find the list of values to use for those parameters, and the test is 208d201506cSStephen Warren parametrized so that it runs once for each combination of values. 209d201506cSStephen Warren 210d201506cSStephen Warren Args: 211d201506cSStephen Warren metafunc: The pytest test function. 212d201506cSStephen Warren 213d201506cSStephen Warren Returns: 214d201506cSStephen Warren Nothing. 215e8debf39SStephen Warren """ 216d201506cSStephen Warren 217d201506cSStephen Warren subconfigs = { 218d201506cSStephen Warren 'brd': console.config.brd, 219d201506cSStephen Warren 'env': console.config.env, 220d201506cSStephen Warren } 221d201506cSStephen Warren for fn in metafunc.fixturenames: 222d201506cSStephen Warren parts = fn.split('__') 223d201506cSStephen Warren if len(parts) < 2: 224d201506cSStephen Warren continue 225d201506cSStephen Warren if parts[0] not in subconfigs: 226d201506cSStephen Warren continue 227d201506cSStephen Warren subconfig = subconfigs[parts[0]] 228d201506cSStephen Warren vals = [] 229d201506cSStephen Warren val = subconfig.get(fn, []) 230d201506cSStephen Warren # If that exact name is a key in the data source: 231d201506cSStephen Warren if val: 232d201506cSStephen Warren # ... use the dict value as a single parameter value. 233d201506cSStephen Warren vals = (val, ) 234d201506cSStephen Warren else: 235d201506cSStephen Warren # ... otherwise, see if there's a key that contains a list of 236d201506cSStephen Warren # values to use instead. 237d201506cSStephen Warren vals = subconfig.get(fn + 's', []) 238d20e5e97SStephen Warren def fixture_id(index, val): 239d20e5e97SStephen Warren try: 240d20e5e97SStephen Warren return val["fixture_id"] 241d20e5e97SStephen Warren except: 242d20e5e97SStephen Warren return fn + str(index) 243d20e5e97SStephen Warren ids = [fixture_id(index, val) for (index, val) in enumerate(vals)] 244d20e5e97SStephen Warren metafunc.parametrize(fn, vals, ids=ids) 245d201506cSStephen Warren 246636f38d8SStephen Warren@pytest.fixture(scope='function') 247d201506cSStephen Warrendef u_boot_console(request): 248e8debf39SStephen Warren """Generate the value of a test's u_boot_console fixture. 249d201506cSStephen Warren 250d201506cSStephen Warren Args: 251d201506cSStephen Warren request: The pytest request. 252d201506cSStephen Warren 253d201506cSStephen Warren Returns: 254d201506cSStephen Warren The fixture value. 255e8debf39SStephen Warren """ 256d201506cSStephen Warren 257636f38d8SStephen Warren console.ensure_spawned() 258d201506cSStephen Warren return console 259d201506cSStephen Warren 260*83357fd5SStephen Warrenanchors = {} 261d201506cSStephen Warrentests_not_run = set() 262d201506cSStephen Warrentests_failed = set() 26378b39cc3SStephen Warrentests_xpassed = set() 26478b39cc3SStephen Warrentests_xfailed = set() 265d201506cSStephen Warrentests_skipped = set() 266d201506cSStephen Warrentests_passed = set() 267d201506cSStephen Warren 268d201506cSStephen Warrendef pytest_itemcollected(item): 269e8debf39SStephen Warren """pytest hook: Called once for each test found during collection. 270d201506cSStephen Warren 271d201506cSStephen Warren This enables our custom result analysis code to see the list of all tests 272d201506cSStephen Warren that should eventually be run. 273d201506cSStephen Warren 274d201506cSStephen Warren Args: 275d201506cSStephen Warren item: The item that was collected. 276d201506cSStephen Warren 277d201506cSStephen Warren Returns: 278d201506cSStephen Warren Nothing. 279e8debf39SStephen Warren """ 280d201506cSStephen Warren 281d201506cSStephen Warren tests_not_run.add(item.name) 282d201506cSStephen Warren 283d201506cSStephen Warrendef cleanup(): 284e8debf39SStephen Warren """Clean up all global state. 285d201506cSStephen Warren 286d201506cSStephen Warren Executed (via atexit) once the entire test process is complete. This 287d201506cSStephen Warren includes logging the status of all tests, and the identity of any failed 288d201506cSStephen Warren or skipped tests. 289d201506cSStephen Warren 290d201506cSStephen Warren Args: 291d201506cSStephen Warren None. 292d201506cSStephen Warren 293d201506cSStephen Warren Returns: 294d201506cSStephen Warren Nothing. 295e8debf39SStephen Warren """ 296d201506cSStephen Warren 297d201506cSStephen Warren if console: 298d201506cSStephen Warren console.close() 299d201506cSStephen Warren if log: 300*83357fd5SStephen Warren with log.section('Status Report', 'status_report'): 301d201506cSStephen Warren log.status_pass('%d passed' % len(tests_passed)) 302d201506cSStephen Warren if tests_skipped: 303d201506cSStephen Warren log.status_skipped('%d skipped' % len(tests_skipped)) 304d201506cSStephen Warren for test in tests_skipped: 305*83357fd5SStephen Warren anchor = anchors.get(test, None) 306*83357fd5SStephen Warren log.status_skipped('... ' + test, anchor) 30778b39cc3SStephen Warren if tests_xpassed: 30878b39cc3SStephen Warren log.status_xpass('%d xpass' % len(tests_xpassed)) 30978b39cc3SStephen Warren for test in tests_xpassed: 310*83357fd5SStephen Warren anchor = anchors.get(test, None) 311*83357fd5SStephen Warren log.status_xpass('... ' + test, anchor) 31278b39cc3SStephen Warren if tests_xfailed: 31378b39cc3SStephen Warren log.status_xfail('%d xfail' % len(tests_xfailed)) 31478b39cc3SStephen Warren for test in tests_xfailed: 315*83357fd5SStephen Warren anchor = anchors.get(test, None) 316*83357fd5SStephen Warren log.status_xfail('... ' + test, anchor) 317d201506cSStephen Warren if tests_failed: 318d201506cSStephen Warren log.status_fail('%d failed' % len(tests_failed)) 319d201506cSStephen Warren for test in tests_failed: 320*83357fd5SStephen Warren anchor = anchors.get(test, None) 321*83357fd5SStephen Warren log.status_fail('... ' + test, anchor) 322d201506cSStephen Warren if tests_not_run: 323d201506cSStephen Warren log.status_fail('%d not run' % len(tests_not_run)) 324d201506cSStephen Warren for test in tests_not_run: 325*83357fd5SStephen Warren anchor = anchors.get(test, None) 326*83357fd5SStephen Warren log.status_fail('... ' + test, anchor) 327d201506cSStephen Warren log.close() 328d201506cSStephen Warrenatexit.register(cleanup) 329d201506cSStephen Warren 330d201506cSStephen Warrendef setup_boardspec(item): 331e8debf39SStephen Warren """Process any 'boardspec' marker for a test. 332d201506cSStephen Warren 333d201506cSStephen Warren Such a marker lists the set of board types that a test does/doesn't 334d201506cSStephen Warren support. If tests are being executed on an unsupported board, the test is 335d201506cSStephen Warren marked to be skipped. 336d201506cSStephen Warren 337d201506cSStephen Warren Args: 338d201506cSStephen Warren item: The pytest test item. 339d201506cSStephen Warren 340d201506cSStephen Warren Returns: 341d201506cSStephen Warren Nothing. 342e8debf39SStephen Warren """ 343d201506cSStephen Warren 344d201506cSStephen Warren mark = item.get_marker('boardspec') 345d201506cSStephen Warren if not mark: 346d201506cSStephen Warren return 347d201506cSStephen Warren required_boards = [] 348d201506cSStephen Warren for board in mark.args: 349d201506cSStephen Warren if board.startswith('!'): 350d201506cSStephen Warren if ubconfig.board_type == board[1:]: 351d201506cSStephen Warren pytest.skip('board not supported') 352d201506cSStephen Warren return 353d201506cSStephen Warren else: 354d201506cSStephen Warren required_boards.append(board) 355d201506cSStephen Warren if required_boards and ubconfig.board_type not in required_boards: 356d201506cSStephen Warren pytest.skip('board not supported') 357d201506cSStephen Warren 358d201506cSStephen Warrendef setup_buildconfigspec(item): 359e8debf39SStephen Warren """Process any 'buildconfigspec' marker for a test. 360d201506cSStephen Warren 361d201506cSStephen Warren Such a marker lists some U-Boot configuration feature that the test 362d201506cSStephen Warren requires. If tests are being executed on an U-Boot build that doesn't 363d201506cSStephen Warren have the required feature, the test is marked to be skipped. 364d201506cSStephen Warren 365d201506cSStephen Warren Args: 366d201506cSStephen Warren item: The pytest test item. 367d201506cSStephen Warren 368d201506cSStephen Warren Returns: 369d201506cSStephen Warren Nothing. 370e8debf39SStephen Warren """ 371d201506cSStephen Warren 372d201506cSStephen Warren mark = item.get_marker('buildconfigspec') 373d201506cSStephen Warren if not mark: 374d201506cSStephen Warren return 375d201506cSStephen Warren for option in mark.args: 376d201506cSStephen Warren if not ubconfig.buildconfig.get('config_' + option.lower(), None): 377d201506cSStephen Warren pytest.skip('.config feature not enabled') 378d201506cSStephen Warren 379d201506cSStephen Warrendef pytest_runtest_setup(item): 380e8debf39SStephen Warren """pytest hook: Configure (set up) a test item. 381d201506cSStephen Warren 382d201506cSStephen Warren Called once for each test to perform any custom configuration. This hook 383d201506cSStephen Warren is used to skip the test if certain conditions apply. 384d201506cSStephen Warren 385d201506cSStephen Warren Args: 386d201506cSStephen Warren item: The pytest test item. 387d201506cSStephen Warren 388d201506cSStephen Warren Returns: 389d201506cSStephen Warren Nothing. 390e8debf39SStephen Warren """ 391d201506cSStephen Warren 392*83357fd5SStephen Warren anchors[item.name] = log.start_section(item.name) 393d201506cSStephen Warren setup_boardspec(item) 394d201506cSStephen Warren setup_buildconfigspec(item) 395d201506cSStephen Warren 396d201506cSStephen Warrendef pytest_runtest_protocol(item, nextitem): 397e8debf39SStephen Warren """pytest hook: Called to execute a test. 398d201506cSStephen Warren 399d201506cSStephen Warren This hook wraps the standard pytest runtestprotocol() function in order 400d201506cSStephen Warren to acquire visibility into, and record, each test function's result. 401d201506cSStephen Warren 402d201506cSStephen Warren Args: 403d201506cSStephen Warren item: The pytest test item to execute. 404d201506cSStephen Warren nextitem: The pytest test item that will be executed after this one. 405d201506cSStephen Warren 406d201506cSStephen Warren Returns: 407d201506cSStephen Warren A list of pytest reports (test result data). 408e8debf39SStephen Warren """ 409d201506cSStephen Warren 410d201506cSStephen Warren reports = runtestprotocol(item, nextitem=nextitem) 41178b39cc3SStephen Warren 41278b39cc3SStephen Warren failure_cleanup = False 41378b39cc3SStephen Warren test_list = tests_passed 41478b39cc3SStephen Warren msg = 'OK' 41578b39cc3SStephen Warren msg_log = log.status_pass 416d201506cSStephen Warren for report in reports: 417d201506cSStephen Warren if report.outcome == 'failed': 41878b39cc3SStephen Warren if hasattr(report, 'wasxfail'): 41978b39cc3SStephen Warren test_list = tests_xpassed 42078b39cc3SStephen Warren msg = 'XPASSED' 42178b39cc3SStephen Warren msg_log = log.status_xpass 42278b39cc3SStephen Warren else: 42378b39cc3SStephen Warren failure_cleanup = True 42478b39cc3SStephen Warren test_list = tests_failed 42578b39cc3SStephen Warren msg = 'FAILED:\n' + str(report.longrepr) 42678b39cc3SStephen Warren msg_log = log.status_fail 427d201506cSStephen Warren break 428d201506cSStephen Warren if report.outcome == 'skipped': 42978b39cc3SStephen Warren if hasattr(report, 'wasxfail'): 43078b39cc3SStephen Warren failure_cleanup = True 43178b39cc3SStephen Warren test_list = tests_xfailed 43278b39cc3SStephen Warren msg = 'XFAILED:\n' + str(report.longrepr) 43378b39cc3SStephen Warren msg_log = log.status_xfail 43478b39cc3SStephen Warren break 43578b39cc3SStephen Warren test_list = tests_skipped 43678b39cc3SStephen Warren msg = 'SKIPPED:\n' + str(report.longrepr) 43778b39cc3SStephen Warren msg_log = log.status_skipped 438d201506cSStephen Warren 43978b39cc3SStephen Warren if failure_cleanup: 440c10eb9d3SStephen Warren console.drain_console() 44178b39cc3SStephen Warren 44278b39cc3SStephen Warren test_list.add(item.name) 443d201506cSStephen Warren tests_not_run.remove(item.name) 444d201506cSStephen Warren 445d201506cSStephen Warren try: 44678b39cc3SStephen Warren msg_log(msg) 447d201506cSStephen Warren except: 448d201506cSStephen Warren # If something went wrong with logging, it's better to let the test 449d201506cSStephen Warren # process continue, which may report other exceptions that triggered 450d201506cSStephen Warren # the logging issue (e.g. console.log wasn't created). Hence, just 451d201506cSStephen Warren # squash the exception. If the test setup failed due to e.g. syntax 452d201506cSStephen Warren # error somewhere else, this won't be seen. However, once that issue 453d201506cSStephen Warren # is fixed, if this exception still exists, it will then be logged as 454d201506cSStephen Warren # part of the test's stdout. 455d201506cSStephen Warren import traceback 456d201506cSStephen Warren print 'Exception occurred while logging runtest status:' 457d201506cSStephen Warren traceback.print_exc() 458d201506cSStephen Warren # FIXME: Can we force a test failure here? 459d201506cSStephen Warren 460d201506cSStephen Warren log.end_section(item.name) 461d201506cSStephen Warren 46278b39cc3SStephen Warren if failure_cleanup: 463d201506cSStephen Warren console.cleanup_spawn() 464d201506cSStephen Warren 465d201506cSStephen Warren return reports 466