llvm/utils/lit/tests/Inputs/googletest-format/DummySubDir/OneTest.py

   1 #!/usr/bin/env python
   2
   3 import os
   4 import sys
   5
   6 if len(sys.argv) == 3 and sys.argv[1] == "--gtest_list_tests":
   7     if sys.argv[2] != "--gtest_filter=-*DISABLED_*":
   8         raise ValueError("unexpected argument: %s" % (sys.argv[2]))
   9     print(
  10         """\
  11 FirstTest.
  12   subTestA
  13   subTestB
  14   subTestC
  15   subTestD
  16 ParameterizedTest/0.
  17   subTest
  18 ParameterizedTest/1.
  19   subTest"""
  20     )
  21     sys.exit(0)
  22 elif len(sys.argv) != 1:
  23     # sharding and json output are specified using environment variables
  24     raise ValueError("unexpected argument: %r" % (" ".join(sys.argv[1:])))
  25
  26 for e in ["GTEST_TOTAL_SHARDS", "GTEST_SHARD_INDEX", "GTEST_OUTPUT"]:
  27     if e not in os.environ:
  28         raise ValueError("missing environment variables: " + e)
  29
  30 if not os.environ["GTEST_OUTPUT"].startswith("json:"):
  31     raise ValueError("must emit json output: " + os.environ["GTEST_OUTPUT"])
  32
  33 output = """\
  34 {
  35 "random_seed": 123,
  36 "testsuites": [
  37     {
  38         "name": "FirstTest",
  39         "testsuite": [
  40             {
  41                 "name": "subTestA",
  42                 "result": "COMPLETED",
  43                 "time": "0.001s"
  44             },
  45             {
  46                 "name": "subTestB",
  47                 "result": "COMPLETED",
  48                 "time": "0.001s",
  49                 "failures": [
  50                     {
  51                         "failure": "I am subTest B, I FAIL\\nAnd I have two lines of output",
  52                         "type": ""
  53                     }
  54                 ]
  55             },
  56             {
  57                 "name": "subTestC",
  58                 "result": "SKIPPED",
  59                 "time": "0.001s"
  60             },
  61             {
  62                 "name": "subTestD",
  63                 "result": "UNRESOLVED",
  64                 "time": "0.001s"
  65             }
  66         ]
  67     },
  68     {
  69         "name": "ParameterizedTest/0",
  70         "testsuite": [
  71             {
  72                 "name": "subTest",
  73                 "result": "COMPLETED",
  74                 "time": "0.001s"
  75             }
  76         ]
  77     },
  78     {
  79         "name": "ParameterizedTest/1",
  80         "testsuite": [
  81             {
  82                 "name": "subTest",
  83                 "result": "COMPLETED",
  84                 "time": "0.001s"
  85             }
  86         ]
  87     }
  88 ]
  89 }"""
  90
  91 dummy_output = """\
  92 {
  93 "testsuites": [
  94 ]
  95 }"""
  96
  97 json_filename = os.environ["GTEST_OUTPUT"].split(":", 1)[1]
  98 with open(json_filename, "w", encoding="utf-8") as f:
  99     if os.environ["GTEST_SHARD_INDEX"] == "0":
 100         print("[ RUN      ] FirstTest.subTestB", flush=True)
 101         print("I am subTest B output", file=sys.stderr, flush=True)
 102         print("[  FAILED  ] FirstTest.subTestB (8 ms)", flush=True)
 103
 104         f.write(output)
 105         exit_code = 1
 106     else:
 107         f.write(dummy_output)
 108         exit_code = 0
 109
 110 sys.exit(exit_code)