| 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 |