t/my-tap-lib.sh

   1 #!/bin/sh
   2 # Copyright (c) 2009, 2010 Eric Wong <normalperson@yhbt.net>
   3 #
   4 # TAP-producing shell library for POSIX-compliant Bourne shells We do
   5 # not _rely_ on Bourne Again features, though we will use "set -o
   6 # pipefail" from ksh93 or bash 3 if available
   7 #
   8 # Only generic, non-project/non-language-specific stuff goes here.  We
   9 # only have POSIX dependencies for the core tests (without --verbose),
  10 # though we'll enable useful non-POSIX things if they're available.
  11 #
  12 # This test library is intentionally unforgiving, it does not support
  13 # skipping tests nor continuing after any failure.  Any failures
  14 # immediately halt execution as do any references to undefined
  15 # variables.
  16 #
  17 # When --verbose is specified, we always prefix stdout/stderr
  18 # output with "#" to avoid confusing TAP consumers.  Otherwise
  19 # the normal stdout/stderr streams are redirected to /dev/null
  20
  21 # dup normal stdout(fd=1) and stderr (fd=2) to fd=3 and fd=4 respectively
  22 # normal TAP output goes to fd=3, nothing should go to fd=4
  23 exec 3>&1 4>&2
  24
  25 # ensure a sane environment
  26 TZ=UTC LC_ALL=C LANG=C
  27 export LANG LC_ALL TZ
  28 unset CDPATH
  29
  30 # pipefail is non-POSIX, but very useful in ksh93/bash
  31 ( set -o pipefail 2>/dev/null ) && set -o pipefail
  32
  33 SED=${SED-sed}
  34
  35 # Unlike other test frameworks, we are unforgiving and bail immediately
  36 # on any failures.  We do this because we're lazy about error handling
  37 # and also because we believe anything broken should not be allowed to
  38 # propagate throughout the rest of the test
  39 set -e
  40 set -u
  41
  42 # name of our test
  43 T=${0##*/}
  44
  45 t_expect_nr=-1
  46 t_nr=0
  47 t_current=
  48 t_complete=false
  49
  50 # list of files to remove unconditionally on exit
  51 T_RM_LIST=
  52
  53 # list of files to remove only on successful exit
  54 T_OK_RM_LIST=
  55
  56 # emit output to stdout, it'll be parsed by the TAP consumer
  57 # so it must be TAP-compliant output
  58 t_echo () {
  59         echo >&3 "$@"
  60 }
  61
  62 # emits non-parsed information to stdout, it will be prefixed with a '#'
  63 # to not throw off TAP consumers
  64 t_info () {
  65         t_echo '#' "$@"
  66 }
  67
  68 # exit with an error and print a diagnostic
  69 die () {
  70         echo >&2 "$@"
  71         exit 1
  72 }
  73
  74 # our at_exit handler, it'll fire for all exits except SIGKILL (unavoidable)
  75 t_at_exit () {
  76         code=$?
  77         set +e
  78         if test $code -eq 0
  79         then
  80                 $t_complete || {
  81                         t_info "t_done not called"
  82                         code=1
  83                 }
  84         elif test -n "$t_current"
  85         then
  86                 t_echo "not ok $t_nr - $t_current"
  87         fi
  88         if test $t_expect_nr -ne -1
  89         then
  90                 test $t_expect_nr -eq $t_nr || {
  91                         t_info "planned $t_expect_nr tests but ran $t_nr"
  92                         test $code -ne 0 || code=1
  93                 }
  94         fi
  95         $t_complete || {
  96                 t_info "unexpected test failure"
  97                 test $code -ne 0 || code=1
  98         }
  99         rm -f $T_RM_LIST
 100         test $code -eq 0 && rm -f $T_OK_RM_LIST
 101         set +x
 102         exec >&3 2>&4
 103         t_close_fds
 104         exit $code
 105 }
 106
 107 # close test-specific extra file descriptors
 108 t_close_fds () {
 109         exec 3>&- 4>&-
 110 }
 111
 112 # call this at the start of your test to specify the number of tests
 113 # you plan to run
 114 t_plan () {
 115         test "$1" -ge 1 || die "must plan at least one test"
 116         test $t_expect_nr -eq -1 || die "tried to plan twice in one test"
 117         t_expect_nr=$1
 118         shift
 119         t_echo 1..$t_expect_nr "#" "$@"
 120         trap t_at_exit EXIT
 121 }
 122
 123 _t_checkup () {
 124         test $t_expect_nr -le 0 && die "no tests planned"
 125         test -n "$t_current" && t_echo "ok $t_nr - $t_current"
 126         true
 127 }
 128
 129 # finalizes any previously test and starts a new one
 130 t_begin () {
 131         _t_checkup
 132         t_nr=$(( $t_nr + 1 ))
 133         t_current="$1"
 134
 135         # just in case somebody wanted to cheat us:
 136         set -e
 137 }
 138
 139 # finalizes the current test without starting a new one
 140 t_end () {
 141         _t_checkup
 142         t_current=
 143 }
 144
 145 # run this to signify the end of your test
 146 t_done () {
 147         _t_checkup
 148         t_current=
 149         t_complete=true
 150         test $t_expect_nr -eq $t_nr || exit 1
 151         exit 0
 152 }
 153
 154 # create and assign named-pipes to variable _names_ passed to this function
 155 t_fifos () {
 156         for _id in "$@"
 157         do
 158                 _name=$_id
 159                 _tmp=$(mktemp -t $T.$$.$_id.XXXXXXXX)
 160                 eval "$_id=$_tmp"
 161                 rm -f $_tmp
 162                 mkfifo $_tmp
 163                 T_RM_LIST="$T_RM_LIST $_tmp"
 164         done
 165 }
 166
 167 t_verbose=false t_trace=false
 168
 169 while test "$#" -ne 0
 170 do
 171         arg="$1"
 172         shift
 173         case $arg in
 174         -v|--verbose) t_verbose=true ;;
 175         --trace) t_trace=true t_verbose=true ;;
 176         *) die "Unknown option: $arg" ;;
 177         esac
 178 done
 179
 180 # we always only setup stdout, nothing should end up in the "real" stderr
 181 if $t_verbose
 182 then
 183         if test x"$(which mktemp 2>/dev/null)" = x
 184         then
 185                 die "mktemp(1) not available for --verbose"
 186         fi
 187         t_fifos t_stdout t_stderr
 188
 189         (
 190                 # use a subshell so seds are not waitable
 191                 $SED -e 's/^/#: /' < $t_stdout &
 192                 $SED -e 's/^/#! /' < $t_stderr &
 193         ) &
 194         wait
 195         exec > $t_stdout 2> $t_stderr
 196 else
 197         exec > /dev/null 2> /dev/null
 198 fi
 199
 200 $t_trace && set -x
 201 true