126 строки
3.8 KiB
Bash
126 строки
3.8 KiB
Bash
#!/usr/bin/env bash
|
|
#
|
|
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you 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_SHORT_OPTIONS="
|
|
h i x s v f:"
|
|
|
|
_RUN_TESTS_LONG_OPTIONS="
|
|
help with-db-init with-xunit nocapture verbose xunit-file:
|
|
"
|
|
|
|
# Note on OSX bash has no associative arrays (Bash 3.2) so we have to fake it
|
|
|
|
_RUN_TESTS_KNOWN_VALUES=""
|
|
|
|
function _get_known_values_run_tests {
|
|
case "$1" in
|
|
*)
|
|
_RUN_TESTS_KNOWN_VALUES=""
|
|
esac
|
|
}
|
|
|
|
_RUN_TESTS_GETOPT_SHORT_OPTIONS=""
|
|
_RUN_TESTS_GETOPT_LONG_OPTIONS=""
|
|
|
|
function _build_options_run_tests {
|
|
local SEPARATOR=""
|
|
local OPTION
|
|
|
|
for OPTION in ${_RUN_TESTS_SHORT_OPTIONS}
|
|
do
|
|
_RUN_TESTS_GETOPT_SHORT_OPTIONS="${_RUN_TESTS_GETOPT_SHORT_OPTIONS}${SEPARATOR}${OPTION}"
|
|
SEPARATOR=","
|
|
done
|
|
|
|
SEPARATOR=""
|
|
for OPTION in ${_RUN_TESTS_LONG_OPTIONS}
|
|
do
|
|
_RUN_TESTS_GETOPT_LONG_OPTIONS="${_RUN_TESTS_GETOPT_LONG_OPTIONS}${SEPARATOR}${OPTION}"
|
|
SEPARATOR=","
|
|
done
|
|
}
|
|
|
|
function _listcontains_run_tests {
|
|
local WORD
|
|
for WORD in $1; do
|
|
[[ ${WORD} = "$2" ]] && return 0
|
|
done
|
|
return 1
|
|
}
|
|
|
|
# A completion function for run_tests
|
|
function _comp_run_tests {
|
|
local ALL_OPTIONS=""
|
|
local EXTRA_ARG_OPTIONS=""
|
|
local OPTION
|
|
local GETOPT_OPTION
|
|
local LAST_COMMAND_PREFIX
|
|
local PREVIOUS_COMMAND
|
|
local ALL_TESTS_FILE="${HOME}/all_tests.txt"
|
|
|
|
for OPTION in ${_RUN_TESTS_SHORT_OPTIONS}
|
|
do
|
|
LAST_CHAR="${OPTION:$((${#OPTION}-1)):1}"
|
|
GETOPT_OPTION='-'${OPTION//:/}
|
|
if [[ "${LAST_CHAR}" == ":" ]]; then
|
|
EXTRA_ARG_OPTIONS="${EXTRA_ARG_OPTIONS} ${GETOPT_OPTION}"
|
|
fi
|
|
ALL_OPTIONS="${ALL_OPTIONS} ${GETOPT_OPTION}"
|
|
done
|
|
for OPTION in ${_RUN_TESTS_LONG_OPTIONS}
|
|
do
|
|
LAST_CHAR="${OPTION:$((${#OPTION}-1)):1}"
|
|
GETOPT_OPTION='--'${OPTION//:/}
|
|
ALL_OPTIONS="${ALL_OPTIONS} ${GETOPT_OPTION}"
|
|
if [[ "${LAST_CHAR}" == ":" ]]; then
|
|
EXTRA_ARG_OPTIONS="${EXTRA_ARG_OPTIONS} ${GETOPT_OPTION}"
|
|
fi
|
|
done
|
|
|
|
LAST_COMMAND_PREFIX="${COMP_WORDS[${#COMP_WORDS[@]}-1]}"
|
|
if [[ ${#COMP_WORDS[@]} -gt 1 ]]; then
|
|
PREVIOUS_COMMAND="${COMP_WORDS[${#COMP_WORDS[@]}-2]}"
|
|
else
|
|
PREVIOUS_COMMAND=""
|
|
fi
|
|
|
|
if _listcontains_run_tests "${EXTRA_ARG_OPTIONS}" "${PREVIOUS_COMMAND}"; then
|
|
COMPREPLY=()
|
|
_get_known_values_run_tests "${PREVIOUS_COMMAND}"
|
|
while IFS='' read -r LINE; do COMPREPLY+=("$LINE"); done \
|
|
< <(compgen -W "${_RUN_TESTS_KNOWN_VALUES}" -- "${LAST_COMMAND_PREFIX}")
|
|
else
|
|
COMPREPLY=()
|
|
while IFS='' read -r LINE; do COMPREPLY+=("$LINE"); done \
|
|
< <(compgen -W "${ALL_OPTIONS}" -- "${LAST_COMMAND_PREFIX}")
|
|
|
|
if [[ -f ${ALL_TESTS_FILE} ]]; then
|
|
while IFS='' read -r LINE; do COMPREPLY+=("$LINE"); done \
|
|
< <(compgen -W "$(cat "${ALL_TESTS_FILE}")" -- "${LAST_COMMAND_PREFIX}")
|
|
fi
|
|
fi
|
|
}
|
|
|
|
_build_options_run_tests
|
|
|
|
# allow completion to contain colon (see http://tiswww.case.edu/php/chet/bash/FAQ)
|
|
COMP_WORDBREAKS=${COMP_WORDBREAKS//:}
|
|
|
|
complete -F _comp_run_tests run-tests
|