plum

#!/usr/bin/env bash

#

# Integration test: compiles and runs every examples/*.plum file through the

# REAL toolchain — the `plum` CLI binary (which resolves `import`s against

# --lib-path, unlike plum-checker/plum-wasm-codegen's unit tests, which parse

# and compile a single file directly) and, if available, an external

# `wasmtime` runtime.

#

# This exists because the unit tests alone missed a real bug: examples/basics.plum

# had an `import std/io` where libs/std/io.plum doesn't exist. The unit tests

# never noticed since they don't touch plum-core's loader/import resolution at

# all — only compiling and running through the actual CLI catches that class

# of problem.

#

# Usage: scripts/test-examples.sh

# Exit code is 0 if every example compiled (and, where checked, ran) correctly.

set -uo pipefail

cd "$(dirname "${BASH_SOURCE[0]}")/.."

WORKDIR=$(mktemp -d)

trap 'rm -rf "$WORKDIR"' EXIT

echo "Building plum-cli..."

if ! cargo build -p plum-cli --quiet 2>"$WORKDIR/build.log"; then

  echo "FAIL: plum-cli failed to build" >&2

  cat "$WORKDIR/build.log" >&2

  exit 1

fi

PLUM_BIN="target/debug/plum"

HAVE_WASMTIME=0

if command -v wasmtime >/dev/null 2>&1; then

  HAVE_WASMTIME=1

else

  echo "note: wasmtime CLI not found on PATH — will compile every example but skip running them" >&2

fi

# Examples that export `main` and are expected to run without trapping. Kept

# in sync with plum-wasm-codegen/tests/examples_test.rs.

has_main() {

  case "$1" in

    basics|closures|match|methods) return 0 ;;

    *) return 1 ;;

  esac

}

# Expected `main` return value for examples whose result is asserted in

# plum-wasm-codegen/tests/examples_test.rs. Empty means "just don't trap"

# (e.g. basics.plum's main returns Unit, not a value to compare).

expected_value() {

  case "$1" in

    closures) echo 120 ;;

    match) echo 5 ;;

    methods) echo 10 ;;

    *) echo "" ;;

  esac

}

failed=0

passed=0

for src in examples/*.plum; do

  name=$(basename "$src" .plum)

  wasm="$WORKDIR/$name.wasm"

  printf '%-14s compile ... ' "$name"

  if ! "$PLUM_BIN" compile "$src" -o "$wasm" >"$WORKDIR/$name.compile.log" 2>&1; then

    echo "FAIL"

    sed 's/^/    /' "$WORKDIR/$name.compile.log"

    failed=$((failed + 1))

    continue

  fi

  echo "ok"

  if [ "$HAVE_WASMTIME" -eq 1 ] && has_main "$name"; then

    printf '%-14s run     ... ' "$name"

    if output=$(wasmtime run --invoke main "$wasm" 2>"$WORKDIR/$name.run.log"); then

      expected=$(expected_value "$name")

      if [ -n "$expected" ] && [ "$output" != "$expected" ]; then

        echo "FAIL (expected $expected, got '$output')"

        failed=$((failed + 1))

        continue

      fi

      if [ -n "$output" ]; then

        echo "ok (-> $output)"

      else

        echo "ok"

      fi

    else

      echo "FAIL (trapped)"

      sed 's/^/    /' "$WORKDIR/$name.run.log"

      failed=$((failed + 1))

      continue

    fi

  fi

  passed=$((passed + 1))

done

echo

echo "$passed passed, $failed failed"

[ "$failed" -eq 0 ]