mirror of https://github.com/grpc/grpc.git
xds-k8s: add run and cleanup helpers, document bin/ scripts (#26344)
Sergii Tkachenko
4 years ago
committed by
GitHub
parent
edae7450f5
commit
3ba36cc0c9
5 changed files with 377 additions and 27 deletions
@ -0,0 +1,58 @@ |
||||
#!/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 |
||||
|
||||
display_usage() { |
||||
cat <<EOF >/dev/stderr |
||||
Performs full TD and K8S resource cleanup |
||||
|
||||
USAGE: $0 [--nosecure] [arguments] |
||||
--nosecure: Skip cleanup for the resources specific for PSM Security |
||||
arguments ...: additional arguments passed to ./run.sh |
||||
|
||||
ENVIRONMENT: |
||||
XDS_K8S_CONFIG: file path to the config flagfile, relative to |
||||
driver root folder. Default: config/local-dev.cfg |
||||
Will be appended as --flagfile="config_absolute_path" argument |
||||
XDS_K8S_DRIVER_VENV_DIR: the path to python virtual environment directory |
||||
Default: $XDS_K8S_DRIVER_DIR/venv |
||||
EXAMPLES: |
||||
$0 |
||||
$0 --secure |
||||
XDS_K8S_CONFIG=./path-to-flagfile.cfg $0 --namespace=override-namespace |
||||
EOF |
||||
exit 1 |
||||
} |
||||
|
||||
if [[ "$1" == "-h" || "$1" == "--help" ]]; then |
||||
display_usage |
||||
fi |
||||
|
||||
readonly SCRIPT_DIR="$( cd -- "$(dirname "$0")" >/dev/null 2>&1 ; pwd -P )" |
||||
readonly XDS_K8S_DRIVER_DIR="${SCRIPT_DIR}/.." |
||||
|
||||
cd "${XDS_K8S_DRIVER_DIR}" |
||||
|
||||
if [[ "$1" == "--nosecure" ]]; then |
||||
shift |
||||
./run.sh bin/run_td_setup.py --cmd=cleanup "$@" && \ |
||||
./run.sh bin/run_test_client.py --cmd=cleanup "$@" && \ |
||||
./run.sh bin/run_test_server.py --cmd=cleanup --cleanup_namespace "$@" |
||||
else |
||||
./run.sh bin/run_td_setup.py --cmd=cleanup --security=mtls "$@" && \ |
||||
./run.sh bin/run_test_client.py --cmd=cleanup --secure "$@" && \ |
||||
./run.sh bin/run_test_server.py --cmd=cleanup --cleanup_namespace --secure "$@" |
||||
fi |
||||
@ -0,0 +1,71 @@ |
||||
#!/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 |
||||
|
||||
readonly XDS_K8S_DRIVER_DIR="$( cd -- "$(dirname "$0")" >/dev/null 2>&1 ; pwd -P )" |
||||
readonly XDS_K8S_DRIVER_VENV_DIR="${XDS_K8S_DRIVER_VENV_DIR:-$XDS_K8S_DRIVER_DIR/venv}" |
||||
readonly XDS_K8S_CONFIG="${XDS_K8S_CONFIG:-$XDS_K8S_DRIVER_DIR/config/local-dev.cfg}" |
||||
|
||||
display_usage() { |
||||
cat <<EOF >/dev/stderr |
||||
Convenience script to execute tests/ and helper bin/ scripts. |
||||
|
||||
USAGE: $0 script_path [arguments] |
||||
script_path: path to python script to execute, relative to driver root folder |
||||
arguments ...: arguments passed to program in sys.argv |
||||
|
||||
ENVIRONMENT: |
||||
XDS_K8S_CONFIG: file path to the config flagfile, relative to |
||||
driver root folder. Default: config/local-dev.cfg |
||||
Will be appended as --flagfile="config_absolute_path" argument |
||||
XDS_K8S_DRIVER_VENV_DIR: the path to python virtual environment directory |
||||
Default: $XDS_K8S_DRIVER_DIR/venv |
||||
DESCRIPTION: |
||||
This tool performs the following: |
||||
1) Ensures python virtual env installed and activated |
||||
2) Exports test driver root in PYTHONPATH |
||||
3) Automatically appends --flagfile="\$XDS_K8S_CONFIG" argument |
||||
|
||||
EXAMPLES: |
||||
$0 bin/run_td_setup.py --help # list script-specific options |
||||
$0 bin/run_td_setup.py --helpfull # list all available options |
||||
XDS_K8S_CONFIG=./path-to-flagfile.cfg $0 bin/run_td_setup.py --namespace=override-namespace |
||||
$0 tests/baseline_test.py |
||||
$0 tests/security_test.py --verbosity=1 --logger_levels=__main__:DEBUG,framework:DEBUG |
||||
$0 tests/security_test.py SecurityTest.test_mtls --nocheck_local_certs |
||||
EOF |
||||
exit 1 |
||||
} |
||||
|
||||
if [[ "$#" -eq 0 || "$1" = "-h" || "$1" = "--help" ]]; then |
||||
display_usage |
||||
fi |
||||
|
||||
if [[ -z "${VIRTUAL_ENV}" ]]; then |
||||
if [[ -d "${XDS_K8S_DRIVER_VENV_DIR}" ]]; then |
||||
# Intentional: No need to check python venv activate script. |
||||
# shellcheck source=/dev/null |
||||
source "${XDS_K8S_DRIVER_VENV_DIR}/bin/activate" |
||||
else |
||||
echo "Missing python virtual environment directory: ${XDS_K8S_DRIVER_VENV_DIR}" >&2 |
||||
echo "Follow README.md installation steps first." >&2 |
||||
exit 1 |
||||
fi |
||||
fi |
||||
|
||||
cd "${XDS_K8S_DRIVER_DIR}" |
||||
export PYTHONPATH="${XDS_K8S_DRIVER_DIR}" |
||||
exec python "$@" --flagfile="${XDS_K8S_CONFIG}" |
||||
Loading…
Reference in new issue