You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

334 lines
9.1 KiB

#! /bin/sh
# vim:et:ft=sh:sts=2:sw=2
#
# shFlags unit test for the flag definition methods
#
# Copyright 2008-2020 Kate Ward. All Rights Reserved.
# Released under the Apache 2.0 license.
#
# Author: kate.ward@forestent.com (Kate Ward)
# https://github.com/kward/shflags
#
### ShellCheck (http://www.shellcheck.net/)
# Disable source following.
# shellcheck disable=SC1090,SC1091
# TODO(kward): assert on FLAGS errors
# TODO(kward): testNonStandardIFS()
# These variables will be overridden by the test helpers.
stdoutF="${TMPDIR:-/tmp}/STDOUT"
stderrF="${TMPDIR:-/tmp}/STDERR"
# Load test helpers.
. ./shflags_test_helpers
testGetoptStandard() {
if ! _flags_getoptStandard '-b' >"${stdoutF}" 2>"${stderrF}"; then
fail "error parsing -b flag"
_showTestOutput
fi
if _flags_getoptStandard '-x' >"${stdoutF}" 2>"${stderrF}"; then
fail "expected error parsing invalid -x flag"
_showTestOutput
fi
}
testGetoptEnhanced() {
if ! flags_getoptIsEnh; then
return
fi
if ! _flags_getoptEnhanced '-b' >"${stdoutF}" 2>"${stderrF}"; then
fail "error parsing -b flag"
_showTestOutput
fi
if ! _flags_getoptEnhanced '--bool' >"${stdoutF}" 2>"${stderrF}"; then
fail "error parsing --bool flag"
_showTestOutput
fi
if _flags_getoptEnhanced '-x' >"${stdoutF}" 2>"${stderrF}"; then
fail "expected error parsing invalid -x flag"
_showTestOutput
fi
if _flags_getoptEnhanced '--xyz' >"${stdoutF}" 2>"${stderrF}"; then
fail "expected error parsing invalid --xyz flag"
_showTestOutput
fi
}
testValidBoolsShort() {
desc='bool_true_arg'
if FLAGS -b >"${stdoutF}" 2>"${stderrF}"; then
assertTrue "${desc}: expected true value" "${FLAGS_bool:-}"
else
fail "${desc}: FLAGS returned a non-zero result ($?)"
fi
th_showOutput
desc='bool2_defined'
DEFINE_boolean bool2 true '2nd boolean' B
if FLAGS >"${stdoutF}" 2>"${stderrF}"; then
assertTrue "${desc}: expected true value" "${FLAGS_bool2:-}"
else
fail "${desc}: FLAGS returned a non-zero result ($?)"
fi
th_showOutput
desc='bool_false_arg'
if FLAGS -B >"${stdoutF}" 2>"${stderrF}"; then
assertFalse "${desc}: expected false value" "${FLAGS_bool2:-}"
else
fail "${desc}: FLAGS returned a non-zero result ($?)"
fi
th_showOutput
}
# TODO(kate): separate into multiple functions to reflect correct usage
testValidBoolsLong() {
flags_getoptIsEnh
[ $? -eq "${FLAGS_FALSE}" ] && return
# Note: the default value of bool is 'false'.
# Leave flag false.
FLAGS --nobool >"${stdoutF}" 2>"${stderrF}"
r3turn=$?
assertTrue "FLAGS returned a non-zero result (${r3turn})" ${r3turn}
assertFalse '--noXX flag resulted in true value.' "${FLAGS_bool:-}"
assertFalse 'expected no output to STDERR' "[ -s '${stderrF}' ]"
th_showOutput ${r3turn} "${stdoutF}" "${stderrF}"
# Flip flag true.
FLAGS --bool >"${stdoutF}" 2>"${stderrF}"
r3turn=$?
assertTrue "FLAGS returned a non-zero result (${r3turn})" ${r3turn}
assertTrue '--XX flag resulted in false value.' "${FLAGS_bool:-}"
assertFalse 'expected no output to STDERR' "[ -s '${stderrF}' ]"
th_showOutput ${r3turn} "${stdoutF}" "${stderrF}"
# Flip flag back false.
FLAGS --nobool >"${stdoutF}" 2>"${stderrF}"
r3turn=$?
assertTrue "FLAGS returned a non-zero result (${r3turn})" ${r3turn}
assertFalse '--noXX flag resulted in true value.' "${FLAGS_bool:-}"
assertFalse 'expected no output to STDERR' "[ -s '${stderrF}' ]"
th_showOutput ${r3turn} "${stdoutF}" "${stderrF}"
}
testFloats() {
_testFloats '-f'
if flags_getoptIsEnh; then
_testFloats '--float'
fi
}
_testFloats() {
flag=$1
for value in ${TH_FLOAT_VALID}; do
if FLAGS "${flag}" "${value}" >"${stdoutF}" 2>"${stderrF}"; then
# shellcheck disable=SC2154
assertEquals "${flag}: incorrect value" "${value}" "${FLAGS_float}"
else
fail "${flag}: unexpected non-zero result ($?)"
th_showOutput
fi
done
for value in ${TH_FLOAT_INVALID}; do
# Wrap FLAGS in () to catch errors.
if (FLAGS "${flag}" "${value}" >"${stdoutF}" 2>"${stderrF}"); then
fail "${flag}: expected a non-zero result"
th_showOutput
else
assertEquals "${flag}: expected an error" $? "${FLAGS_ERROR}"
fi
done
}
testIntegers() {
_testIntegers '-i'
if flags_getoptIsEnh; then
_testIntegers '--int'
fi
}
_testIntegers() {
flag=$1
for value in ${TH_INT_VALID}; do
if FLAGS "${flag}" "${value}" >"${stdoutF}" 2>"${stderrF}"; then
# shellcheck disable=SC2154
assertEquals "${flag}: incorrect value" "${value}" "${FLAGS_int}"
else
fail "${flag}: unexpected non-zero result ($?)"
th_showOutput
fi
done
for value in ${TH_INT_INVALID}; do
# Wrap FLAGS in () to catch errors.
if (FLAGS "${flag}" "${value}" >"${stdoutF}" 2>"${stderrF}"); then
fail "${flag}: expected a non-zero result"
th_showOutput
else
assertEquals "${flag}: expected an error" $? "${FLAGS_ERROR}"
fi
done
}
testStrings() {
_testStrings 'std_single_word' -s single_word
if flags_getoptIsEnh; then
_testStrings 'enh_single_word' --str single_word
_testStrings 'enh_multi_word' --str 'string with spaces'
fi
}
_testStrings() {
desc=$1
flag=$2
value=$3
if FLAGS "${flag}" "${value}" >"${stdoutF}" 2>"${stderrF}"; then
# shellcheck disable=SC2154
assertEquals "${desc}: incorrect value" "${value}" "${FLAGS_str}"
else
fail "${desc}: unexpected non-zero result ($?)"
# Validate that an error is thrown for unsupported getopt uses.
assertFatalMsg '.* spaces in options'
th_showOutput
fi
}
testMultipleFlags() {
_testMultipleFlags '-b' '-i' '-f' '-s'
flags_getoptIsEnh
[ $? -eq "${FLAGS_FALSE}" ] && return
_testMultipleFlags '--bool' '--int' '--float' '--str'
}
_testMultipleFlags() {
boolFlag=$1
intFlag=$2
floatFlag=$3
strFlag=$4
FLAGS \
"${boolFlag}" \
"${intFlag}" 567 \
"${floatFlag}" 123.45678 \
"${strFlag}" 'some_string' \
>"${stdoutF}" 2>"${stderrF}"
r3turn=$?
assertTrue "use of multiple flags returned a non-zero result" ${r3turn}
assertTrue 'boolean test failed.' "${FLAGS_bool}"
assertNotSame 'float test failed.' 0 "${FLAGS_float}"
assertNotSame 'integer test failed.' 0 "${FLAGS_int}"
assertNotSame 'string test failed.' '' "${FLAGS_str}"
assertFalse 'expected no output to STDERR' "[ -s '${stderrF}' ]"
th_showOutput ${r3turn} "${stdoutF}" "${stderrF}"
}
_testNonFlagArgs() {
argc=$1
shift
FLAGS "$@" >"${stdoutF}" 2>"${stderrF}"
r3turn=$?
assertTrue 'parse returned non-zero value.' ${r3turn}
th_showOutput ${r3turn} "${stdoutF}" "${stderrF}"
eval set -- "${FLAGS_ARGV}"
assertEquals 'wrong count of argv arguments returned.' "${argc}" $#
}
testSingleNonFlagArg() { _testNonFlagArgs 1 argOne; }
testMultipleNonFlagArgs() { _testNonFlagArgs 3 argOne argTwo arg3; }
testMultipleNonFlagStringArgsWithSpaces() {
if flags_getoptIsEnh; then
_testNonFlagArgs 3 argOne 'arg two' arg3
fi
}
testFlagsWithEquals() {
if ! flags_getoptIsEnh; then
return
fi
FLAGS --str='str_flag' 'non_flag' >"${stdoutF}" 2>"${stderrF}"
assertTrue 'FLAGS returned a non-zero result' $?
assertEquals 'string flag not set properly' 'str_flag' "${FLAGS_str}"
th_showOutput ${r3turn} "${stdoutF}" "${stderrF}"
eval set -- "${FLAGS_ARGV}"
assertEquals 'wrong count of argv arguments returned.' 1 $#
}
testComplicatedCommandLineStandard() {
flags_getoptIsEnh
[ $? -eq "${FLAGS_TRUE}" ] && return
# Note: standard getopt stops parsing after first non-flag argument, which
# results in the remaining flags being treated as arguments instead.
FLAGS -i 1 non_flag_1 -s 'two' non_flag_2 -f 3 non_flag_3 \
>"${stdoutF}" 2>"${stderrF}"
r3turn=$?
assertTrue 'FLAGS returned a non-zero result' ${r3turn}
assertEquals 'failed int test' 1 "${FLAGS_int}"
th_showOutput ${r3turn} "${stdoutF}" "${stderrF}"
eval set -- "${FLAGS_ARGV}"
assertEquals 'incorrect number of argv values' 7 $#
}
testComplicatedCommandLineEnhanced() {
flags_getoptIsEnh
[ $? -eq "${FLAGS_FALSE}" ] && return
FLAGS -i 1 non_flag_1 --str='two' non_flag_2 --float 3 'non flag 3' \
>"${stdoutF}" 2>"${stderrF}"
r3turn=$?
assertTrue 'FLAGS returned a non-zero result' ${r3turn}
assertEquals 'failed int test' 1 "${FLAGS_int}"
assertEquals 'failed str test' 'two' "${FLAGS_str}"
assertEquals 'failed float test' 3 "${FLAGS_float}"
th_showOutput ${r3turn} "${stdoutF}" "${stderrF}"
eval set -- "${FLAGS_ARGV}"
assertEquals 'incorrect number of argv values' 3 $#
}
oneTimeSetUp() {
th_oneTimeSetUp
if flags_getoptIsStd; then
th_warn 'Standard version of getopt found. Enhanced tests will be skipped.'
else
th_warn 'Enhanced version of getopt found. Standard tests will be skipped.'
fi
}
setUp() {
DEFINE_boolean bool false 'boolean test' 'b'
DEFINE_float float 0.0 'float test' 'f'
DEFINE_integer int 0 'integer test' 'i'
DEFINE_string str '' 'string test' 's'
}
tearDown() {
flags_reset
}
# showTestOutput for the most recently run test.
_showTestOutput() { th_showOutput "${SHUNIT_FALSE}" "${stdoutF}" "${stderrF}"; }
# Load and run shUnit2.
# shellcheck disable=SC2034
[ -n "${ZSH_VERSION:-}" ] && SHUNIT_PARENT=$0
. "${TH_SHUNIT}"