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

use PerconaTest;
use Sandbox;
require "$trunk/bin/pt-archiver";

my $dp   = new DSNParser(opts=>$dsn_opts);
my $sb   = new Sandbox(basedir => '/tmp', DSNParser => $dp);
my $dbh  = $sb->get_dbh_for('master');
my $dbh2 = $sb->get_dbh_for('slave1');

if ( !$dbh ) {
   plan skip_all => 'Cannot connect to sandbox master';
}
else {
   plan tests => 6;
}

my $output;
my $cnf = "/tmp/12345/my.sandbox.cnf";
my $cmd = "$trunk/bin/pt-archiver";

$sb->create_dbs($dbh, [qw(test)]);

SKIP: {
   skip 'Sandbox master does not have the sakila database', 1
      unless @{$dbh->selectcol_arrayref('SHOW DATABASES LIKE "sakila"')};

   $output = `$cmd --source F=$cnf,h=127.1,D=sakila,t=film --no-check-charset  --where "film_id < 100" --purge --dry-run --port 12345 | diff $trunk/t/pt-archiver/samples/issue-248.txt -`;
   is(
      $output,
      '',
      'DSNs inherit from standard connection options (issue 248)'
   );
};


# Test with a sentinel file
$sb->load_file('master', 't/pt-archiver/samples/table1.sql');
diag(`touch sentinel`);
$output = output(
   sub { pt_archiver::main(qw(--where 1=1 --why-quit --sentinel sentinel), "--source", "D=test,t=table_1,F=$cnf", qw(--purge)) },
);
like($output, qr/because sentinel/, 'Exits because of sentinel');
$output = `/tmp/12345/use -N -e "select count(*) from test.table_1"`;
is($output + 0, 4, 'No rows were deleted');
`rm sentinel`;

# Test --stop, which sets the sentinel
$output = output(
   sub { pt_archiver::main(qw(--sentinel sentinel --stop)) },
);
like($output, qr/Successfully created file sentinel/, 'Created the sentinel OK');
diag(`rm -f sentinel >/dev/null`);

# #############################################################################
# Issue 391: Add --pid option to mk-table-sync
# #############################################################################
`touch /tmp/mk-archiver.pid`;
$output = `$cmd --where 1=1 --source F=$cnf,D=test,t=issue_131_src --statistics --dest t=issue_131_dst --pid /tmp/mk-archiver.pid 2>&1`;
like(
   $output,
   qr{PID file /tmp/mk-archiver.pid already exists},
   'Dies if PID file already exists (issue 391)'
);

`rm -rf /tmp/mk-archiver.pid`;

# #############################################################################
# Issue 460: mk-archiver does not inherit DSN as documented 
# #############################################################################
SKIP: {
   skip 'Cannot connect to sandbox slave1', 1 unless $dbh2;

   # This test will achive rows from dbh:test.table_1 to dbh2:test.table_2.
   $sb->load_file('master', 't/pt-archiver/samples/tables1-4.sql');

   # Change passwords so defaults files won't work.
   $dbh->do('SET PASSWORD FOR msandbox = PASSWORD("foo")');
   $dbh2->do('SET PASSWORD FOR msandbox = PASSWORD("foo")');

   $dbh2->do('TRUNCATE TABLE test.table_2');

   $output = `MKDEBUG=1 $trunk/bin/pt-archiver --where 1=1 --source h=127.1,P=12345,D=test,t=table_1,u=msandbox,p=foo --dest P=12346,t=table_2 --statistics 2>&1`;
   my $r = $dbh2->selectall_arrayref('SELECT * FROM test.table_2');
   is(
      scalar @$r,
      4,
      '--dest inherited from --source'
   );

   # Set the passwords back.  If this fails we should bail out because
   # nothing else is going to work.
   eval {
      $dbh->do("SET PASSWORD FOR msandbox = PASSWORD('msandbox')");
      $dbh2->do("SET PASSWORD FOR msandbox = PASSWORD('msandbox')");
   };
   if ( $EVAL_ERROR ) {
      BAIL_OUT('Failed to reset the msandbox password on the master or slave '
         . 'sandbox.  Check the Maatkit test environment with "test-env '
         . 'status" and restart with "test-env restart".  The error was: '
         . $EVAL_ERROR);
   }

};

# #############################################################################
# Done.
# #############################################################################
$sb->wipe_clean($dbh);
$sb->wipe_clean($dbh2) if $dbh2;
exit;