urchin/urchin

#!/bin/bash

stdout_file=/tmp/urchin_stdout

indent() {
  level="$1"
  printf "%$((2 * ${level}))s"
}

recurse() {
  potential_test="$1"
  indent_level="$2"

  [ "$potential_test" = 'setup_dir' ] && return
  [ "$potential_test" = 'teardown_dir' ] && return
  [ "$potential_test" = 'setup' ] && return
  [ "$potential_test" = 'teardown' ] && return

  echo > "$stdout_file"

  if [ -d "$potential_test" ]
    then
    (
      indent $indent_level
      echo "  ${potential_test}"
      cd "$potential_test"
      [ -f setup_dir ] && [ -x setup_dir ] && ./setup_dir &>> $stdout_file
      for test in *
        do
        [ -f setup ] && [ -x setup ] && ./setup &>> $stdout_file

        # $2 instead of $indent_level so it doesn't clash
        recurse "${test}" $(( $2 + 1 ))

        [ -f teardown ] && [ -x teardown ] && ./teardown &>> $stdout_file
      done
      [ -f teardown_dir ] && [ -x teardown_dir ] && ./teardown_dir &>> $stdout_file
      echo
    )
  elif [ -x "$potential_test" ]
    then

    [ -f setup ] && [ -x setup ] && ./setup &>> $stdout_file

    # Run the test
    ./"$potential_test" &>> $stdout_file
    exit_code="$?"

    [ -f teardown ] && [ -x teardown ] && ./teardown &>> $stdout_file

    indent $indent_level
    if [ "$exit_code" = '0' ]
      then
      # On success, print a '✓'
      echo -ne '\033[32m✓ \033[0m'
      echo "${potential_test}"
      echo "${potential_test} passed" >> "$logfile"
    else
      # On fail, print a red '✗'
      echo -ne '\033[31m✗ \033[0m'
      echo "${potential_test}"
      echo "${potential_test} failed" >> "$logfile"
      cat $stdout_file
    fi
    rm $stdout_file
  fi
}

if [ "$#" = '1' ] && [ -d "$1" ]
  then
  echo Running tests

  logfile=$(readlink -f "$1/.urchin.log")
  echo > "$logfile"
  recurse "$1" 0

  echo
  echo Done
  echo $(grep -e 'passed$' "$logfile"|wc -l) tests passed.
  echo $(grep -e 'failed$' "$logfile"|wc -l) tests failed.
else
  echo "usage: $0 <test directory>"
  echo 'Go to http://www.urchin.sh for documentation on writing tests.'
fi