#!/usr/bin/env perl

BEGIN {
   die "The PERCONA_TOOLKIT_BRANCH environment variable is not set.\n"
      unless $ENV{PERCONA_TOOLKIT_BRANCH} && -d $ENV{PERCONA_TOOLKIT_BRANCH};
   unshift @INC, "$ENV{PERCONA_TOOLKIT_BRANCH}/lib";
};

use strict;
use warnings FATAL => 'all';
use English qw(-no_match_vars);
use Test::More tests => 15;

use PerconaTest;
require "$trunk/bin/pt-diskstats";

# pt-diskstats is an interactive-only tool.  It waits for user input
# (i.e. menu commands) via STDIN.  So we cannot just run it with input,
# get ouput, then test that output.  We have to tie STDIN to these subs
# so that we can fake sending pt-diskstats menu commands via STDIN.
# All we do is send 'q', the command to quit.  See the note in the bottom
# of this file about *DATA. Please don't close it.
{
sub Test::TIEHANDLE {
   return bless {}, "Test";
}

sub Test::FILENO {
   return fileno(*DATA);
}

sub Test::READLINE {
   return "q";
}
}

sub test_diskstats_file {
   my (%args) = @_;
   my $file = "$trunk/t/pt-diskstats/samples/$args{file}";
   die "$file does not exist" unless -f $file;
   foreach my $groupby ( qw(all disk sample) ) {
      ok(
         no_diff(
            sub {
               tie local *STDIN, "Test";
               pt_diskstats::main(
                  qw(--zero-rows --group-by), $groupby,
                  '--columns','cnc|rt|mb|busy|prg',
                  $file);
            },
            "t/pt-diskstats/expected/${groupby}_int_$args{file}",
            keep_output=>1,
         ),
         "$args{file} --group-by $groupby"
      );
   }
}

foreach my $file ( map "diskstats-00$_.txt", 1..5 ) {
   test_diskstats_file(file => $file);
}

# ###########################################################################
# --save-samples and --iterations
# ###########################################################################

# TODO: fix this

#my $iterations = 2;
#my $out = output( sub {
#   pt_diskstats::main(
#      "--group-by"     => "all",
#      "--columns"      => "cnc|rt|mb|busy|prg",
#      "--save-samples" => $tempfile,
#      "--iterations"   => $iterations,
#      "--zero-rows",
#   );
#});
#
#is(
#   $count,
#   $iterations,
#   "--save-samples and --iterations work"
#);


# ###########################################################################
# Done.
# ###########################################################################
exit;

__DATA__
Leave this here!
We tie STDIN during the tests, and fake the fileno by giving it *DATA's result;
These lines here make Perl leave *DATA open.