#!/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)