#!/bin/sh

# Run binary tests.

# Usage: test/run [<test.rad|test.ras>...]

#

# If no arguments are provided, runs all `.rad` and `.ras` tests in

# `test/tests/`.

#

# For each test:

#   - If a `.ril` file exists alongside it, the IL output is checked

#     against it via the runner binary.

#   - If `//! session-support` appears, the IL check uses the fixture allocator

#     package required by session allocation.

#   - If `//! returns: N` appears in the file, the test is compiled to

#     a binary and executed; the exit code must match N.

#   - If `//! rw-data-size: N` appears in the file, the emitted image header's

#     rwdata size must match N bytes.

#   - If `//! rejects: TEXT` appears, the compiler must reject the source with

#     that diagnostic. Crashes and timeouts are failures.

RUNNER="test/runner.rv64"

TEST_DIR="test/tests"

EMU="${RAD_EMULATOR:-emulator} -stack-size=1024 -run"

EMU_RUN="${RAD_EMULATOR:-emulator} -no-jit -run"

if [ ! -f "$RUNNER" ]; then

  echo "error: runner binary not found: $RUNNER" >&2

  echo "hint: run 'make test' first" >&2

  exit 1

fi

# Disable core dumps for tests.

ulimit -c 0

# Collect tests.

if [ $# -eq 0 ]; then

  tests=$(find "$TEST_DIR" \( -name '*.rad' -o -name '*.ras' \) | sort)

else

  tests="$*"

fi

if [ -z "$tests" ]; then

  echo "error: no tests found" >&2

  exit 1

fi

passed=0

failed=0

for test in $tests; do

  case "$test" in

    *.rad) base="${test%.rad}" ;;

    *.ras) base="${test%.ras}" ;;

    *) base="$test" ;;

  esac

  ril="${base}.ril"

  bin="${base}.rv64"

  if grep -q '^//! rejects: ' "$test"; then

    echo -n "test $test rejection ... "

    if sh test/reject "$test"; then

      echo "ok"

      passed=$((passed + 1))

    else

      echo "FAILED"

      failed=$((failed + 1))

    fi

    continue

  fi

  # IL check: run the runner if a .ril file exists.

  if [ -f "$ril" ]; then

    if grep -q '^//! session-support$' "$test"; then

      echo -n "test $test ... "

      if test/package-golden "$test"; then

        echo "ok"

        passed=$((passed + 1))

      else

        echo "FAILED"

        failed=$((failed + 1))

      fi

    elif $EMU "$RUNNER" -- "$test"; then

      passed=$((passed + 1))

    else

      failed=$((failed + 1))

    fi

  fi

  # Execution check: run the binary if //! returns: is present.

  returns=$(grep -m1 '^//! returns:' "$test" | sed 's/^\/\/! returns: *//')

  if [ -n "$returns" ]; then

    echo -n "test $test ... "

    if [ ! -f "$bin" ]; then

      echo "FAILED (binary not found: $bin)"

      failed=$((failed + 1))

      continue

    fi

    $EMU_RUN "$bin" >/dev/null 2>&1

    ret=$?

    if [ "$ret" -eq "$returns" ]; then

      echo "ok"

      passed=$((passed + 1))

    else

      echo "FAILED (expected: $returns, got: $ret)"

      failed=$((failed + 1))

    fi

  fi

  rw_data_size=$(grep -m1 '^//! rw-data-size:' "$test" | sed 's/^\/\/! rw-data-size: *//')

  if [ -n "$rw_data_size" ]; then

    echo -n "test $test rw.data ... "

    if [ ! -f "$bin" ]; then

      echo "FAILED (binary not found: $bin)"

      failed=$((failed + 1))

      continue

    fi

    actual_rw_data_size=$(od -An -tu4 -j16 -N4 "$bin" | tr -d ' ')

    if [ "$actual_rw_data_size" -eq "$rw_data_size" ]; then

      echo "ok"

      passed=$((passed + 1))

    else

      echo "FAILED (expected: $rw_data_size, got: $actual_rw_data_size)"

      failed=$((failed + 1))

    fi

  fi

done

echo

if [ "$failed" -eq 0 ]; then

  echo "test result: ok. $passed passed; $failed failed"

  exit 0

else

  echo "test result: FAILED. $passed passed; $failed failed"

  exit 1

fi