percona-toolkit/t/pt-kill/execute_command.t

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

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

my $dp = new DSNParser(opts=>$dsn_opts);
my $sb = new Sandbox(basedir => '/tmp', DSNParser => $dp);
my $source_dbh = $sb->get_dbh_for('source');

my $output;
my $cnf ='/tmp/12345/my.sandbox.cnf';
my $cmd = "$trunk/bin/pt-kill -F $cnf -h 127.1";
my $out = "/tmp/mk-kill-test.txt";

# #############################################################################
# Test --execute-command action.
# #############################################################################
diag(`rm $out 2>/dev/null`);

$output = `$cmd --test-matching $trunk/t/lib/samples/pl/recset001.txt --match-command Query --execute-command 'echo hello > $out'`;
is(
   $output, 
   '',
   'No output without --print'
);

chomp($output = `cat $out 2>/dev/null`),
is(
   $output,
   'hello',
   '--execute-command'
);

diag(`rm $out 2>/dev/null`);

SKIP: {
   skip 'Cannot connect to sandbox source', 2 unless $source_dbh;
   $source_dbh->do("CREATE DATABASE IF NOT EXISTS pt_kill_zombie_test");

   system "/tmp/12345/use -e 'select sleep(2)' >/dev/null 2>&1 &";

   $output = `$cmd --match-info 'select sleep' --run-time 2 --interval 1 --print --execute-command 'echo batty > $out'`;

   like(
      $output,
      qr/KILL .+ select sleep\(2\)/,
      '--print with --execute-command'
   );

   chomp($output = `cat $out 2>/dev/null`);
   is(
      $output,
      'batty',
      '--execute-command (online)'
   );

   # Let our select sleep(2) go away before other tests are ran.
   sleep 1;
   diag(`rm $out 2>/dev/null`);

   # Don't make zombies (https://bugs.launchpad.net/percona-toolkit/+bug/919819)

   # We have to store existing zombies, otherwise test occasionally fails
   my $our_zombies = `ps x | grep Z | grep -v grep`;

   $source_dbh->do("USE pt_kill_zombie_test");

   my $sentinel = "/tmp/pt-kill-test.$PID.stop";
   my $pid_file = "/tmp/pt-kill-test.$PID.pid";
   my $log_file = "/tmp/pt-kill-test.$PID.log";
   diag(`rm $sentinel 2>/dev/null`);
   diag(`rm $pid_file 2>/dev/null`);
   diag(`rm $log_file 2>/dev/null`);

   `$cmd --daemonize --match-db pt_kill_zombie_test --interval 1 --print --execute-command 'echo zombie > $out' --verbose --pid $pid_file --log $log_file --sentinel $sentinel`;
   PerconaTest::wait_for_files($pid_file, $log_file, $out);
   $output = `grep Executed $log_file`;
   like(
      $output,
      qr/Executed echo zombie/,
      "Executed zombie command"
   );

   $output = `ps x | grep Z | grep -v grep`;
   is(
      $output,
      $our_zombies,
      "No zombies"
   ) or diag($output);

   diag(`touch $sentinel`);
   sleep 1;
   ok(
      !-f $pid_file,
      "pt-kill stopped"
   );
   $output = `ps x | grep Z | grep -v grep`;
   is(
      $output,
      $our_zombies,
      "No zombies"
   ) or diag($output);

   diag(`rm $sentinel 2>/dev/null`);
   diag(`rm $pid_file 2>/dev/null`);
   diag(`rm $log_file 2>/dev/null`);
}

# #############################################################################
# Done.
# #############################################################################
diag(`rm $out 2>/dev/null`);
$sb->wipe_clean($source_dbh) if $source_dbh;
ok($sb->ok(), "Sandbox servers") or BAIL_OUT(__FILE__ . " broke the sandbox");
exit;