Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(357)

Side by Side Diff: build/android/pylib/host_driven/python_test_caller.py

Issue 19537004: [Android] Converts host driven tests to common test_dispatcher (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@sharding_refactoring
Patch Set: Converts --official-build into a boolean flag Created 7 years, 4 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
(Empty)
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
8 import logging
9 import sys
10 import time
11 import traceback
12
13 from pylib.base import base_test_result
14 from pylib.instrumentation import test_result
15
16
17 class PythonExceptionTestResult(test_result.InstrumentationTestResult):
18 """Helper class for creating a test result from python exception."""
19
20 def __init__(self, test_name, start_date_ms, exc_info):
21 """Constructs an PythonExceptionTestResult object.
22
23 Args:
24 test_name: name of the test which raised an exception.
25 start_date_ms: the starting time for the test.
26 exc_info: exception info, ostensibly from sys.exc_info().
27 """
28 exc_type, exc_value, exc_traceback = exc_info
29 trace_info = ''.join(traceback.format_exception(exc_type, exc_value,
30 exc_traceback))
31 log_msg = 'Exception:\n' + trace_info
32 duration_ms = (int(time.time()) * 1000) - start_date_ms
33
34 super(PythonExceptionTestResult, self).__init__(
35 'PythonWrapper#' + test_name,
36 base_test_result.ResultType.FAIL,
37 start_date_ms,
38 duration_ms,
39 log=str(exc_type) + ' ' + log_msg)
40
41
42 def CallPythonTest(test, options):
43 """Invokes a test function and translates Python exceptions into test results.
44
45 This method invokes SetUp()/TearDown() on the test. It is intended to be
46 resilient to exceptions in SetUp(), the test itself, and TearDown(). Any
47 Python exception means the test is marked as failed, and the test result will
48 contain information about the exception.
49
50 If SetUp() raises an exception, the test is not run.
51
52 If TearDown() raises an exception, the test is treated as a failure. However,
53 if the test itself raised an exception beforehand, that stack trace will take
54 precedence whether or not TearDown() also raised an exception.
55
56 shard_index is not applicable in single-device scenarios, when test execution
57 is serial rather than parallel. Tests can use this to bring up servers with
58 unique port numbers, for example. See also python_test_sharder.
59
60 Args:
61 test: an object which is ostensibly a subclass of PythonTestBase.
62 options: Options to use for setting up tests.
63
64 Returns:
65 A TestRunResults object which contains any results produced by the test or,
66 in the case of a Python exception, the Python exception info.
67 """
68
69 start_date_ms = int(time.time()) * 1000
70 failed = False
71
72 try:
73 test.SetUp(options)
74 except Exception:
75 failed = True
76 logging.exception(
77 'Caught exception while trying to run SetUp() for test: ' +
78 test.qualified_name)
79 # Tests whose SetUp() method has failed are likely to fail, or at least
80 # yield invalid results.
81 exc_info = sys.exc_info()
82 results = base_test_result.TestRunResults()
83 results.AddResult(PythonExceptionTestResult(
84 test.qualified_name, start_date_ms, exc_info))
85 return results
86
87 try:
88 results = test.Run()
89 except Exception:
90 # Setting this lets TearDown() avoid stomping on our stack trace from Run()
91 # should TearDown() also raise an exception.
92 failed = True
93 logging.exception('Caught exception while trying to run test: ' +
94 test.qualified_name)
95 exc_info = sys.exc_info()
96 results = base_test_result.TestRunResults()
97 results.AddResult(PythonExceptionTestResult(
98 test.qualified_name, start_date_ms, exc_info))
99
100 try:
101 test.TearDown()
102 except Exception:
103 logging.exception(
104 'Caught exception while trying run TearDown() for test: ' +
105 test.qualified_name)
106 if not failed:
107 # Don't stomp the error during the test if TearDown blows up. This is a
108 # trade-off: if the test fails, this will mask any problem with TearDown
109 # until the test is fixed.
110 exc_info = sys.exc_info()
111 results = base_test_result.TestRunResults()
112 results.AddResult(PythonExceptionTestResult(
113 test.qualified_name, start_date_ms, exc_info))
114
115 return results
OLDNEW
« no previous file with comments | « build/android/pylib/host_driven/python_test_base.py ('k') | build/android/pylib/host_driven/python_test_sharder.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698