set -Eo pipefail

# copy FDs
exec 5<&0
exec 6>&1
exec 7>&2

test_num=0
test_count=$(@GREP@ -c "begin_test" "$0")
echo "1..$test_count"

do_clean=yes
test_helper_tmpdirs_=()
trap test_helper_cleanup_ EXIT
script_error_disable_=
trap script_error ERR

script_error() {
    if [ -z "$script_error_disable_" ]; then
	n="${#FUNCNAME[@]}"
	test_err="${test_err}Error in bash, callstack:
"
	let n--;
	for ((; n>0; --n)); do
	    let m=n-1
	    test_err="${test_err} $n:  ${BASH_SOURCE[$n]}:${BASH_LINENO[$m]} in ${FUNCNAME[$n]}()
"
	done
    fi
}

maketmpdir() {
    local tmp
    tmp=$(@MKTEMP@ -d 2>/dev/null || mktemp -d -t 'mytmpdir')
    test_helper_tmpdirs_+=($tmp)
    eval $1=$tmp
}

test_helper_cleanup_() {
    if test x"$do_clean" == x"yes"; then
	for dir in "${test_helper_tmpdirs_[@]}"; do
	    echo "Removing $dir"
	    rm -rf "$dir"
	done
    else
	test_name=$(basename $0)
	for dir in "${test_helper_tmpdirs_[@]}"; do
	    tgt=test_failures/$test_name
	    echo "Copying $dir to $tgt"
	    rm -rf $tgt
	    mkdir -p $tgt
	    mv "$dir"/* $tgt
	    rm -rf "$dir"
	done
    fi
}

capture_stdout() {
    echo "----- Running \"$*\""
    script_error_disable_=1
    output=$(eval "$SINA_TEST_WRAPPER $*" | @TEE@ /dev/fd/6)
    exit_code=$?
    script_error_disable_=
    echo "----- Finished with exit code $exit_code (cmd: \"$*\""
}
capture_stderr() {
    echo "----- Running \"$*\""
    script_error_disable_=1
    output=$(eval "$SINA_TEST_WRAPPER $*" 2>&1 >&6 | @TEE@ /dev/fd/7)
    exit_code=$?
    script_error_disable_=
    echo "----- Finished with exit code $exit_code (cmd: \"$*\""
}
capture_stdouterr() {
    echo "----- Running \"$*\""
    script_error_disable_=1
    output=$(eval "$SINA_TEST_WRAPPER $*" 2>&1 | @TEE@ /dev/fd/6)
    exit_code=$?
    script_error_disable_=
    echo "----- Finished with exit code $exit_code (cmd: \"$*\""
}

begin_test() {
    let ++test_num
    test_name=$1
    test_err=
    test_var=
    echo ""
    echo "$test_num $test_name"
    echo "$test_num $test_name" | sed 's/./=/g'
    echo ""
}

end_test() {
    if [ -n "$test_err" ]; then
	do_clean=no
	echo -n "not "
    fi
    echo "ok $test_num - $test_name"
    if [ -n "$test_var" ]; then
	echo "#$test_var"
    fi
    if [ -n "$test_err" ]; then
	echo "--- BEGIN ERR ---"
	echo "$test_err"
	echo "--- END ERR ---"
    fi
}

assert_exit_success() {
    if [ -n "$1" ]; then
	capture_stdouterr "$@"
    fi
    if [ $exit_code -ne 0 ]; then
	test_err="${test_err}# command exited with $exit_code
"
    fi
}

assert_exit_failure() {
    if [ -n "$1" ]; then
	capture_stdouterr "$@"
    fi
    if [ $exit_code -eq 0 ]; then
	test_err="${test_err}# command exited with $exit_code
"
    fi
}

assert_output_contains() {
    if echo "$output" | @GREP@ "$1" >/dev/null; then
	:
    else
	test_err="${test_err}# command output did not include '$1'
OUTPUT_BEGIN
$output
OUTPUT_END
"
    fi
}

assert_output_not_contains() {
    if echo "$output" | @GREP@ "$1" >/dev/null; then
	test_err="${test_err}# command should not have included '$1'
OUTPUT_BEGIN
$output
OUTPUT_END
"
    fi
}

assert_output_count() {
    if test $(echo "$output" | @GREP@ -c "$1") $2 $3; then
	:
    else
	test_err="${test_err}# command output did not match $@
"
    fi
}

assert_output_value() {
    # convert 8.4e-10 to 8.4*10^-10 (from https://stackoverflow.com/questions/12882611)
    snot2bc='s/([+-]?[0-9.]+)[eE]\+?(-?)([0-9]+)/(\1*10^\2\3)/g'
    var="$1"
    shift
    val=$(echo "$output" | @SED@ -n 's/.*'$var'//p' | @SED@ -E "$snot2bc")
    test_var="${test_var} $var$val"
    while test $# -gt 0; do
	if (( $(echo "$val $1" | @BC@) )); then
	    :
	else
	    test_err="${test_err}# value for key $var ($val) did not match $1
"
	fi
	shift
    done
}