shpy-shunit2

#!/usr/bin/env sh

assertCallCount() {
    if [ $# -lt 2 ] || [ $# -gt 3 ]; then
        echo 'Usage: assertCallCount [message] spy count'
        echo
        echo 'Assert the number of times the spy was invoked'
        return 1
    fi

    if [ $# -eq 2 ]; then
        assertEquals "$2" "$(getSpyCallCount "$1")"
    else
        assertEquals "$1" "$3" "$(getSpyCallCount "$2")"
    fi
}

assertCalledWith() {
    if [ $# -lt 1 ]; then
        echo 'Usage: assertCalledWith spy [arg ...]'
        echo
        echo 'Assert the arguments for the first invocation of the spy'
        echo
        echo 'Subsequent calls will assert the second invocation, etc'
        return 1
    fi

    # shellcheck disable=SC3043
    local status_code actual_call current_spy_call

    wasSpyCalledWith "$@"
    status_code=$?

    actual_call=$1
    if [ $status_code -ne 0 ]; then
        current_spy_call=$(( $(_shpyGetCurrentSpyCallIndex "$1") + 1 ))
        if ! actual_call="$actual_call $(getArgsForCall "$1" "$current_spy_call")" 2>/dev/null; then
            actual_call='not called'
        fi
    fi

    examineNextSpyCall "$1"

    assertTrue "expected:<$*> but was:<$actual_call>" $status_code
}

assertCalledWith_() {
    if [ $# -lt 2 ]; then
        echo 'Usage: assertCalledWith_ message spy [arg ...]'
        echo
        echo 'Assert the arguments for the first invocation of the spy'
        echo 'When the assertion fails, the custom error message is used'
        echo
        echo 'Subsequent calls will assert the second invocation, etc'
        return 1
    fi

    # shellcheck disable=SC3043
    local message status_code

    message=$1
    shift

    wasSpyCalledWith "$@"
    status_code=$?

    examineNextSpyCall "$1"

    assertTrue "$message" $status_code

}

assertCalledOnceWith() {
    if [ $# -lt 1 ]; then
        echo 'Usage: assertCalledOnceWith spy [arg ...]'
        echo
        echo 'Assert the arguments of the first and only invocation of the spy'
        return 1
    fi

    assertCallCount "$1" 1 && assertCalledWith "$@"
}

assertCalledOnceWith_() {
    if [ $# -lt 2 ]; then
        echo 'Usage: assertCalledOnceWith_ message spy [arg ...]'
        echo
        echo 'Assert the arguments of the first and only invocation of the spy'
        echo 'When the assertion fails, the custom error message is used'
        return 1
    fi

    assertCallCount "$1" "$2" 1 && assertCalledWith_ "$@"
}

assertNeverCalled() {
    if [ $# -lt 1 ] || [ $# -gt 2 ]; then
        echo 'Usage: assertNeverCalled [message] spy'
        echo
        echo 'Assert the spy was never invoked with an optional error message'
        return 1
    fi

    assertCallCount "$@" 0
}