1# Copyright (c) 2012 The Chromium Authors. All rights reserved. 2# Use of this source code is governed by a BSD-style license that can be 3# found in the LICENSE file. 4 5"""Helper module for calling python-based tests.""" 6 7 8import logging 9import sys 10import time 11 12from test_result import TestResults 13 14 15def CallPythonTest(test, options): 16 """Invokes a test function and translates Python exceptions into test results. 17 18 This method invokes SetUp()/TearDown() on the test. It is intended to be 19 resilient to exceptions in SetUp(), the test itself, and TearDown(). Any 20 Python exception means the test is marked as failed, and the test result will 21 contain information about the exception. 22 23 If SetUp() raises an exception, the test is not run. 24 25 If TearDown() raises an exception, the test is treated as a failure. However, 26 if the test itself raised an exception beforehand, that stack trace will take 27 precedence whether or not TearDown() also raised an exception. 28 29 shard_index is not applicable in single-device scenarios, when test execution 30 is serial rather than parallel. Tests can use this to bring up servers with 31 unique port numbers, for example. See also python_test_sharder. 32 33 Args: 34 test: an object which is ostensibly a subclass of PythonTestBase. 35 options: Options to use for setting up tests. 36 37 Returns: 38 A TestResults object which contains any results produced by the test or, in 39 the case of a Python exception, the Python exception info. 40 """ 41 42 start_date_ms = int(time.time()) * 1000 43 failed = False 44 45 try: 46 test.SetUp(options) 47 except Exception: 48 failed = True 49 logging.exception( 50 'Caught exception while trying to run SetUp() for test: ' + 51 test.qualified_name) 52 # Tests whose SetUp() method has failed are likely to fail, or at least 53 # yield invalid results. 54 exc_info = sys.exc_info() 55 return TestResults.FromPythonException(test.qualified_name, start_date_ms, 56 exc_info) 57 58 try: 59 result = test.Run() 60 except Exception: 61 # Setting this lets TearDown() avoid stomping on our stack trace from Run() 62 # should TearDown() also raise an exception. 63 failed = True 64 logging.exception('Caught exception while trying to run test: ' + 65 test.qualified_name) 66 exc_info = sys.exc_info() 67 result = TestResults.FromPythonException(test.qualified_name, start_date_ms, 68 exc_info) 69 70 try: 71 test.TearDown() 72 except Exception: 73 logging.exception( 74 'Caught exception while trying run TearDown() for test: ' + 75 test.qualified_name) 76 if not failed: 77 # Don't stomp the error during the test if TearDown blows up. This is a 78 # trade-off: if the test fails, this will mask any problem with TearDown 79 # until the test is fixed. 80 exc_info = sys.exc_info() 81 result = TestResults.FromPythonException(test.qualified_name, 82 start_date_ms, exc_info) 83 84 return result 85