load 'test_helper/bats-support/load'
load 'test_helper/bats-assert/load'

NAME=tvial/docker-mailserver:testing

# default timeout is 120 seconds
TEST_TIMEOUT_IN_SECONDS=${TEST_TIMEOUT_IN_SECONDS-120}
NUMBER_OF_LOG_LINES=${NUMBER_OF_LOG_LINES-10}

# @param $1 timeout
# @param --fatal-test <command eval string> additional test whose failure aborts immediately
# @param ... test to run
function repeat_until_success_or_timeout {
    local fatal_failure_test_command
    if [[ "$1" == "--fatal-test" ]]; then
        fatal_failure_test_command="$2"
        shift 2
    fi
    if ! [[ "$1" =~ ^[0-9]+$ ]]; then
        echo "First parameter for timeout must be an integer, recieved \"$1\""
        return 1
    fi
    TIMEOUT=$1
    STARTTIME=$SECONDS
    shift 1
    until "$@"
    do
        if [[ -n "$fatal_failure_test_command" ]] && ! eval "$fatal_failure_test_command"; then
            echo "\`$fatal_failure_test_command\` failed, early aborting repeat_until_success of \`$*\`" >&2
            exit 1
        fi
        sleep 5
        if [[ $(( SECONDS - STARTTIME )) -gt $TIMEOUT ]]; then
            echo "Timed out on command: $*" >&2
            return 1
        fi
    done
}

# like repeat_until_success_or_timeout but with wrapping the command to run into `run` for later bats consumption
# @param $1 timeout
# @param ... test command to run
function run_until_success_or_timeout {
    if ! [[ "$1" =~ ^[0-9]+$ ]]; then
        echo "First parameter for timeout must be an integer, recieved \"$1\""
        return 1
    fi
    TIMEOUT=$1
    STARTTIME=$SECONDS
    shift 1
    until run "$@" && [[ $status -eq 0 ]]
    do
        sleep 1
        if [[ $(( SECONDS - STARTTIME )) -gt $TIMEOUT ]]; then
            echo "Timed out on command: $*" >&2
            return 1
        fi
    done
}

# @param $1 timeout
# @param $2 container name
# @param ... test command for container
function repeat_in_container_until_success_or_timeout() {
    timeout="$1"
    container_name="$2"
    shift 2
    repeat_until_success_or_timeout --fatal-test "container_is_running $container_name" "$timeout" docker exec "$container_name" "$@"
}

function container_is_running() {
    [[ "$(docker inspect -f '{{.State.Running}}' "$1")" == "true" ]]
}

# @param $1 port
# @param $2 container name
function wait_for_tcp_port_in_container() {
    repeat_until_success_or_timeout --fatal-test "container_is_running $2" "$TEST_TIMEOUT_IN_SECONDS" docker exec $2 /bin/sh -c "nc -z 0.0.0.0 $1"
}

# @param $1 name of the postfix container
function wait_for_smtp_port_in_container() {
    wait_for_tcp_port_in_container 25 "$1"
}

# @param $1 name of the postfix container
function wait_for_amavis_port_in_container() {
    wait_for_tcp_port_in_container 10024 "$1"
}

# @param $1 name of the postfix container
function wait_for_finished_setup_in_container() {
    local status=0
    repeat_until_success_or_timeout --fatal-test "container_is_running $1" "$TEST_TIMEOUT_IN_SECONDS" sh -c "docker logs $1 | grep 'is up and running'" || status=1
    if [[ $status -eq 1 ]]; then
        echo "Last $NUMBER_OF_LOG_LINES lines of container \`$1\`'s log"
        docker logs "$1" | tail -n "$NUMBER_OF_LOG_LINES"
    fi
    return $status
}

SETUP_FILE_MARKER="$BATS_TMPDIR/$(basename "$BATS_TEST_FILENAME").setup_file"

function native_setup_teardown_file_support() {
    VERSION_REGEX='([0-9]+)\.([0-9]+)\.([0-9]+)'
    # bats versions that support setup_file out of the box don't need this
    if [[ "$BATS_VERSION" =~ $VERSION_REGEX ]]; then
        numeric_version=$(( (BASH_REMATCH[1] * 100 + BASH_REMATCH[2]) * 100 + BASH_REMATCH[3] ))
        if [[ $numeric_version -ge 10201 ]]; then
            if [ "$BATS_TEST_NAME" == 'test_first' ]; then
                skip 'This version natively supports setup/teardown_file'
            fi
            return 0
        fi
    fi
    return 1
}

# use in setup() in conjunction with a `@test "first" {}` to trigger setup_file reliably
function run_setup_file_if_necessary() {
    native_setup_teardown_file_support && return 0
    if [ "$BATS_TEST_NAME" == 'test_first' ]; then
        # prevent old markers from marking success or get an error if we cannot remove due to permissions
        rm -f "$SETUP_FILE_MARKER"

        setup_file

        touch "$SETUP_FILE_MARKER"
    else
        if [ ! -f "$SETUP_FILE_MARKER" ]; then
            skip "setup_file failed"
            return 1
        fi
    fi
}

# use in teardown() in conjunction with a `@test "last" {}` to trigger teardown_file reliably
function run_teardown_file_if_necessary() {
    native_setup_teardown_file_support && return 0
    if [ "$BATS_TEST_NAME" == 'test_last' ]; then
        # cleanup setup file marker
        rm -f "$SETUP_FILE_MARKER"
        teardown_file
    fi
}

# get the private config path for the given container or test file, if no container name was given
function private_config_path() {
    echo "$PWD/test/duplicate_configs/${1:-$(basename "$BATS_TEST_FILENAME")}"
}

# @param $1 relative source in test/config folder
# @param $2 (optional) container name, defaults to $BATS_TEST_FILENAME
# @return path to the folder where the config is duplicated
function duplicate_config_for_container() {
    output="$(private_config_path "$2")"
    rm -rf "${output:?}/" # cleanup
    mkdir -p "$output"
    cp -r "$PWD/test/config/${1:?}/." "$output"
    echo "$output"
}

function container_has_service_running() {
    containerName="$1"
    serviceName="$2"
    docker exec "$containerName" /usr/bin/supervisorctl status "$serviceName" | grep RUNNING >/dev/null
}

function wait_for_service() {
    containerName="$1"
    serviceName="$2"
    repeat_until_success_or_timeout --fatal-test "container_is_running $containerName" 60 \
        container_has_service_running "$containerName" "$serviceName"
}

function count_processed_changes() {
    containerName=$1
    docker exec "$containerName" cat /var/log/supervisor/changedetector.log | grep "Change detected" -c \
        || [[ $? == 1 ]] # don't error when no matches were found
}


function wait_for_changes_to_be_detected_in_container() {
    containerName="$1"
    timeout=${TEST_TIMEOUT_IN_SECONDS}
    repeat_in_container_until_success_or_timeout "$timeout" "$containerName" \
        bash -c 'source /usr/local/bin/helper_functions.sh; cmp --silent -- <(_monitored_files_checksums) "$CHKSUM_FILE"  >/dev/null'
}