#!/bin/sh

# Verify test discovery, package selection, and descriptor capacity.

set -eu

ulimit -c 0

work=$(mktemp -d)

trap 'rm -rf "$work"' EXIT HUP INT TERM

mkdir "$work/selected"

# The dependency test must not enter the selected package's runner.

printf '%s\n' '/// Test outside the entry package.' '@test fn excluded() { assert false; }' > "$work/dependency.rad"

printf '%s\n' 'use std::testing;' '/// Test in a child module.' \

    '@test fn childTest() throws (testing::TestError) { assert true; }' > "$work/selected/child.rad"

# Compile through Make so the driver uses the standard memory settings.

compile() {

    timeout 300 "${MAKE:-make}" --no-print-directory driver-test-image \

        RAD_BIN="${RAD_BIN:-bin/radiance.rv64.dev}" \

        RAD_EMULATOR="${RAD_EMULATOR:-emulator}" \

        DRIVER_TEST_INPUTS="-pkg dependency -mod $work/dependency.rad -pkg selected -mod $work/selected.rad $1" \

        DRIVER_TEST_FLAGS="${2:-}" \

        DRIVER_TEST_OUTPUT="$work/result.rv64" > "$work/compile.log" 2>&1

}

# Check one successful runner and its exact number of executed tests.

run() {

    if ! compile "$1"; then

        cat "$work/compile.log"

        exit 1

    fi

    if ! timeout 10 "${RAD_EMULATOR:-emulator}" -run "$work/result.rv64" > "$work/run.log" 2>&1; then

        cat "$work/run.log"

        exit 1

    fi

    grep -Fq "test result: ok. $2 passed; 0 failed" "$work/run.log"

    test "$(grep -c '^test selected::' "$work/run.log")" -eq "$2"

}

printf '%s\n' 'use std::testing;' 'mod child;' \

    '/// Test in the package root.' '@test fn rootTest() throws (testing::TestError) { assert true; }' > "$work/selected.rad"

run "-mod $work/selected/child.rad" 2

grep -Fq 'test selected::rootTest ... ok' "$work/run.log"

grep -Fq 'test selected::child::childTest ... ok' "$work/run.log"

printf '%s\n' 'use std::testing;' > "$work/selected.rad"

if compile ''; then

    echo 'driver: empty test discovery unexpectedly succeeded'

    exit 1

fi

grep -Fq 'no test functions found' "$work/compile.log"

# MAX_TESTS is the compiler's supported discovery capacity.

for count in 1024 1025; do

    awk -v count="$count" -v directory="$work/selected" 'BEGIN {

        print "use std::testing;"

        for (i = 0; i < count; i++) {

            part = int(i / 256)

            path = directory "/part" part ".rad"

            if (i % 256 == 0) {

                printf "/// Capacity test group.\nmod part%d;\n", part

                print "use std::testing;" > path

            }

            print "/// Generated capacity test." > path

            printf "@test fn test%d() throws (testing::TestError) { assert true; }\n", i > path

        }

    }' > "$work/selected.rad"

    inputs=''

    for path in "$work/selected"/part*.rad; do

        inputs="$inputs -mod $path"

    done

    if [ "$count" -eq 1024 ]; then

        if ! compile "$inputs" '-dump il'; then

            cat "$work/compile.log"

            exit 1

        fi

        grep -Fq 'selected: found 1024 test(s)' "$work/compile.log"

        test "$(grep -c 'call.*std::testing::test' "$work/compile.log")" -eq "$count"

        run "$inputs" "$count"

    else

        if compile "$inputs"; then

            echo 'driver: test discovery overflow unexpectedly succeeded'

            exit 1

        fi

        grep -Fq 'Runtime error (EBREAK)' "$work/compile.log"

    fi

done

echo 'driver tests: 4 passed'