1 files changed, 85 insertions, 0 deletions
diff --git a/examples/shellmath/assert.sh b/examples/shellmath/assert.sh
new file mode 100644
index 0000000..bc4122e
--- /dev/null
+++ b/examples/shellmath/assert.sh
@@ -0,0 +1,85 @@
+#!/bin/env bash
+###############################################################################
+# Internal test engine functions
+###############################################################################
+
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+NO_COLOR='\033[0m'
+
+function _shellmath_assert_returnCode()
+{
+    _shellmath_assert_functionReturn -c "$@"
+    return $?
+}
+
+function _shellmath_assert_returnString()
+{
+    _shellmath_assert_functionReturn "$@"
+    return $?
+}
+
+function _shellmath_assert_functionReturn()
+{
+    if [[ $# -lt 2 ]]; then
+        echo "USAGE: ${FUNCNAME[0]} [-c] returnStringOrCode functionName [ functionArgs ... ]"
+        echo "    By default, asserts against the string output by the function."
+        echo "    Use -c to assert against the numeric return code instead."
+        return "${__shellmath_returnCodes[FAIL]}"
+    fi
+
+    if [[ "${1,,}" == '-c' ]]; then
+        mode=RETURN_CODE
+        shift
+    else
+        mode=RETURN_STRING
+    fi
+
+    expectedReturn="$1"
+    func="$2"
+    shift 2
+
+    args=("$@")
+
+    # Exercise the function in optimized mode; it will run faster by avoiding
+    # subshelling. This also suppresses dumping of function output to stdout.
+    __shellmath_isOptimized=${__shellmath_true}
+    "$func" "${args[@]}"
+    returnCode=$?
+    __shellmath_isOptimized=${__shellmath_false}
+
+    # Fetch the return value(s)
+    local numReturnValues
+    declare -a actualReturn
+    _shellmath_getReturnValueCount numReturnValues
+    if ((numReturnValues == 1)); then
+        _shellmath_getReturnValue actualReturn[0]
+    else
+        # Multiple returns? Join them into one string
+        local _i evalString="_shellmath_getReturnValues"
+        for ((_i=0; _i<numReturnValues; _i++)); do
+            evalString+=" actualReturn[$_i]"
+        done
+        eval "$evalString"
+    fi
+
+    if [[ $mode == RETURN_STRING ]]; then
+        if [[ "${actualReturn[*]}" == "$expectedReturn" ]]; then
+            _shellmath_setReturnValue  "${GREEN}ok${NO_COLOR}   "
+            return "$__shellmath_SUCCESS"
+        else
+            _shellmath_setReturnValue "${RED}FAIL${NO_COLOR} (${actualReturn[*]}) "
+            return "$__shellmath_FAIL"
+        fi
+    elif [[ $mode == RETURN_CODE ]]; then
+        if [[ "$returnCode" == "$expectedReturn" ]]; then
+            _shellmath_setReturnValue  "${GREEN}ok${NO_COLOR}   "
+            return "$__shellmath_SUCCESS"
+        else
+            _shellmath_setReturnValue "${RED}FAIL${NO_COLOR} ($returnCode) "
+            return "$__shellmath_FAIL"
+        fi
+    fi
+
+}
+