d3efa577f5
When a unit test crashes or timeout, print the shard's stdout and stderr. When a unit test fails, attaches the test's output to the LIT output to help debugging. While at it, concatenating shard's environment variables using space instead of newline to make the reproducer script user friendly. Based on D123797. (Thanks to @lenary)
69 lines
1.7 KiB
Python
69 lines
1.7 KiB
Python
#!/usr/bin/env python
|
|
|
|
import os
|
|
import sys
|
|
|
|
if len(sys.argv) == 3 and sys.argv[1] == "--gtest_list_tests":
|
|
if sys.argv[2] != '--gtest_filter=-*DISABLED_*':
|
|
raise ValueError("unexpected argument: %s" % (sys.argv[2]))
|
|
print("""\
|
|
T.
|
|
QuickSubTest
|
|
InfiniteLoopSubTest
|
|
""")
|
|
sys.exit(0)
|
|
elif len(sys.argv) != 1:
|
|
# sharding and json output are specified using environment variables
|
|
raise ValueError("unexpected argument: %r" % (' '.join(sys.argv[1:])))
|
|
|
|
for e in ['GTEST_TOTAL_SHARDS', 'GTEST_SHARD_INDEX', 'GTEST_OUTPUT', 'GTEST_FILTER']:
|
|
if e not in os.environ:
|
|
raise ValueError("missing environment variables: " + e)
|
|
|
|
if not os.environ['GTEST_OUTPUT'].startswith('json:'):
|
|
raise ValueError("must emit json output: " + os.environ['GTEST_OUTPUT'])
|
|
|
|
output = """\
|
|
{
|
|
"testsuites": [
|
|
{
|
|
"name": "T",
|
|
"testsuite": [
|
|
{
|
|
"name": "QuickSubTest",
|
|
"result": "COMPLETED",
|
|
"time": "2s"
|
|
}
|
|
]
|
|
}
|
|
]
|
|
}"""
|
|
|
|
dummy_output = """\
|
|
{
|
|
"testsuites": [
|
|
]
|
|
}"""
|
|
|
|
json_filename = os.environ['GTEST_OUTPUT'].split(':', 1)[1]
|
|
|
|
if os.environ['GTEST_SHARD_INDEX'] == '0':
|
|
test_name = os.environ['GTEST_FILTER']
|
|
if test_name == 'QuickSubTest':
|
|
with open(json_filename, 'w') as f:
|
|
f.write(output)
|
|
exit_code = 0
|
|
elif test_name == 'InfiniteLoopSubTest':
|
|
print('[ RUN ] T.InfiniteLoopSubTest', flush=True)
|
|
print('some in progess output', file=sys.stderr, flush=True)
|
|
while True:
|
|
pass
|
|
else:
|
|
raise SystemExit("error: invalid test name: %r" % (test_name,))
|
|
else:
|
|
with open(json_filename, 'w') as f:
|
|
f.write(dummy_output)
|
|
exit_code = 0
|
|
|
|
sys.exit(exit_code)
|