test/core/end2end/gen_build_yaml.py

   1 # Copyright 2015 gRPC authors.
   2 #
   3 # Licensed under the Apache License, Version 2.0 (the "License");
   4 # you may not use this file except in compliance with the License.
   5 # You may obtain a copy of the License at
   6 #
   7 #     http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 # Unless required by applicable law or agreed to in writing, software
  10 # distributed under the License is distributed on an "AS IS" BASIS,
  11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 # See the License for the specific language governing permissions and
  13 # limitations under the License.
  14 """Generates the list of end2end test cases from generate_tests.bzl"""
  15
  16 import os
  17 import sys
  18 import yaml
  19
  20 _ROOT = os.path.abspath(os.path.join(os.path.dirname(sys.argv[0]), '../../..'))
  21 os.chdir(_ROOT)
  22
  23
  24 def load(*args):
  25     """Replacement of bazel's load() function"""
  26     pass
  27
  28
  29 def struct(**kwargs):
  30     return kwargs  # all the args as a dict
  31
  32
  33 # generate_tests.bzl is now the source of truth for end2end tests.
  34 # The .bzl file is basically a python file and we can "execute" it
  35 # to get access to the variables it defines.
  36 exec(
  37     compile(
  38         open('test/core/end2end/generate_tests.bzl', "rb").read(),
  39         'test/core/end2end/generate_tests.bzl', 'exec'))
  40
  41
  42 def main():
  43     json = {
  44         # needed by end2end_tests.cc.template and end2end_nosec_tests.cc.template
  45         'core_end2end_tests':
  46             dict((t, END2END_TESTS[t]['secure']) for t in END2END_TESTS.keys())
  47     }
  48     print(yaml.dump(json))
  49
  50
  51 if __name__ == '__main__':
  52     main()