67 lines
1.6 KiB
Python
67 lines
1.6 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', encoding='utf-8') as f:
|
|
f.write(output)
|
|
exit_code = 0
|
|
elif test_name == 'InfiniteLoopSubTest':
|
|
while True:
|
|
pass
|
|
else:
|
|
raise SystemExit("error: invalid test name: %r" % (test_name,))
|
|
else:
|
|
with open(json_filename, 'w', encoding='utf-8') as f:
|
|
f.write(dummy_output)
|
|
exit_code = 0
|
|
|
|
sys.exit(exit_code)
|