#!/usr/bin/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 => 7;

use ReplicaLagWaiter;
use OptionParser;
use DSNParser;
use Cxn;
use PerconaTest;

my $oktorun = 1;
my @waited  = ();
my @lag     = ();
my @sleep   = ();
my @worst   = ();

sub oktorun {
   return $oktorun;
}

sub get_lag {
   my ($cxn) = @_;
   push @waited, $cxn->dbh();
   my $lag = shift @lag || 0;
   return $lag;
}

sub sleep {
   push @worst, [@_];
   my $t = shift @sleep || 0;
   sleep $t;
}

my $dp = new DSNParser(opts=>$dsn_opts);
my $o  = new OptionParser(description => 'Cxn');
$o->get_specs("$trunk/bin/pt-table-checksum");
$o->get_opts();
$dp->prop('set-vars', $o->get('set-vars'));

my $r1 = new Cxn(dsn=>{n=>'slave1'}, dbh=>1, DSNParser=>$dp, OptionParser=>$o);
my $r2 = new Cxn(dsn=>{n=>'slave2'}, dbh=>2, DSNParser=>$dp, OptionParser=>$o);

my $rll = new ReplicaLagWaiter(
   oktorun => \&oktorun,
   get_lag => \&get_lag,
   sleep   => \&sleep,
   max_lag => 1,
   slaves  => [$r1, $r2],
);

@lag = (0, 0);
my $t = time;
$rll->wait();
ok(
   time - $t < 0.5,
   "wait() returns immediately if all slaves are ready"
);

is_deeply(
   \@waited,
   [1,2],
   "Waited for all slaves"
);

is_deeply(
   \@worst,
   [],
   "Did not call sleep callback"
);

@waited = ();
@lag    = (3, 5, 0, 0);
@sleep  = (1, 1, 1, 1);
@worst  = ();
$t   = time;
$rll->wait(),
ok(
   time - $t >= 0.9,
   "wait() waited a second"
);

is_deeply(
   \@waited,
   [1, 2, 2, 1],
   "wait() waited for first slave"
);

# This tests that the code sorts the lag correctly.  r2 is lagging
# the worst (5 > 3) so it should be passed to the sleep callback.
is_deeply(
   \@worst,
   [
      [ $r2, 5 ],
   ],
   "Called sleep callback with worst lagger"
);

# #############################################################################
# Done.
# #############################################################################
my $output = '';
{
   local *STDERR;
   open STDERR, '>', \$output;
   $rll->_d('Complete test coverage');
}
like(
   $output,
   qr/Complete test coverage/,
   '_d() works'
);
exit;