#!/usr/bin/perl -w
# keine_test.pl - Don Yang (uguu.org)
#
# ./keine_test.pl <script>
#
# 09/02/07

use strict;
use constant TMP_PREFIX => "keine_test.tmp";


# Test report summary functions
sub TestStats($)
{
   my ($script) = @_;
   my $data = TMP_PREFIX . ".log";
   my $input = TMP_PREFIX . ".in";

   # Generate test data
   my $file;
   open $file, "> $data" or die $!;
   print $file <<'EOT';
2008-02-28 00:00:00 > BEGIN LOG
2008-02-28 00:01:00 > task 1
2008-02-28 00:02:00 > task 2
2008-03-01 00:02:00 > task 1
2008-03-01 00:02:00 # a=abbrev
2008-03-01 01:02:01 > END LOG

2008-04-01 00:00:00 > BEGIN LOG
2008-04-01 00:01:00 > task 3
2008-04-01 00:02:00 < task 4
2008-04-01 00:03:00 > END LOG
EOT
   close $file;

   # Generate test input
   open $file, "> $input" or die $!;
   print $file "?\n\n!\n";
   close $file;

   # Run command
   my $cmd = "perl $script $data < $input";
   my @output = `$cmd`;
   die "'$cmd' failed: $?\n" if $?;
   foreach my $line (@output)
   {
      if( $line =~ /(BEGIN LOG)/ || $line =~ /(END LOG)/ )
      {
         $line = "<$1>\n";
      }
   }

   my $stats = <<'EOT';
Total time = 176581 (2d 01:03:01)
   task 2 = 172800 (2d 00:00:00)
   task 1 = 3661 (0d 01:01:01)
   task 3 = 60 (0d 00:01:00)
   task 4 = 60 (0d 00:01:00)
Interrupts = 1 / 60 (0d 00:01:00)
   task 4 = 1 / 60 (0d 00:01:00)
Context switch = 4
EOT
   my $expected_output = <<"EOT";
Loaded 10 events from keine_test.tmp.log

$stats
<BEGIN LOG>
Commands:
   task        Record context switch to "task"
   > task      Record context switch to "task"
   < task      Record interrupt to "task"
   - task      Record interrupt to "task"

   . prefix    Record context switch to previous task matching "prefix"
   , prefix    Record interrupt to previous task matching "prefix"
   / substr    Record context switch to previous task containing "substr"
   \\ substr    Record interrupt to previous task containing "substr"

   key=task    Define abbreviation and record task
   # key=task  Define abbreviation without recording task

   ?                 Show commands and abbreviations
   !                 Reload events and abbreviations from file
   (empty string)    Show stats
   (EOF)             Exit
Abbreviations:
   a = abbrev
$stats
Loaded 11 events from keine_test.tmp.log

$stats
<END LOG>
Stopped logging to keine_test.tmp.log
EOT
   my $actual_output = join '', @output;
   if( $expected_output ne $actual_output )
   {
      die "Output mismatched";
   }

   # Cleanup
   unlink $data;
   unlink $input;
}

# Test generating summary from stdin
sub TestSummary($)
{
   my ($script) = @_;

   # Generate test data
   my $data = TMP_PREFIX . ".log";
   my $file;
   open $file, "> $data" or die $!;
   print $file <<"EOT";
2007-02-28 00:00:00 > BEGIN LOG
2007-02-28 00:01:00 > task 1
2007-03-01 00:02:00 > END LOG

2007-03-02 00:00:00 > BEGIN LOG
2007-03-02 00:01:00 > task 1
2007-03-02 00:02:00 > END LOG

EOT
   close $file;

   # Run command
   my $cmd = "perl $script < $data";
   my @output = `$cmd`;
   die "'$cmd' failed: $?\n" if $?;

   my $expected_output = <<'EOT';
Total time = 86520 (1d 00:02:00)
   task 1 = 86520 (1d 00:02:00)
Context switch = 2

EOT
   my $actual_output = join '', @output;
   if( $expected_output ne $actual_output )
   {
      die "Output mismatched";
   }

   # Cleanup
   unlink $data;
}

# Test input matching and abbreviation substitution functions
sub TestInput($)
{
   my ($script) = @_;
   my $data = TMP_PREFIX . ".log";
   my $input = TMP_PREFIX . ".in";

   # Generate test input
   my $file;
   open $file, "> $input" or die $!;
   print $file <<'EOT';
hoge task 1
piyo task 1
>hoge task 2
>piyo task 2
-hage task 3
<hogehoge task
,p
.h
/yo
\ge
x=hogepiyo
y=<piyopiyo
x
y
#z=hage
hoge
<z
>y
EOT
   close $file;

   # Run command
   unlink $data;
   my $cmd = "perl $script $data < $input";
   my @output = `$cmd`;
   die "'$cmd' failed: $?\n" if $?;

   # Check for expected output
   open $file, "< $data" or die $!;
   my $actual_output = "";
   while(my $line = <$file>)
   {
      $line =~ s/\d\d\d\d-\d\d-\d\d \d\d:\d\d:\d\d/*timestamp*/;
      $actual_output .= $line;
   }
   close $file;

   my $expected_output = <<'EOT';
*timestamp* > BEGIN LOG
*timestamp* > hoge task 1
*timestamp* > piyo task 1
*timestamp* > hoge task 2
*timestamp* > piyo task 2
*timestamp* < hage task 3
*timestamp* < hogehoge task
*timestamp* < piyo task 2
*timestamp* > hogehoge task
*timestamp* > piyo task 2
*timestamp* < hogehoge task
*timestamp* # x=hogepiyo
*timestamp* > hogepiyo
*timestamp* # y=<piyopiyo
*timestamp* < piyopiyo
*timestamp* > hogepiyo
*timestamp* < piyopiyo
*timestamp* # z=hage
*timestamp* > hoge
*timestamp* < hage
*timestamp* > piyopiyo
*timestamp* > END LOG

EOT

   if( $expected_output ne $actual_output )
   {
      die "Output mismatched";
   }

   # Cleanup
   unlink $data;
   unlink $input;
}


die "$0 <script to test>\n" if $#ARGV != 0;
TestStats($ARGV[0]);
TestSummary($ARGV[0]);
TestInput($ARGV[0]);
exit 0;