add a script for running tests locally

Takashi Matsuo · Takashi Matsuo · commit 74cb963b94f0 · 2020-05-28T22:43:46.000Z
diff --git a/scripts/run_tests_local.sh b/scripts/run_tests_local.sh
@@ -0,0 +1,64 @@
+#!/usr/bin/env bash
+# Copyright 2020 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.
+
+# run_tests_local.sh
+#
+# This script is a helper script for running tests with
+# .kokoro/trampoline_v2.sh.
+# run_tests_local.sh directory (sessions..)
+
+set -euo pipefail
+
+sessions=(
+    "lint"
+    "py-3.6"
+    "py-3.7"
+)
+
+# The only required argument is a directory for running the tests.
+if [[ $# -lt 1 ]]; then
+    echo "Please provide at least one argument."
+    echo "Usage: run_tests_local.sh directory (sessions..)"
+    exit 1
+fi
+
+function repo_root() {
+    local dir="$1"
+    while [[ ! -d "${dir}/.git" ]]; do
+	dir="$(dirname "$dir")"
+    done
+    echo "${dir}"
+}
+
+PROGRAM_PATH="$(realpath "$0")"
+PROGRAM_DIR="$(dirname "${PROGRAM_PATH}")"
+PROJECT_ROOT="$(repo_root "${PROGRAM_DIR}")"
+
+directory="$(realpath "$1")"
+relative_dir=${directory#"${PROJECT_ROOT}/"}
+export RUN_TESTS_DIRS="${relative_dir}"
+
+if [[ $# -ge 2 ]]; then
+    sessions=("${@:2}")
+fi
+
+echo "Running tests for directory: ${directory}"
+echo "Sessions: ${sessions[@]}"
+
+for session in "${sessions[@]}"
+do
+    export RUN_TESTS_SESSION="${session}"
+    "${PROJECT_ROOT}/.kokoro/trampoline_v2.sh"
+done
