114 lines
3.5 KiB
Bash
Executable File
114 lines
3.5 KiB
Bash
Executable File
#!/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.
|
|
|
|
set -e
|
|
function usage {
|
|
echo "Usage: ${0} [options] TestSomeTestName [TestOtherTest...]"
|
|
echo ""
|
|
echo " --repeat times number of times to repeat if successful"
|
|
echo " --force-timeout seconds Seconds to wait before killing a given test run"
|
|
echo " --maven-local-repo /path/to/use Path for maven artifacts while building"
|
|
echo " --surefire-fork-count set the fork-count. only useful if multiple " \
|
|
"tests running (default 0.5C)"
|
|
echo " --log-output /path/to/use path to directory to hold attempt log"
|
|
echo " --hadoop-profile profile specify a value passed via -Dhadoop.profile"
|
|
exit 1
|
|
}
|
|
# Get arguments
|
|
declare -i force_timeout=7200
|
|
declare fork_count="0.5C"
|
|
declare -i attempts=1
|
|
declare maven_repo="${HOME}/.m2/repository"
|
|
declare output="."
|
|
declare hadoop_profile
|
|
while [ $# -gt 0 ]
|
|
do
|
|
case "$1" in
|
|
--force-timeout) shift; force_timeout=$1; shift;;
|
|
--maven-local-repo) shift; maven_repo=$1; shift;;
|
|
--repeat) shift; attempts=$1; shift;;
|
|
--log-output) shift; output=$1; shift;;
|
|
--surefire-fork-count) shift; fork_count=$1; shift;;
|
|
--hadoop-profile) shift; hadoop_profile=$1; shift;;
|
|
--) shift; break;;
|
|
-*) usage ;;
|
|
*) break;; # terminate while loop
|
|
esac
|
|
done
|
|
|
|
if [ "$#" -lt 1 ]; then
|
|
usage
|
|
fi
|
|
|
|
function find_modules
|
|
{
|
|
declare testmaybepattern=$1
|
|
declare path
|
|
while IFS= read -r -d $'\0' path; do
|
|
while [ -n "${path}" ]; do
|
|
path=$(dirname "${path}")
|
|
if [ -f "${path}/pom.xml" ]; then
|
|
echo "${path}"
|
|
break
|
|
fi
|
|
done
|
|
done < <(find . -name "${testmaybepattern}.java" -a -type f -a -not -path '*/target/*' -print0)
|
|
}
|
|
|
|
function echo_run_redirect
|
|
{
|
|
declare log=$1
|
|
shift
|
|
echo "${*}" >"${log}"
|
|
"${@}" >>"${log}" 2>&1
|
|
}
|
|
|
|
declare -a modules
|
|
|
|
for test in "${@}"; do
|
|
for module in $(find_modules "${test}"); do
|
|
if [[ ! "${modules[*]}" =~ ${module} ]]; then
|
|
echo "adding module '${module}' to set."
|
|
modules+=("${module}")
|
|
fi
|
|
done
|
|
done
|
|
|
|
declare -a mvn_module_arg
|
|
for module in "${modules[@]}"; do
|
|
mvn_module_arg+=(-pl "${module}")
|
|
done
|
|
|
|
declare tests="${*}"
|
|
declare -a maven_args=('--batch-mode' "-Dmaven.repo.local=${maven_repo}" "-Dtest=${tests// /,}"
|
|
'-Dsurefire.rerunFailingTestsCount=0' "-Dsurefire.parallel.forcedTimeout=${force_timeout}"
|
|
'-Dsurefire.shutdown=kill' '-DtrimStackTrace=false' '-am' "${mvn_module_arg[@]}"
|
|
"-DforkCount=${fork_count}" test)
|
|
if [[ -n "${hadoop_profile}" ]] ; then
|
|
maven_args+=("-Dhadoop.profile=${hadoop_profile}")
|
|
fi
|
|
|
|
if [[ ! -d "${output}" ]] ; then
|
|
mkdir -p "${output}"
|
|
fi
|
|
|
|
for attempt in $(seq "${attempts}"); do
|
|
echo "Attempt ${attempt}" >&2
|
|
echo_run_redirect "${output}/mvn_test.log" mvn "${maven_args[@]}"
|
|
done
|