OLD | NEW |
| (Empty) |
1 #!/usr/bin/python2.4 | |
2 # | |
3 # | |
4 # Copyright 2008, The Android Open Source Project | |
5 # | |
6 # Licensed under the Apache License, Version 2.0 (the "License"); | |
7 # you may not use this file except in compliance with the License. | |
8 # You may obtain a copy of the License at | |
9 # | |
10 # http://www.apache.org/licenses/LICENSE-2.0 | |
11 # | |
12 # Unless required by applicable law or agreed to in writing, software | |
13 # distributed under the License is distributed on an "AS IS" BASIS, | |
14 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
15 # See the License for the specific language governing permissions and | |
16 # limitations under the License. | |
17 | |
18 """Module that assists in parsing the output of "am instrument" commands run on | |
19 the device.""" | |
20 | |
21 import re | |
22 import string | |
23 | |
24 | |
25 def ParseAmInstrumentOutput(result): | |
26 """Given the raw output of an "am instrument" command that targets and | |
27 InstrumentationTestRunner, return structured data. | |
28 | |
29 Args: | |
30 result (string): Raw output of "am instrument" | |
31 | |
32 Return | |
33 (test_results, inst_finished_bundle) | |
34 | |
35 test_results (list of am_output_parser.TestResult) | |
36 inst_finished_bundle (dict): Key/value pairs contained in the bundle that is | |
37 passed into ActivityManager.finishInstrumentation(). Included in this bundle
is the return | |
38 code of the Instrumentation process, any error codes reported by the | |
39 activity manager, and any results explicity added by the instrumentation | |
40 code. | |
41 """ | |
42 | |
43 re_status_code = re.compile(r'INSTRUMENTATION_STATUS_CODE: (?P<status_code>-?\
d)$') | |
44 test_results = [] | |
45 inst_finished_bundle = {} | |
46 | |
47 result_block_string = "" | |
48 for line in result.splitlines(): | |
49 result_block_string += line + '\n' | |
50 | |
51 if "INSTRUMENTATION_STATUS_CODE:" in line: | |
52 test_result = TestResult(result_block_string) | |
53 if test_result.GetStatusCode() == 1: # The test started | |
54 pass | |
55 elif test_result.GetStatusCode() in [0, -1, -2]: | |
56 test_results.append(test_result) | |
57 else: | |
58 pass | |
59 result_block_string = "" | |
60 if "INSTRUMENTATION_CODE:" in line: | |
61 inst_finished_bundle = _ParseInstrumentationFinishedBundle(result_block_st
ring) | |
62 result_block_string = "" | |
63 | |
64 return (test_results, inst_finished_bundle) | |
65 | |
66 | |
67 def _ParseInstrumentationFinishedBundle(result): | |
68 """Given the raw output of "am instrument" returns a dictionary of the | |
69 key/value pairs from the bundle passed into | |
70 ActivityManager.finishInstrumentation(). | |
71 | |
72 Args: | |
73 result (string): Raw output of "am instrument" | |
74 | |
75 Return: | |
76 inst_finished_bundle (dict): Key/value pairs contained in the bundle that is | |
77 passed into ActivityManager.finishInstrumentation(). Included in this bundle
is the return | |
78 code of the Instrumentation process, any error codes reported by the | |
79 activity manager, and any results explicity added by the instrumentation | |
80 code. | |
81 """ | |
82 | |
83 re_result = re.compile(r'INSTRUMENTATION_RESULT: ([^=]+)=(.*)$') | |
84 re_code = re.compile(r'INSTRUMENTATION_CODE: (\-?\d)$') | |
85 result_dict = {} | |
86 key = '' | |
87 val = '' | |
88 last_tag = '' | |
89 | |
90 for line in result.split('\n'): | |
91 line = line.strip(string.whitespace) | |
92 if re_result.match(line): | |
93 last_tag = 'INSTRUMENTATION_RESULT' | |
94 key = re_result.search(line).group(1).strip(string.whitespace) | |
95 if key.startswith('performance.'): | |
96 key = key[len('performance.'):] | |
97 val = re_result.search(line).group(2).strip(string.whitespace) | |
98 try: | |
99 result_dict[key] = float(val) | |
100 except ValueError: | |
101 result_dict[key] = val | |
102 except TypeError: | |
103 result_dict[key] = val | |
104 elif re_code.match(line): | |
105 last_tag = 'INSTRUMENTATION_CODE' | |
106 key = 'code' | |
107 val = re_code.search(line).group(1).strip(string.whitespace) | |
108 result_dict[key] = val | |
109 elif 'INSTRUMENTATION_ABORTED:' in line: | |
110 last_tag = 'INSTRUMENTATION_ABORTED' | |
111 key = 'INSTRUMENTATION_ABORTED' | |
112 val = '' | |
113 result_dict[key] = val | |
114 elif last_tag == 'INSTRUMENTATION_RESULT': | |
115 result_dict[key] += '\n' + line | |
116 | |
117 if not result_dict.has_key('code'): | |
118 result_dict['code'] = '0' | |
119 result_dict['shortMsg'] = "No result returned from instrumentation" | |
120 | |
121 return result_dict | |
122 | |
123 | |
124 class TestResult(object): | |
125 """A class that contains information about a single test result.""" | |
126 | |
127 def __init__(self, result_block_string): | |
128 """ | |
129 Args: | |
130 result_block_string (string): Is a single "block" of output. A single | |
131 "block" would be either a "test started" status report, or a "test | |
132 finished" status report. | |
133 """ | |
134 | |
135 self._test_name = None | |
136 self._status_code = None | |
137 self._failure_reason = None | |
138 self._fields_map = {} | |
139 | |
140 re_status_code = re.search(r'INSTRUMENTATION_STATUS_CODE: ' | |
141 '(?P<status_code>1|0|-1|-2)', result_block_string) | |
142 re_fields = re.compile(r'INSTRUMENTATION_STATUS: ' | |
143 '(?P<key>[\w.]+)=(?P<value>.*?)(?=\nINSTRUMENTATION_STATUS)', re.DOTALL) | |
144 | |
145 for field in re_fields.finditer(result_block_string): | |
146 key, value = (field.group('key').strip(), field.group('value').strip()) | |
147 if key.startswith('performance.'): | |
148 key = key[len('performance.'):] | |
149 self._fields_map[key] = value | |
150 self._fields_map.setdefault('class') | |
151 self._fields_map.setdefault('test') | |
152 | |
153 self._test_name = '%s:%s' % (self._fields_map['class'], | |
154 self._fields_map['test']) | |
155 self._status_code = int(re_status_code.group('status_code')) | |
156 if 'stack' in self._fields_map: | |
157 self._failure_reason = self._fields_map['stack'] | |
158 | |
159 def GetTestName(self): | |
160 return self._test_name | |
161 | |
162 def GetStatusCode(self): | |
163 return self._status_code | |
164 | |
165 def GetFailureReason(self): | |
166 return self._failure_reason | |
167 | |
168 def GetResultFields(self): | |
169 return self._fields_map | |
OLD | NEW |