#!@PERL@

#

# Run unit tests.

#

# Syntax:

#          All: run-tests.pl

#     All in file: run-tests.pl file

#     Nth in file: run-tests.pl file N

#

use strict;

use POSIX qw(WIFEXITED WEXITSTATUS WIFSIGNALED WTERMSIG);

use File::Basename qw(basename dirname);

use FileHandle;

use IPC::Open2 qw(open2);

my @TYPES = qw(tfn op action);

my $TEST = "./msc_test";

my $SCRIPT = basename($0);

my $SCRIPTDIR = dirname($0);

my $PASSED = 0;

my $TOTAL = 0;

my $DEBUG = $ENV{MSC_TEST_DEBUG} || 0;

if (defined $ARGV[0]) {

    runfile(dirname($ARGV[0]), basename($ARGV[0]), $ARGV[1]);

    done();

}

for my $type (sort @TYPES) {

    my $dir = "$SCRIPTDIR/$type";

    my @cfg = ();

    # Get test names

    opendir(DIR, "$dir") or quit(1, "Failed to open \"$dir\": $!");

    @cfg = grep { /\.t$/ && -f "$dir/$_" } readdir(DIR);

    closedir(DIR);

    for my $cfg (sort @cfg) {

        runfile($dir, $cfg);

    }

}

done();

sub runfile {

    my($dir, $cfg, $testnum) = @_;

    my $fn = "$dir/$cfg";

    my @data = ();

    my $edata;

    my @C = ();

    my @test = ();

    my $teststr;

    my $n = 0;

    my $pass = 0;

    open(CFG, "<$fn") or quit(1, "Failed to open \"$fn\": $!");

    @data = <CFG>;

    $edata = q/@C = (/ . join("", @data) . q/)/;

    eval $edata;

    quit(1, "Failed to read test data \"$cfg\": $@") if ($@);

    unless (@C) {

        msg("\nNo tests defined for $fn");

        return;

    }

    msg("\nLoaded ".@C." tests from $fn");

    for my $t (@C) {

        $n++;

        next if (defined $testnum and $n != $testnum);

        my %t = %{$t || {}};

        my $id = sprintf("%6d", $n);

        my $in = (exists($t{input}) and defined($t{input})) ? $t{input} : "";

        my $out;

        my $test_in = new FileHandle();

        my $test_out = new FileHandle();

        my $test_pid;

        my $rc = 0;

        my $param;

        if ($t{type} eq "tfn") {

            $param = escape($t{output});

        }

        elsif ($t{type} eq "op") {

            $param = escape($t{param});

        }

        elsif ($t{type} eq "action") {

            $param = escape($t{param});

        }

        else {

            quit(1, "Unknown type \"$t{type}\" - should be one of: " . join(",",@TYPES));

        }

        @test = ("-t", $t{type}, "-n", $t{name}, "-p", $param, "-D", "$DEBUG", (exists($t{ret}) ? ("-r", $t{ret}) : ()), (exists($t{iterations}) ? ("-I", $t{iterations}) : ()), (exists($t{prerun}) ? ("-P", $t{prerun}) : ()));

        $teststr = "$TEST " . join(" ", map { "\"$_\"" } @test);

        $test_pid = open2($test_out, $test_in, $TEST, @test) or quit(1, "Failed to execute test: $teststr\": $!");

        print $test_in "$in";

        close $test_in;

        $out = join("\\n", split(/\n/, <$test_out>));

        close $test_out;

        waitpid($test_pid, 0);

        $rc = $?;

        if ( WIFEXITED($rc) ) {

            $rc = WEXITSTATUS($rc);

        }

        elsif( WIFSIGNALED($rc) ) {

            msg("Test exited with signal " . WTERMSIG($rc) . ".");

            msg("Executed: $teststr");

            $rc = -1;

        }

        else {

            msg("Test exited with unknown error.");

            $rc = -1;

        }

        if ($rc == 0) {

            $pass++;

        }

        msg(sprintf("%s) %s \"%s\"%s: %s%s", $id, $t{type}, $t{name}, (exists($t{comment}) ? " $t{comment}" : ""), ($rc ? "failed" : "passed"), ((defined($out) && $out ne "")? " ($out)" : "")));

    }

    $TOTAL += $testnum ? 1 : $n;

    $PASSED += $pass;

    msg(sprintf("Passed: %2d; Failed: %2d", $pass, $testnum ? (1 - $pass) : ($n - $pass)));

}

sub escape {

    my @new = ();

    for my $c (split(//, $_[0])) {

        push @new, ((ord($c) >= 0x20 and ord($c) <= 0x7e) ? $c : sprintf("\\x%02x", ord($c)));

    }

    join('', @new);

}

sub msg {

    print STDOUT "@_\n" if (@_);

}

sub quit {

    my($ec,$msg) = @_;

    $ec = 0 unless (defined $_[0]);

    msg("$msg") if (defined $msg);

    exit $ec;

}

sub done {

    if ($PASSED != $TOTAL) {

        quit(1, "\n$PASSED/$TOTAL tests passed.");

    }

    quit(0, "\nAll tests passed ($TOTAL).");

}