1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
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
}
|
