tools/internal_ci/linux/grpc_xds_k8s_lb.sh

   1 #!/usr/bin/env bash
   2 # Copyright 2021 gRPC authors.
   3 #
   4 # Licensed under the Apache License, Version 2.0 (the "License");
   5 # you may not use this file except in compliance with the License.
   6 # You may obtain a copy of the License at
   7 #
   8 #     http://www.apache.org/licenses/LICENSE-2.0
   9 #
  10 # Unless required by applicable law or agreed to in writing, software
  11 # distributed under the License is distributed on an "AS IS" BASIS,
  12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 # See the License for the specific language governing permissions and
  14 # limitations under the License.
  15
  16 set -eo pipefail
  17
  18 # Constants
  19 readonly GITHUB_REPOSITORY_NAME="grpc"
  20 # GKE Cluster
  21 readonly GKE_CLUSTER_NAME="interop-test-psm-lb-v1-us-central1-a"
  22 readonly GKE_CLUSTER_ZONE="us-central1-a"
  23 readonly SECONDARY_GKE_CLUSTER_NAME="interop-test-psm-lb-v1-us-west1-b"
  24 readonly SECONDARY_GKE_CLUSTER_ZONE="us-west1-b"
  25 ## xDS test client Docker images
  26 readonly SERVER_IMAGE_NAME="gcr.io/grpc-testing/xds-interop/cpp-server"
  27 readonly CLIENT_IMAGE_NAME="gcr.io/grpc-testing/xds-interop/cpp-client"
  28 readonly FORCE_IMAGE_BUILD="${FORCE_IMAGE_BUILD:-0}"
  29 readonly BUILD_APP_PATH="interop-testing/build/install/grpc-interop-testing"
  30
  31 #######################################
  32 # Builds test app Docker images and pushes them to GCR
  33 # Globals:
  34 #   BUILD_APP_PATH
  35 #   SERVER_IMAGE_NAME: Test server Docker image name
  36 #   CLIENT_IMAGE_NAME: Test client Docker image name
  37 #   GIT_COMMIT: SHA-1 of git commit being built
  38 # Arguments:
  39 #   None
  40 # Outputs:
  41 #   Writes the output of `gcloud builds submit` to stdout, stderr
  42 #######################################
  43 build_test_app_docker_images() {
  44   echo "Building C++ xDS interop test app Docker images"
  45   docker build -f "${SRC_DIR}/tools/dockerfile/interoptest/grpc_interop_cxx_xds/Dockerfile.xds_client" -t "${CLIENT_IMAGE_NAME}:${GIT_COMMIT}" "${SRC_DIR}"
  46   docker build -f "${SRC_DIR}/tools/dockerfile/interoptest/grpc_interop_cxx_xds/Dockerfile.xds_server" -t "${SERVER_IMAGE_NAME}:${GIT_COMMIT}" "${SRC_DIR}"
  47   gcloud -q auth configure-docker
  48   docker push "${CLIENT_IMAGE_NAME}:${GIT_COMMIT}"
  49   docker push "${SERVER_IMAGE_NAME}:${GIT_COMMIT}"
  50 }
  51
  52 #######################################
  53 # Builds test app and its docker images unless they already exist
  54 # Globals:
  55 #   SERVER_IMAGE_NAME: Test server Docker image name
  56 #   CLIENT_IMAGE_NAME: Test client Docker image name
  57 #   GIT_COMMIT: SHA-1 of git commit being built
  58 #   FORCE_IMAGE_BUILD
  59 # Arguments:
  60 #   None
  61 # Outputs:
  62 #   Writes the output to stdout, stderr
  63 #######################################
  64 build_docker_images_if_needed() {
  65   # Check if images already exist
  66   server_tags="$(gcloud_gcr_list_image_tags "${SERVER_IMAGE_NAME}" "${GIT_COMMIT}")"
  67   printf "Server image: %s:%s\n" "${SERVER_IMAGE_NAME}" "${GIT_COMMIT}"
  68   echo "${server_tags:-Server image not found}"
  69
  70   client_tags="$(gcloud_gcr_list_image_tags "${CLIENT_IMAGE_NAME}" "${GIT_COMMIT}")"
  71   printf "Client image: %s:%s\n" "${CLIENT_IMAGE_NAME}" "${GIT_COMMIT}"
  72   echo "${client_tags:-Client image not found}"
  73
  74   # Build if any of the images are missing, or FORCE_IMAGE_BUILD=1
  75   if [[ "${FORCE_IMAGE_BUILD}" == "1" || -z "${server_tags}" || -z "${client_tags}" ]]; then
  76     build_test_app_docker_images
  77   else
  78     echo "Skipping C++ test app build"
  79   fi
  80 }
  81
  82 #######################################
  83 # Executes the test case
  84 # Globals:
  85 #   TEST_DRIVER_FLAGFILE: Relative path to test driver flagfile
  86 #   KUBE_CONTEXT: The name of kubectl context with GKE cluster access
  87 #   SECONDARY_KUBE_CONTEXT: The name of kubectl context with secondary GKE cluster access, if any
  88 #   TEST_XML_OUTPUT_DIR: Output directory for the test xUnit XML report
  89 #   SERVER_IMAGE_NAME: Test server Docker image name
  90 #   CLIENT_IMAGE_NAME: Test client Docker image name
  91 #   GIT_COMMIT: SHA-1 of git commit being built
  92 # Arguments:
  93 #   Test case name
  94 # Outputs:
  95 #   Writes the output of test execution to stdout, stderr
  96 #   Test xUnit report to ${TEST_XML_OUTPUT_DIR}/${test_name}/sponge_log.xml
  97 #######################################
  98 run_test() {
  99   # Test driver usage:
 100   # https://github.com/grpc/grpc/tree/master/tools/run_tests/xds_k8s_test_driver#basic-usage
 101   local test_name="${1:?Usage: run_test test_name}"
 102   python -m "tests.${test_name}" \
 103     --flagfile="${TEST_DRIVER_FLAGFILE}" \
 104     --kube_context="${KUBE_CONTEXT}" \
 105     --secondary_kube_context="${SECONDARY_KUBE_CONTEXT}" \
 106     --server_image="${SERVER_IMAGE_NAME}:${GIT_COMMIT}" \
 107     --client_image="${CLIENT_IMAGE_NAME}:${GIT_COMMIT}" \
 108     --xml_output_file="${TEST_XML_OUTPUT_DIR}/${test_name}/sponge_log.xml"
 109 }
 110
 111 #######################################
 112 # Main function: provision software necessary to execute tests, and run them
 113 # Globals:
 114 #   KOKORO_ARTIFACTS_DIR
 115 #   GITHUB_REPOSITORY_NAME
 116 #   SRC_DIR: Populated with absolute path to the source repo
 117 #   TEST_DRIVER_REPO_DIR: Populated with the path to the repo containing
 118 #                         the test driver
 119 #   TEST_DRIVER_FULL_DIR: Populated with the path to the test driver source code
 120 #   TEST_DRIVER_FLAGFILE: Populated with relative path to test driver flagfile
 121 #   TEST_XML_OUTPUT_DIR: Populated with the path to test xUnit XML report
 122 #   GIT_ORIGIN_URL: Populated with the origin URL of git repo used for the build
 123 #   GIT_COMMIT: Populated with the SHA-1 of git commit being built
 124 #   GIT_COMMIT_SHORT: Populated with the short SHA-1 of git commit being built
 125 #   KUBE_CONTEXT: Populated with name of kubectl context with GKE cluster access
 126 #   SECONDARY_KUBE_CONTEXT: Populated with name of kubectl context with secondary GKE cluster access, if any
 127 # Arguments:
 128 #   None
 129 # Outputs:
 130 #   Writes the output of test execution to stdout, stderr
 131 #######################################
 132 main() {
 133   local script_dir
 134   script_dir="$(dirname "$0")"
 135   # shellcheck source=tools/internal_ci/linux/grpc_xds_k8s_install_test_driver.sh
 136   source "${script_dir}/grpc_xds_k8s_install_test_driver.sh"
 137   set -x
 138   if [[ -n "${KOKORO_ARTIFACTS_DIR}" ]]; then
 139     kokoro_setup_test_driver "${GITHUB_REPOSITORY_NAME}"
 140   else
 141     local_setup_test_driver "${script_dir}"
 142   fi
 143   build_docker_images_if_needed
 144   # Run tests
 145   cd "${TEST_DRIVER_FULL_DIR}"
 146   local failed_tests=0
 147   test_suites=("change_backend_service_test" "failover_test" "remove_neg_test" "round_robin_test" "affinity_test")
 148   for test in "${test_suites[@]}"; do
 149     run_test $test || (( failed_tests++ ))
 150   done
 151   echo "Failed test suites: ${failed_tests}"
 152   if (( failed_tests > 0 )); then
 153     exit 1
 154   fi
 155
 156 }
 157
 158 main "$@"