OLD | NEW |
| (Empty) |
1 #!/usr/bin/env python | |
2 # | |
3 # Copyright 2008, Google Inc. | |
4 # All rights reserved. | |
5 # | |
6 # Redistribution and use in source and binary forms, with or without | |
7 # modification, are permitted provided that the following conditions are | |
8 # met: | |
9 # | |
10 # * Redistributions of source code must retain the above copyright | |
11 # notice, this list of conditions and the following disclaimer. | |
12 # * Redistributions in binary form must reproduce the above | |
13 # copyright notice, this list of conditions and the following disclaimer | |
14 # in the documentation and/or other materials provided with the | |
15 # distribution. | |
16 # * Neither the name of Google Inc. nor the names of its | |
17 # contributors may be used to endorse or promote products derived from | |
18 # this software without specific prior written permission. | |
19 # | |
20 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS | |
21 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT | |
22 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR | |
23 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT | |
24 # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, | |
25 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT | |
26 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, | |
27 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY | |
28 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | |
29 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | |
30 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | |
31 | |
32 """Tests the text output of Google C++ Testing Framework. | |
33 | |
34 SYNOPSIS | |
35 gtest_output_test.py --build_dir=BUILD/DIR --gengolden | |
36 # where BUILD/DIR contains the built gtest_output_test_ file. | |
37 gtest_output_test.py --gengolden | |
38 gtest_output_test.py | |
39 """ | |
40 | |
41 __author__ = 'wan@google.com (Zhanyong Wan)' | |
42 | |
43 import os | |
44 import re | |
45 import sys | |
46 import gtest_test_utils | |
47 | |
48 | |
49 # The flag for generating the golden file | |
50 GENGOLDEN_FLAG = '--gengolden' | |
51 CATCH_EXCEPTIONS_ENV_VAR_NAME = 'GTEST_CATCH_EXCEPTIONS' | |
52 | |
53 IS_WINDOWS = os.name == 'nt' | |
54 | |
55 # TODO(vladl@google.com): remove the _lin suffix. | |
56 GOLDEN_NAME = 'gtest_output_test_golden_lin.txt' | |
57 | |
58 PROGRAM_PATH = gtest_test_utils.GetTestExecutablePath('gtest_output_test_') | |
59 | |
60 # At least one command we exercise must not have the | |
61 # --gtest_internal_skip_environment_and_ad_hoc_tests flag. | |
62 COMMAND_LIST_TESTS = ({}, [PROGRAM_PATH, '--gtest_list_tests']) | |
63 COMMAND_WITH_COLOR = ({}, [PROGRAM_PATH, '--gtest_color=yes']) | |
64 COMMAND_WITH_TIME = ({}, [PROGRAM_PATH, | |
65 '--gtest_print_time', | |
66 '--gtest_internal_skip_environment_and_ad_hoc_tests', | |
67 '--gtest_filter=FatalFailureTest.*:LoggingTest.*']) | |
68 COMMAND_WITH_DISABLED = ( | |
69 {}, [PROGRAM_PATH, | |
70 '--gtest_also_run_disabled_tests', | |
71 '--gtest_internal_skip_environment_and_ad_hoc_tests', | |
72 '--gtest_filter=*DISABLED_*']) | |
73 COMMAND_WITH_SHARDING = ( | |
74 {'GTEST_SHARD_INDEX': '1', 'GTEST_TOTAL_SHARDS': '2'}, | |
75 [PROGRAM_PATH, | |
76 '--gtest_internal_skip_environment_and_ad_hoc_tests', | |
77 '--gtest_filter=PassingTest.*']) | |
78 | |
79 GOLDEN_PATH = os.path.join(gtest_test_utils.GetSourceDir(), GOLDEN_NAME) | |
80 | |
81 | |
82 def ToUnixLineEnding(s): | |
83 """Changes all Windows/Mac line endings in s to UNIX line endings.""" | |
84 | |
85 return s.replace('\r\n', '\n').replace('\r', '\n') | |
86 | |
87 | |
88 def RemoveLocations(test_output): | |
89 """Removes all file location info from a Google Test program's output. | |
90 | |
91 Args: | |
92 test_output: the output of a Google Test program. | |
93 | |
94 Returns: | |
95 output with all file location info (in the form of | |
96 'DIRECTORY/FILE_NAME:LINE_NUMBER: 'or | |
97 'DIRECTORY\\FILE_NAME(LINE_NUMBER): ') replaced by | |
98 'FILE_NAME:#: '. | |
99 """ | |
100 | |
101 return re.sub(r'.*[/\\](.+)(\:\d+|\(\d+\))\: ', r'\1:#: ', test_output) | |
102 | |
103 | |
104 def RemoveStackTraceDetails(output): | |
105 """Removes all stack traces from a Google Test program's output.""" | |
106 | |
107 # *? means "find the shortest string that matches". | |
108 return re.sub(r'Stack trace:(.|\n)*?\n\n', | |
109 'Stack trace: (omitted)\n\n', output) | |
110 | |
111 | |
112 def RemoveStackTraces(output): | |
113 """Removes all traces of stack traces from a Google Test program's output.""" | |
114 | |
115 # *? means "find the shortest string that matches". | |
116 return re.sub(r'Stack trace:(.|\n)*?\n\n', '', output) | |
117 | |
118 | |
119 def RemoveTime(output): | |
120 """Removes all time information from a Google Test program's output.""" | |
121 | |
122 return re.sub(r'\(\d+ ms', '(? ms', output) | |
123 | |
124 | |
125 def RemoveTypeInfoDetails(test_output): | |
126 """Removes compiler-specific type info from Google Test program's output. | |
127 | |
128 Args: | |
129 test_output: the output of a Google Test program. | |
130 | |
131 Returns: | |
132 output with type information normalized to canonical form. | |
133 """ | |
134 | |
135 # some compilers output the name of type 'unsigned int' as 'unsigned' | |
136 return re.sub(r'unsigned int', 'unsigned', test_output) | |
137 | |
138 | |
139 def NormalizeToCurrentPlatform(test_output): | |
140 """Normalizes platform specific output details for easier comparison.""" | |
141 | |
142 if IS_WINDOWS: | |
143 # Removes the color information that is not present on Windows. | |
144 test_output = re.sub('\x1b\\[(0;3\d)?m', '', test_output) | |
145 # Changes failure message headers into the Windows format. | |
146 test_output = re.sub(r': Failure\n', r': error: ', test_output) | |
147 # Changes file(line_number) to file:line_number. | |
148 test_output = re.sub(r'((\w|\.)+)\((\d+)\):', r'\1:\3:', test_output) | |
149 | |
150 return test_output | |
151 | |
152 | |
153 def RemoveTestCounts(output): | |
154 """Removes test counts from a Google Test program's output.""" | |
155 | |
156 output = re.sub(r'\d+ tests?, listed below', | |
157 '? tests, listed below', output) | |
158 output = re.sub(r'\d+ FAILED TESTS', | |
159 '? FAILED TESTS', output) | |
160 output = re.sub(r'\d+ tests? from \d+ test cases?', | |
161 '? tests from ? test cases', output) | |
162 output = re.sub(r'\d+ tests? from ([a-zA-Z_])', | |
163 r'? tests from \1', output) | |
164 return re.sub(r'\d+ tests?\.', '? tests.', output) | |
165 | |
166 | |
167 def RemoveMatchingTests(test_output, pattern): | |
168 """Removes output of specified tests from a Google Test program's output. | |
169 | |
170 This function strips not only the beginning and the end of a test but also | |
171 all output in between. | |
172 | |
173 Args: | |
174 test_output: A string containing the test output. | |
175 pattern: A regex string that matches names of test cases or | |
176 tests to remove. | |
177 | |
178 Returns: | |
179 Contents of test_output with tests whose names match pattern removed. | |
180 """ | |
181 | |
182 test_output = re.sub( | |
183 r'.*\[ RUN \] .*%s(.|\n)*?\[( FAILED | OK )\] .*%s.*\n' % ( | |
184 pattern, pattern), | |
185 '', | |
186 test_output) | |
187 return re.sub(r'.*%s.*\n' % pattern, '', test_output) | |
188 | |
189 | |
190 def NormalizeOutput(output): | |
191 """Normalizes output (the output of gtest_output_test_.exe).""" | |
192 | |
193 output = ToUnixLineEnding(output) | |
194 output = RemoveLocations(output) | |
195 output = RemoveStackTraceDetails(output) | |
196 output = RemoveTime(output) | |
197 return output | |
198 | |
199 | |
200 def GetShellCommandOutput(env_cmd): | |
201 """Runs a command in a sub-process, and returns its output in a string. | |
202 | |
203 Args: | |
204 env_cmd: The shell command. A 2-tuple where element 0 is a dict of extra | |
205 environment variables to set, and element 1 is a string with | |
206 the command and any flags. | |
207 | |
208 Returns: | |
209 A string with the command's combined standard and diagnostic output. | |
210 """ | |
211 | |
212 # Spawns cmd in a sub-process, and gets its standard I/O file objects. | |
213 # Set and save the environment properly. | |
214 environ = os.environ.copy() | |
215 environ.update(env_cmd[0]) | |
216 p = gtest_test_utils.Subprocess(env_cmd[1], env=environ) | |
217 | |
218 return p.output | |
219 | |
220 | |
221 def GetCommandOutput(env_cmd): | |
222 """Runs a command and returns its output with all file location | |
223 info stripped off. | |
224 | |
225 Args: | |
226 env_cmd: The shell command. A 2-tuple where element 0 is a dict of extra | |
227 environment variables to set, and element 1 is a string with | |
228 the command and any flags. | |
229 """ | |
230 | |
231 # Disables exception pop-ups on Windows. | |
232 environ, cmdline = env_cmd | |
233 environ = dict(environ) # Ensures we are modifying a copy. | |
234 environ[CATCH_EXCEPTIONS_ENV_VAR_NAME] = '1' | |
235 return NormalizeOutput(GetShellCommandOutput((environ, cmdline))) | |
236 | |
237 | |
238 def GetOutputOfAllCommands(): | |
239 """Returns concatenated output from several representative commands.""" | |
240 | |
241 return (GetCommandOutput(COMMAND_WITH_COLOR) + | |
242 GetCommandOutput(COMMAND_WITH_TIME) + | |
243 GetCommandOutput(COMMAND_WITH_DISABLED) + | |
244 GetCommandOutput(COMMAND_WITH_SHARDING)) | |
245 | |
246 | |
247 test_list = GetShellCommandOutput(COMMAND_LIST_TESTS) | |
248 SUPPORTS_DEATH_TESTS = 'DeathTest' in test_list | |
249 SUPPORTS_TYPED_TESTS = 'TypedTest' in test_list | |
250 SUPPORTS_THREADS = 'ExpectFailureWithThreadsTest' in test_list | |
251 SUPPORTS_STACK_TRACES = False | |
252 | |
253 CAN_GENERATE_GOLDEN_FILE = (SUPPORTS_DEATH_TESTS and | |
254 SUPPORTS_TYPED_TESTS and | |
255 SUPPORTS_THREADS) | |
256 | |
257 | |
258 class GTestOutputTest(gtest_test_utils.TestCase): | |
259 def RemoveUnsupportedTests(self, test_output): | |
260 if not SUPPORTS_DEATH_TESTS: | |
261 test_output = RemoveMatchingTests(test_output, 'DeathTest') | |
262 if not SUPPORTS_TYPED_TESTS: | |
263 test_output = RemoveMatchingTests(test_output, 'TypedTest') | |
264 test_output = RemoveMatchingTests(test_output, 'TypedDeathTest') | |
265 test_output = RemoveMatchingTests(test_output, 'TypeParamDeathTest') | |
266 if not SUPPORTS_THREADS: | |
267 test_output = RemoveMatchingTests(test_output, | |
268 'ExpectFailureWithThreadsTest') | |
269 test_output = RemoveMatchingTests(test_output, | |
270 'ScopedFakeTestPartResultReporterTest') | |
271 test_output = RemoveMatchingTests(test_output, | |
272 'WorksConcurrently') | |
273 if not SUPPORTS_STACK_TRACES: | |
274 test_output = RemoveStackTraces(test_output) | |
275 | |
276 return test_output | |
277 | |
278 def testOutput(self): | |
279 output = GetOutputOfAllCommands() | |
280 | |
281 golden_file = open(GOLDEN_PATH, 'rb') | |
282 # A mis-configured source control system can cause \r appear in EOL | |
283 # sequences when we read the golden file irrespective of an operating | |
284 # system used. Therefore, we need to strip those \r's from newlines | |
285 # unconditionally. | |
286 golden = ToUnixLineEnding(golden_file.read()) | |
287 golden_file.close() | |
288 | |
289 # We want the test to pass regardless of certain features being | |
290 # supported or not. | |
291 | |
292 # We still have to remove type name specifics in all cases. | |
293 normalized_actual = RemoveTypeInfoDetails(output) | |
294 normalized_golden = RemoveTypeInfoDetails(golden) | |
295 | |
296 if CAN_GENERATE_GOLDEN_FILE: | |
297 self.assertEqual(normalized_golden, normalized_actual) | |
298 else: | |
299 normalized_actual = NormalizeToCurrentPlatform( | |
300 RemoveTestCounts(normalized_actual)) | |
301 normalized_golden = NormalizeToCurrentPlatform( | |
302 RemoveTestCounts(self.RemoveUnsupportedTests(normalized_golden))) | |
303 | |
304 # This code is very handy when debugging golden file differences: | |
305 if os.getenv('DEBUG_GTEST_OUTPUT_TEST'): | |
306 open(os.path.join( | |
307 gtest_test_utils.GetSourceDir(), | |
308 '_gtest_output_test_normalized_actual.txt'), 'wb').write( | |
309 normalized_actual) | |
310 open(os.path.join( | |
311 gtest_test_utils.GetSourceDir(), | |
312 '_gtest_output_test_normalized_golden.txt'), 'wb').write( | |
313 normalized_golden) | |
314 | |
315 self.assertEqual(normalized_golden, normalized_actual) | |
316 | |
317 | |
318 if __name__ == '__main__': | |
319 if sys.argv[1:] == [GENGOLDEN_FLAG]: | |
320 if CAN_GENERATE_GOLDEN_FILE: | |
321 output = GetOutputOfAllCommands() | |
322 golden_file = open(GOLDEN_PATH, 'wb') | |
323 golden_file.write(output) | |
324 golden_file.close() | |
325 else: | |
326 message = ( | |
327 """Unable to write a golden file when compiled in an environment | |
328 that does not support all the required features (death tests, typed tests, | |
329 and multiple threads). Please generate the golden file using a binary built | |
330 with those features enabled.""") | |
331 | |
332 sys.stderr.write(message) | |
333 sys.exit(1) | |
334 else: | |
335 gtest_test_utils.Main() | |
OLD | NEW |