75 lines
1.7 KiB
Bash
Executable File
75 lines
1.7 KiB
Bash
Executable File
#! /usr/bin/env bash
|
|
#
|
|
# Author: Bert Van Vreckem <bert.vanvreckem@gmail.com>
|
|
#
|
|
# Run BATS test files in the current directory, and the ones in the subdirectory
|
|
# matching the host name.
|
|
#
|
|
# The script installs BATS if needed. It's best to put ${bats_install_dir} in
|
|
# your .gitignore.
|
|
|
|
set -o errexit # abort on nonzero exitstatus
|
|
set -o nounset # abort on unbound variable
|
|
|
|
#{{{ Variables
|
|
|
|
test_dir="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
|
|
|
|
bats_archive="v1.1.0.tar.gz"
|
|
bats_url="https://github.com/bats-core/bats-core/archive/${bats_archive}"
|
|
bats_install_dir="/opt"
|
|
bats="${bats_install_dir}/bats/libexec/bats"
|
|
|
|
test_file_pattern="*.bats"
|
|
|
|
# color definitions
|
|
Blue='\e[0;34m'
|
|
Yellow='\e[0;33m'
|
|
Reset='\e[0m'
|
|
|
|
#}}}
|
|
#{{{ Functions
|
|
|
|
# Usage: install_bats_if_needed
|
|
install_bats_if_needed() {
|
|
pushd "${bats_install_dir}" > /dev/null
|
|
if [[ ! -d "${bats_install_dir}/bats" ]]; then
|
|
wget "${bats_url}"
|
|
tar xf "${bats_archive}"
|
|
mv bats-* bats
|
|
rm "${bats_archive}"
|
|
fi
|
|
popd > /dev/null
|
|
}
|
|
|
|
# find_tests DIR [MAX_DEPTH]
|
|
find_tests() {
|
|
local max_depth=""
|
|
if [ "$#" -eq "2" ]; then
|
|
max_depth="-maxdepth $2"
|
|
fi
|
|
|
|
local tests
|
|
tests=$(find "$1" "${max_depth}" -type f -name "${test_file_pattern}" -printf '%p\n' 2> /dev/null)
|
|
|
|
echo "${tests}"
|
|
}
|
|
#}}}
|
|
# Script proper
|
|
|
|
install_bats_if_needed
|
|
|
|
# List all test cases (i.e. files in the test dir matching the test file
|
|
# pattern)
|
|
# Tests to be run on all hosts
|
|
global_tests=$(find_tests "${test_dir}" 1)
|
|
|
|
# Tests for individual hosts
|
|
host_tests=$(find_tests "${test_dir}/${HOSTNAME}")
|
|
|
|
# Loop over test files
|
|
for test_case in ${global_tests} ${host_tests}; do
|
|
echo -e "${Blue}Running test ${Yellow}${test_case}${Reset}"
|
|
${bats} "${test_case}"
|
|
done
|