#!/bin/sh

usage() {
   echo "Usage: $0 [path/to/test/output/dir] [OPTIONAL: --exclusive] [OPTIONAL: path/to/tests/to/skip-or-run.txt] [OPTIONAL: path/to/existing/test/logdir/]"
   echo "if you pass --exclusive, ONLY run tests in skip-or-run.txt, otherwise, run all tests EXCEPT those in skip-or-run.txt"
   exit 1
}

if [ $# -lt 1 ]; then
   usage $0
fi

RUN_SCENARIO="blockstack-test-scenario"
CHECK_SERIALIZATION="blockstack-test-check-serialization"
SCENARIOS_PYTHON="blockstack_integration_tests.scenarios"
SCENARIOS="$(BLOCKSTACK_TESTNET=1 python -c "import blockstack_integration_tests; import blockstack_integration_tests.scenarios; print blockstack_integration_tests.scenarios.__path__[0]")"
if [ $? -ne 0 ]; then
   echo >&2 "Failed to load blockstack integration test scenarios"
   exit 1
fi

OUTPUTS="$1"

if [ "$2" = "--exclusive" ]; then
    EXCLUSIVE=1
    TESTS_SKIP="$3"
    EXISTING_LOGS="$4"
else
    EXCLUSIVE=0
    TESTS_SKIP="$2"
    EXISTING_LOGS="$3"
fi

test -d "$OUTPUTS" || mkdir -p "$OUTPUTS"

while IFS= read SCENARIO_FILE; do

   if ! [ "$(echo "$SCENARIO_FILE" | egrep '.py$')" ]; then
      continue
   fi

   if [ "$SCENARIO_FILE" = "__init__.py" ] || [ "$SCENARIO_FILE" = "testlib.py" ]; then
      continue
   fi

   SCENARIO_MODULE_BASE="$(echo "$SCENARIO_FILE" | sed 's/\.py//g')"
   SCENARIO_MODULE="$SCENARIOS_PYTHON.$SCENARIO_MODULE_BASE"

   if [ "0" -eq "$EXCLUSIVE" ] && [ -n "$TESTS_SKIP" ] && [ -n "$(grep "^$SCENARIO_MODULE_BASE\$" "$TESTS_SKIP")" ]; then
      continue
   fi

   if [ "1" -eq "$EXCLUSIVE" ] && [ -n "$TESTS_SKIP" ]; then
       if ! [ -n "$(grep "^$SCENARIO_MODULE_BASE\$" "$TESTS_SKIP")" ]; then
           continue
       fi
   fi

   TESTDIR="/tmp/blockstack-test"

   if ! [ -f "$OUTPUTS/$SCENARIO_MODULE_BASE.log" ]; then

      echo "$SCENARIO_MODULE ... "

      mkdir -p "$TESTDIR"
      "$RUN_SCENARIO" "$SCENARIO_MODULE" "$TESTDIR" > "$OUTPUTS/$SCENARIO_MODULE_BASE.log" 2>&1 &
      TEST_PID=$!
      while ps -p $TEST_PID > /dev/null; do
          etime=$(ps -p $TEST_PID -o etimes | tail -n 1 | awk '{ print $1 }')
          if [ $etime -ge 1200 ]; then
              echo -n " KILL for time "
              kill -s INT $TEST_PID
          fi
          sleep 10
      done

      grep -q "SUCCESS" "$OUTPUTS/$SCENARIO_MODULE_BASE.log"
      RC=$?

      # make the xunit file
      OUTPUT_FILE="$OUTPUTS/$SCENARIO_MODULE_BASE.xml"

      echo '<?xml version="1.0" encoding="UTF-8"?><testsuite>' > "$OUTPUT_FILE"
      echo "<testcase name=\"$SCENARIO_MODULE_BASE\" classname=\"test\">" >> "$OUTPUT_FILE"
      if [ $RC -ne 0 ]; then
          echo '<failure message="Integration test failed."></failure>' >> "$OUTPUT_FILE"
      fi
      echo "</testcase></testsuite>" >> "$OUTPUT_FILE"

      if [ $RC -ne 0 ]; then
         # failed
         echo " FAILURE"
         mv "$TESTDIR" "$OUTPUTS/$SCENARIO_MODULE_BASE.d"

         continue
      fi

      # compare serialized consensus fields with older release
      EXISTING_LOG="$EXISTING_LOGS/$SCENARIO_MODULE_BASE.log"
      if [ -f "$EXISTING_LOG" ]; then

         "$CHECK_SERIALIZATION" "$EXISTING_LOGS/$SCENARIO_MODULE_BASE.log" "$OUTPUTS/$SCENARIO_MODULE_BASE.log"
         RC=$?

         if [ $RC -ne 0 ]; then

            if [ $RC -eq 1 ]; then
                # generated incorrect serialization output
                echo " (ERROR: mismatched serialization) FAILURE"
                mv "$TESTDIR" "$OUTPUTS/$SCENARIO_MODULE_BASE.d"

                # TODO: only exit with option
                exit 1
            else
               # exit 2 means no serialization check happened
               echo " (WARN: SKIPPED SERIALIZATION CHECK) SUCCESS"
            fi
         fi
      else
         echo -n " (WARN: no existing log) "
      fi

      bitcoin_pid=$(pidof bitcoind)
      if [ "$bitcoin_pid" != "" ]; then
          echo " killing still-living bitcoind "
          kill -9 $bitcoin_pid
      fi

      echo " SUCCESS"
      rm -rf "$TESTDIR"
   fi

done <<EOF
$(ls "$SCENARIOS")
EOF

exit 0