mirror of
https://github.com/troglobit/finit.git
synced 2026-10-02 05:52:48 +07:00
145 lines
3.6 KiB
Bash
Executable File
145 lines
3.6 KiB
Bash
Executable File
#!/bin/sh
|
|
# Small framework for executing tests in a test environment.
|
|
#
|
|
# Copyright (c) 2021 Jacques de Laval <jacques@de-laval.se>
|
|
#
|
|
# Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
# of this software and associated documentation files (the "Software"), to deal
|
|
# in the Software without restriction, including without limitation the rights
|
|
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
# copies of the Software, and to permit persons to whom the Software is
|
|
# furnished to do so, subject to the following conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be included in
|
|
# all copies or substantial portions of the Software.
|
|
#
|
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
# THE SOFTWARE.
|
|
|
|
assert_num_children() {
|
|
assert "$1 services are running" "$(texec pgrep -P 1 "$2" | wc -l)" -eq "$1"
|
|
}
|
|
|
|
texec() {
|
|
# shellcheck disable=SC2154
|
|
"$TEST_DIR/tenv/exec.sh" "$finit_pid" "$@"
|
|
}
|
|
|
|
pause() {
|
|
echo "Press any key to continue... "
|
|
read -r REPLY
|
|
}
|
|
|
|
toggle_finit_debug() {
|
|
say 'Toggle finit debug'
|
|
texec initctl debug
|
|
sleep 0.5
|
|
}
|
|
|
|
color_reset='\e[0m'
|
|
fg_red='\e[1;31m'
|
|
fg_green='\e[1;32m'
|
|
fg_yellow='\e[1;33m'
|
|
log() {
|
|
printf "< %s > %b%b%b %s\n" "$0" "$1" "$2" "$color_reset" "$3"
|
|
}
|
|
|
|
assert() {
|
|
__assert_msg=$1
|
|
shift
|
|
if [ ! "$@" ]; then
|
|
log "$fg_red" ✘ "$__assert_msg ($*)"
|
|
return 1
|
|
fi
|
|
log "$fg_green" ✔ "$__assert_msg"
|
|
}
|
|
|
|
retry() {
|
|
__retry_cmd=$1
|
|
shift
|
|
case "$#" in
|
|
2)
|
|
__retry_n=$1
|
|
__retry_sleep=$2
|
|
;;
|
|
1)
|
|
__retry_n=$1
|
|
__retry_sleep=0.1
|
|
;;
|
|
*)
|
|
__retry_n=50
|
|
__retry_sleep=0.1
|
|
;;
|
|
esac
|
|
|
|
for _ in $(seq 1 "$__retry_n"); do
|
|
sleep "$__retry_sleep"
|
|
__retry_cmd_out=$(eval "$__retry_cmd") && \
|
|
echo "$__retry_cmd_out" && \
|
|
return 0
|
|
done
|
|
__retry_cmd_exit="$?"
|
|
echo "$__retry_cmd_out"
|
|
return "$__retry_cmd_exit"
|
|
}
|
|
|
|
say() {
|
|
log "$fg_yellow" • "$@"
|
|
}
|
|
|
|
teardown() {
|
|
test_status="$?"
|
|
|
|
if type test_teardown > /dev/null 2>&1 ; then
|
|
test_teardown
|
|
fi
|
|
|
|
log "$color_reset" '--' ''
|
|
if [ "$test_status" -eq 0 ]; then
|
|
log "$fg_green" 'TEST PASS' ''
|
|
else
|
|
log "$fg_red" 'TEST FAIL' ''
|
|
fi
|
|
if [ -n "${finit_pid+x}" ]; then
|
|
texec kill -SIGUSR2 1
|
|
fi
|
|
|
|
wait
|
|
|
|
if [ -d "$TENV_ROOT/var/lock" ]; then
|
|
chmod +r "$TENV_ROOT/var/lock"
|
|
fi
|
|
rm -f "$TENV_ROOT"/running_test.pid
|
|
}
|
|
|
|
trap teardown EXIT
|
|
|
|
TENV_ROOT="${TENV_ROOT:-$(pwd)/${TEST_DIR}/tenv-root}"
|
|
export TENV_ROOT
|
|
|
|
# shellcheck source=/dev/null
|
|
. "$TENV_ROOT/../test.env"
|
|
|
|
# Setup test environment
|
|
if [ -n "${DEBUG:-}" ]; then
|
|
FINIT_ARGS="${FINIT_ARGS:-} finit.debug=on"
|
|
fi
|
|
# shellcheck disable=2086
|
|
"$TEST_DIR/tenv/start.sh" finit ${FINIT_ARGS:-} &
|
|
finit_ppid=$!
|
|
echo "$finit_ppid" > "$TENV_ROOT"/running_test.pid
|
|
|
|
>&2 echo "Hint: Execute 'TENV_ROOT=$TENV_ROOT $TEST_DIR/tenv/enter.sh' to enter the test namespace"
|
|
log "$color_reset" 'Setup of test environment done' ''
|
|
|
|
finit_pid=$(retry "pgrep -P $finit_ppid")
|
|
|
|
tty=/dev/$(texec cat /sys/class/tty/console/active)
|
|
texec cat "$tty" &
|
|
sleep 1
|