#!/usr/bin/env bash # Copyright 2021 gRPC authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. set -eo pipefail # Constants readonly GITHUB_REPOSITORY_NAME="grpc" readonly TEST_DRIVER_INSTALL_SCRIPT_URL="https://raw.githubusercontent.com/${TEST_DRIVER_REPO_OWNER:-grpc}/psm-interop/${TEST_DRIVER_BRANCH:-main}/.kokoro/psm_interop_kokoro_lib.sh" ## xDS test server/client Docker images readonly DOCKER_REGISTRY="us-docker.pkg.dev" readonly SERVER_IMAGE_NAME="us-docker.pkg.dev/grpc-testing/psm-interop/python-server" readonly CLIENT_IMAGE_NAME="us-docker.pkg.dev/grpc-testing/psm-interop/python-client" readonly FORCE_IMAGE_BUILD="${FORCE_IMAGE_BUILD:-0}" readonly BUILD_APP_PATH="interop-testing/build/install/grpc-interop-testing" readonly LANGUAGE_NAME="Python" ####################################### # Builds test app Docker images and pushes them to GCR # Globals: # BUILD_APP_PATH # SERVER_IMAGE_NAME: Test server Docker image name # CLIENT_IMAGE_NAME: Test client Docker image name # GIT_COMMIT: SHA-1 of git commit being built # TESTING_VERSION: version branch under test, f.e. v1.42.x, master # Arguments: # None # Outputs: # Writes the output of `gcloud builds submit` to stdout, stderr ####################################### build_test_app_docker_images() { echo "Building ${LANGUAGE_NAME} xDS interop test app Docker images" pushd "${SRC_DIR}" docker build \ -f src/python/grpcio_tests/tests_py3_only/interop/Dockerfile.client \ -t "${CLIENT_IMAGE_NAME}:${GIT_COMMIT}" \ . docker build \ -f src/python/grpcio_tests/tests_py3_only/interop/Dockerfile.server \ -t "${SERVER_IMAGE_NAME}:${GIT_COMMIT}" \ . popd gcloud -q auth configure-docker "${DOCKER_REGISTRY}" docker push "${CLIENT_IMAGE_NAME}:${GIT_COMMIT}" docker push "${SERVER_IMAGE_NAME}:${GIT_COMMIT}" if is_version_branch "${TESTING_VERSION}"; then tag_and_push_docker_image "${CLIENT_IMAGE_NAME}" "${GIT_COMMIT}" "${TESTING_VERSION}" tag_and_push_docker_image "${SERVER_IMAGE_NAME}" "${GIT_COMMIT}" "${TESTING_VERSION}" fi } ####################################### # Builds test app and its docker images unless they already exist # Globals: # SERVER_IMAGE_NAME: Test server Docker image name # CLIENT_IMAGE_NAME: Test client Docker image name # GIT_COMMIT: SHA-1 of git commit being built # FORCE_IMAGE_BUILD # Arguments: # None # Outputs: # Writes the output to stdout, stderr ####################################### build_docker_images_if_needed() { # Check if images already exist server_tags="$(gcloud_gcr_list_image_tags "${SERVER_IMAGE_NAME}" "${GIT_COMMIT}")" printf "Server image: %s:%s\n" "${SERVER_IMAGE_NAME}" "${GIT_COMMIT}" echo "${server_tags:-Server image not found}" client_tags="$(gcloud_gcr_list_image_tags "${CLIENT_IMAGE_NAME}" "${GIT_COMMIT}")" printf "Client image: %s:%s\n" "${CLIENT_IMAGE_NAME}" "${GIT_COMMIT}" echo "${client_tags:-Client image not found}" # Build if any of the images are missing, or FORCE_IMAGE_BUILD=1 if [[ "${FORCE_IMAGE_BUILD}" == "1" || -z "${server_tags}" || -z "${client_tags}" ]]; then build_test_app_docker_images else echo "Skipping ${LANGUAGE_NAME} test app build" fi } ####################################### # Executes the test case # Globals: # TEST_DRIVER_FLAGFILE: Relative path to test driver flagfile # KUBE_CONTEXT: The name of kubectl context with GKE cluster access # TEST_XML_OUTPUT_DIR: Output directory for the test xUnit XML report # SERVER_IMAGE_NAME: Test server Docker image name # CLIENT_IMAGE_NAME: Test client Docker image name # GIT_COMMIT: SHA-1 of git commit being built # TESTING_VERSION: version branch under test: used by the framework to determine the supported PSM # features. # Arguments: # Test case name # Outputs: # Writes the output of test execution to stdout, stderr # Test xUnit report to ${TEST_XML_OUTPUT_DIR}/${test_name}/sponge_log.xml ####################################### run_test() { # Test driver usage: # https://github.com/grpc/psm-interop#basic-usage local test_name="${1:?Usage: run_test test_name}" local out_dir="${TEST_XML_OUTPUT_DIR}/${test_name}" mkdir -pv "${out_dir}" set -x python3 -m "tests.${test_name}" \ --flagfile="${TEST_DRIVER_FLAGFILE}" \ --kube_context="${KUBE_CONTEXT}" \ --server_image="${SERVER_IMAGE_NAME}:${GIT_COMMIT}" \ --client_image="${CLIENT_IMAGE_NAME}:${GIT_COMMIT}" \ --testing_version="${TESTING_VERSION}" \ --nocheck_local_certs \ --force_cleanup \ --collect_app_logs \ --log_dir="${out_dir}" \ --xml_output_file="${out_dir}/sponge_log.xml" \ |& tee "${out_dir}/sponge_log.log" set +x } ####################################### # Main function: provision software necessary to execute tests, and run them # Globals: # KOKORO_ARTIFACTS_DIR # GITHUB_REPOSITORY_NAME # SRC_DIR: Populated with absolute path to the source repo # TEST_DRIVER_REPO_DIR: Populated with the path to the repo containing # the test driver # TEST_DRIVER_FULL_DIR: Populated with the path to the test driver source code # TEST_DRIVER_FLAGFILE: Populated with relative path to test driver flagfile # TEST_XML_OUTPUT_DIR: Populated with the path to test xUnit XML report # GIT_ORIGIN_URL: Populated with the origin URL of git repo used for the build # GIT_COMMIT: Populated with the SHA-1 of git commit being built # GIT_COMMIT_SHORT: Populated with the short SHA-1 of git commit being built # KUBE_CONTEXT: Populated with name of kubectl context with GKE cluster access # Arguments: # None # Outputs: # Writes the output of test execution to stdout, stderr ####################################### main() { local script_dir script_dir="$(dirname "$0")" # Source the test driver from the master branch. echo "Sourcing test driver install script from: ${TEST_DRIVER_INSTALL_SCRIPT_URL}" source /dev/stdin <<< "$(curl -s "${TEST_DRIVER_INSTALL_SCRIPT_URL}")" activate_gke_cluster GKE_CLUSTER_PSM_SECURITY set -x if [[ -n "${KOKORO_ARTIFACTS_DIR}" ]]; then kokoro_setup_test_driver "${GITHUB_REPOSITORY_NAME}" else local_setup_test_driver "${script_dir}" fi build_docker_images_if_needed # Run tests cd "${TEST_DRIVER_FULL_DIR}" local failed_tests=0 test_suites=("baseline_test" "security_test" "authz_test") for test in "${test_suites[@]}"; do run_test $test || (( ++failed_tests )) done echo "Failed test suites: ${failed_tests}" } main "$@"