percona-toolkit/t/pt-kill/standard_options.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 => 6;

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

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

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

# #########################################################################
# Check that it daemonizes.
# #########################################################################

SKIP: {
   skip 'Cannot connect to sandbox master', 4 unless $master_dbh;

   # There's no hung queries so we'll just make sure it outputs anything,
   # its debug stuff in this case.
   `$cmd --print --interval 1s --run-time 2 --pid /tmp/pt-kill.pid --log /tmp/pt-kill.log --daemonize`;
   $output = `ps -eaf | grep 'pt-kill \-F'`;
   like(
      $output,
      qr/pt-kill -F /,
      'It lives daemonized'
   );
   ok(
      -f '/tmp/pt-kill.pid',
      'PID file created'
   );
   ok(
      -f '/tmp/pt-kill.log',
      'Log file created'
   );

   wait_until(sub { return !-f '/tmp/pt-kill.pid' });
   ok(
      !-f '/tmp/pt-kill.pid',
      'PID file removed'
   );

   diag(`rm -rf /tmp/pt-kill.log 2>/dev/null`);
}

# #########################################################################
# Issue 391: Add --pid option to all scripts
# #########################################################################
diag(`touch /tmp/pt-script.pid`);
$output = `$cmd --test-matching $trunk/t/lib/samples/pl/recset006.txt --match-state Locked  --print --pid /tmp/pt-script.pid 2>&1`;
like(
   $output,
   qr{PID file /tmp/pt-script.pid exists},
   'Dies if PID file already exists (--pid without --daemonize) (issue 391)'
);
diag(`rm -rf /tmp/pt-script.pid 2>/dev/null`);

# #############################################################################
# Done.
# #############################################################################
$sb->wipe_clean($master_dbh) if $master_dbh;
ok($sb->ok(), "Sandbox servers") or BAIL_OUT(__FILE__ . " broke the sandbox");
exit;