#!/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':
        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', encoding='utf-8') as f:
        f.write(dummy_output)
    exit_code = 0

sys.exit(exit_code)