gen_tests.rb

#!/usr/bin/env ruby

# Purpose: Generate test scripts
# Author : Anh K. Huynh
# License: MIT
# Date   : 2016 July 18th
# Example:
#
#   $ ruby -n ./bin/gen_tests.rb < lib/dpkg.txt > ./test.sh
#   $ docker run --rm \
#       -v $PWD/test.sh:/tmp/test.sh \
#       -v $PWD/pacapt.dev:/usr/bin/pacman \
#       ubuntu:14.04 \
#       /bin/sh /tmp/test.sh 2>test.log
#

BEGIN {
  new_test = true
  puts "#!/bin/bash"
  puts ""
  puts "export PATH=/usr/bin:$PATH"
  puts ""
  puts "N_TEST=0  # Total number of tests"
  puts "N_FAIL=0  # Number of failed tests"
  puts "F_TMP=    # Temporary file"
  puts "T_FAIL=0  # Number of failed tests in current block"
  puts ""
  puts "set -u"
  puts ""
  puts "_log()  { echo \":: $*\" 1>&2 ; }"
  puts "_fail() { _log \"Fail $*\"; echo \"Fail: $*\"; }" # red
  puts "_erro() { _log \"Erro $*\"; echo \"Erro: $*\"; }" # red
  puts "_info() { _log \"Info $*\"; echo \"Info: $*\"; }" # cyan
  puts "_pass() { _log \"Pass $*\"; echo \"Pass: $*\"; }" # cyan
  puts "_exec() { _log \"Exec $*\"; echo \"Exec: $*\"; }" # yellow
  puts "_warn() { _log \"Warn $*\"; echo \"Warn: $*\"; }" # yellow
  puts "_slog() {"
  puts "  if [ -n \"${F_TMP:-}\" ]; then"
  puts "    echo 1>&2 ':: Exec output'"
  puts "    cat 1>&2 $F_TMP"
  puts "    if [ $T_FAIL -ge 1 ]; then"
  puts "      cat $F_TMP \\"
  puts "      | awk '{ if (NR <= 100) { printf(\" > %s\\n\", $0); }}"
  puts "         END { if (NR > 100) { printf(\" > ...\\n\");}}'"
  puts "    fi"
  puts "    rm -f \"${F_TMP}\""
  puts "    echo 1>&2"
  puts "  fi"
  puts "  export T_FAIL=0"
  puts "}"
}

if gs = $_.match(/^in(.*)/)
  if new_test
    puts ""
    outputs = []
    new_test = false

    puts "_slog"
    puts "export F_TMP=\"$(mktemp)\""
    puts "if [ -z ${F_TMP:-} ]; then"
    puts "  _fail 'Unable to create temporary file.'"
    puts "fi"
  end

  cmd = gs[1].strip.gsub("\$LOG", "$F_TMP")

  cmd = \
  case cmd
  when "clear" then
    "echo > $F_TMP"
  when /\A! (.+)\z/ then
    "#{$1.strip}"
  else
    "pacman #{cmd}"
  end

  # FIXME: We wanted to use `tee` here, but that will create new pipes,
  # FIXME: and hence some feature didn't work (e.g., export FOO=).
  puts "if [ -n \"${F_TMP:-}\" ]; then"
  puts "  _exec \"#{cmd}\""
  puts "  { #{cmd} ; } 1>>$F_TMP 2>&1"
  puts "fi"

elsif gs = $_.match(/^ou(.*)/)
  new_test = true
  expected = gs[1].strip

  puts "N_TEST=$(( N_TEST + 1 ))"
  puts "if [ -n \"${F_TMP:-}\" ]; then"
  if expected.empty? or expected == "empty"
    puts "  ret=\"$(grep -Ec '.+' $F_TMP)\""
    puts "  if [ $ret -ge 1 ]; then"
  else
    puts "  ret=\"$(grep -Ec \"#{expected}\" $F_TMP)\""
    puts "  if [ $ret -eq 0 ]; then"
  end
  puts "    _fail Expected \"#{expected}\""
  puts "    N_FAIL=$(( N_FAIL + 1 ))"
  puts "    T_FAIL=$(( T_FAIL + 1 ))"
  puts "  else"
  puts "    _pass Matched \"#{expected}\""
  puts "  fi"
  puts "else"
  puts "  N_FAIL=$(( N_FAIL + 1 ))"
  puts "fi"
end

END {
  puts "_slog"
  puts "if [ $N_FAIL -ge 1 ]; then"
  puts "  _fail \"$N_FAIL/$N_TEST test(s) failed.\""
  puts "  exit 1"
  puts "else"
  puts "  _pass \"All $N_TEST tests(s) passed.\""
  puts "fi"
}