tools/run_tests/xds_k8s_test_driver/bin/run_test_server.py

   1 # Copyright 2020 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 import logging
  15
  16 from absl import app
  17 from absl import flags
  18
  19 from framework import xds_flags
  20 from framework import xds_k8s_flags
  21 from framework.infrastructure import k8s
  22 from framework.test_app import server_app
  23
  24 logger = logging.getLogger(__name__)
  25 # Flags
  26 _CMD = flags.DEFINE_enum('cmd',
  27                          default='run',
  28                          enum_values=['run', 'cleanup'],
  29                          help='Command')
  30 _SECURE = flags.DEFINE_bool("secure",
  31                             default=False,
  32                             help="Run server in the secure mode")
  33 _REUSE_NAMESPACE = flags.DEFINE_bool("reuse_namespace",
  34                                      default=True,
  35                                      help="Use existing namespace if exists")
  36 _CLEANUP_NAMESPACE = flags.DEFINE_bool(
  37     "cleanup_namespace",
  38     default=False,
  39     help="Delete namespace during resource cleanup")
  40 flags.adopt_module_key_flags(xds_flags)
  41 flags.adopt_module_key_flags(xds_k8s_flags)
  42
  43
  44 def main(argv):
  45     if len(argv) > 1:
  46         raise app.UsageError('Too many command-line arguments.')
  47
  48     # Base namespace
  49     namespace = xds_flags.NAMESPACE.value
  50     server_namespace = namespace
  51
  52     runner_kwargs = dict(
  53         deployment_name=xds_flags.SERVER_NAME.value,
  54         image_name=xds_k8s_flags.SERVER_IMAGE.value,
  55         gcp_service_account=xds_k8s_flags.GCP_SERVICE_ACCOUNT.value,
  56         network=xds_flags.NETWORK.value,
  57         reuse_namespace=_REUSE_NAMESPACE.value)
  58
  59     if _SECURE.value:
  60         runner_kwargs.update(
  61             td_bootstrap_image=xds_k8s_flags.TD_BOOTSTRAP_IMAGE.value,
  62             xds_server_uri=xds_flags.XDS_SERVER_URI.value,
  63             deployment_template='server-secure.deployment.yaml')
  64
  65     k8s_api_manager = k8s.KubernetesApiManager(xds_k8s_flags.KUBE_CONTEXT.value)
  66     server_runner = server_app.KubernetesServerRunner(
  67         k8s.KubernetesNamespace(k8s_api_manager, server_namespace),
  68         **runner_kwargs)
  69
  70     if _CMD.value == 'run':
  71         logger.info('Run server, secure_mode=%s', _SECURE.value)
  72         server_runner.run(test_port=xds_flags.SERVER_PORT.value,
  73                           secure_mode=_SECURE.value)
  74
  75     elif _CMD.value == 'cleanup':
  76         logger.info('Cleanup server')
  77         server_runner.cleanup(force=True,
  78                               force_namespace=_CLEANUP_NAMESPACE.value)
  79
  80
  81 if __name__ == '__main__':
  82     app.run(main)