1 #!/usr/bin/env python 2 3 import sys 4 5 if len(sys.argv) != 2: 6 raise ValueError("unexpected number of args") 7 8 if sys.argv[1] == "--gtest_list_tests": 9 print("""\ 10 FirstTest. 11 subTestA 12 subTestB 13 ParameterizedTest/0. 14 subTest 15 ParameterizedTest/1. 16 subTest""") 17 sys.exit(0) 18 elif not sys.argv[1].startswith("--gtest_filter="): 19 raise ValueError("unexpected argument: %r" % (sys.argv[1])) 20 21 test_name = sys.argv[1].split('=',1)[1] 22 if test_name == 'FirstTest.subTestA': 23 print('I am subTest A, I PASS') 24 sys.exit(0) 25 elif test_name == 'FirstTest.subTestB': 26 print('I am subTest B, I FAIL') 27 print('And I have two lines of output') 28 sys.exit(1) 29 elif test_name in ('ParameterizedTest/0.subTest', 30 'ParameterizedTest/1.subTest'): 31 print('I am a parameterized test, I also PASS') 32 sys.exit(0) 33 else: 34 raise SystemExit("error: invalid test name: %r" % (test_name,)) 35
