| 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("""\ |
| FirstTest. |
| subTestA""") |
| 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']: |
| 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": "FirstTest", |
| "testsuite": [ |
| { |
| "name": "subTestA", |
| "result": "COMPLETED", |
| "time": "0.001s" |
| } |
| ] |
| } |
| ] |
| }""" |
| |
| dummy_output = """\ |
| { |
| "testsuites": [ |
| ] |
| }""" |
| |
| json_filename = os.environ['GTEST_OUTPUT'].split(':', 1)[1] |
| with open(json_filename, 'w', encoding='utf-8') as f: |
| if os.environ['GTEST_SHARD_INDEX'] == '0': |
| f.write(output) |
| else: |
| f.write(dummy_output) |
| |
| sys.exit(0) |