kokoro/testutils/run_bazel_tests.sh

#!/bin/bash
# Copyright 2022 Google LLC
#
# 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.
################################################################################

# This script runs all the Bazel tests within a given workspace directory.
#
# Users must spcify the WORKSPACE directory. Optionally, the user can specify
# a set of additional manual targets to run.
#
# Usage:
#   ./kokoro/testutils/run_bazel_tests.sh [-mh] <workspace directory> \
#     [<manual target> <manual target> ...]

# Note: -E extends the trap to shell functions, command substitutions, and
# commands executed in a subshell environment.
set -eEo pipefail
# Print some debug output on error before exiting.
trap print_debug_output ERR

usage() {
  echo "Usage: $0 [-mh] <workspace directory> \\"
  echo "         [<manual target> <manual target> ...]"
  echo "  -m: Runs only the manual targets. If set, manual targets must be"
  echo "      provided."
  echo "  -h: Help. Print this usage information."
  exit 1
}

readonly PLATFORM="$(uname | tr '[:upper:]' '[:lower:]')"
MANUAL_ONLY="false"
WORKSPACE_DIR=
MANUAL_TARGETS=

#######################################
# Process command line arguments.
#
# Globals:
#   WORKSPACE_DIR
#   MANUAL_TARGETS
#######################################
process_args() {
  # Parse options.
  while getopts "mh" opt; do
    case "${opt}" in
      d) MANUAL_ONLY="true" ;;
      h) usage ;;
      *) usage ;;
    esac
  done
  shift $((OPTIND - 1))

  WORKSPACE_DIR="$1"
  readonly WORKSPACE_DIR

  if [[ -z "${WORKSPACE_DIR}" ]]; then
    usage
  fi

  shift 1
  MANUAL_TARGETS=("$@")
  readonly MANUAL_TARGETS

  if [[ "${MANUAL_ONLY}" == "true" ]] && (( ${#MANUAL_TARGETS[@]} == 0 )); then
    usage
  fi
}

#######################################
# Print some debugging output.
#######################################
print_debug_output() {
  ls -l
  df -h
}

main() {
  process_args "$@"

  local -a test_flags=(
    --strategy=TestRunner=standalone
    --test_output=all
  )
  if [[ "${PLATFORM}" == 'darwin' ]]; then
    test_flags+=( --jvmopt="-Djava.net.preferIPv6Addresses=true" )
  fi
  readonly test_flags
  (
    cd "${WORKSPACE_DIR}"
    if [[ "${MANUAL_ONLY}" == "false" ]]; then
      time bazel build -- ...
      # Exit code 4 means targets build correctly but no tests were found. See
      # https://bazel.build/docs/scripts#exit-codes.
      bazel_test_return=0
      time bazel test "${test_flags[@]}" -- ... || bazel_test_return="$?"
      if (( $bazel_test_return != 0 && $bazel_test_return != 4 )); then
        return "${bazel_test_return}"
      fi
    fi
    # Run specific manual targets.
    if (( ${#MANUAL_TARGETS[@]} > 0 )); then
      time bazel build -- "${MANUAL_TARGETS[@]}"
      time bazel test "${test_flags[@]}"  -- "${MANUAL_TARGETS[@]}"
    fi
  )
}

main "$@"