Add MySQLStatusWaiter.pm.

t/lib/MySQLStatusWaiter.t

@@ -0,0 +1,221 @@
+#!/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 => 12;
+
+use MySQLStatusWaiter;
+use PerconaTest;
+
+my $oktorun = 1;
+my @checked = ();
+my $slept   = 0;
+my @vals    = ();
+
+sub oktorun {
+   return $oktorun;
+}
+
+sub get_status {
+   my ($var) = @_;
+   push @checked, $var;
+   my $vals = shift @vals;
+   if ( $vals ) {
+      return $vals->{$var} || 0;
+   }
+   $oktorun = 0;
+   return;
+}
+
+sub sleep {
+   $slept++;
+   return;
+}
+
+# ############################################################################
+# Use initial vals + 20%.
+# ############################################################################
+@vals = (
+   # initial values
+   { Threads_connected => 10, },
+   { Threads_running   => 5,  },
+
+   # first check, no wait
+   { Threads_connected => 1, },
+   { Threads_running   => 1, },
+
+   # second check, wait
+   { Threads_connected => 12, }, # too high
+   { Threads_running   => 6,  }, # too high
+
+   # third check, wait
+   { Threads_connected => 12, }, # too high
+   { Threads_running   => 5,  },
+
+   # fourth check, wait
+   { Threads_connected => 10, },
+   { Threads_running   => 6,  }, # too high
+   
+   # fifth check, no wait
+   { Threads_connected => 10, },
+   { Threads_running   => 5,  },
+);
+
+my $sw = new MySQLStatusWaiter(
+   oktorun    => \&oktorun,
+   get_status => \&get_status,
+   sleep      => \&sleep,
+   spec       => [qw(Threads_connected Threads_running)],
+);
+
+is_deeply(
+   $sw->max_values(),
+   {
+      Threads_connected => int(10 + (10 * 0.20)),
+      Threads_running   => int(5  + (5  * 0.20)),
+   },
+   "Initial values +20%"
+);
+
+# first check
+@checked = ();
+$slept   = 0;
+$sw->wait();
+
+is_deeply(
+   \@checked,
+   [qw(Threads_connected Threads_running)],
+   "Checked both vars"
+);
+
+is(
+   $slept,
+   0,
+   "Vals not too high, did not sleep"
+);
+
+# second through fifth checks
+@checked = ();
+$slept   = 0;
+$sw->wait();
+
+is_deeply(
+   \@checked,
+   [qw(
+      Threads_connected Threads_running
+      Threads_connected Threads_running
+      Threads_connected Threads_running
+      Threads_connected Threads_running
+   )],
+   "Rechecked all variables"
+);
+
+is(
+   $slept,
+   3,
+   "Slept until values low enough"
+);
+
+# ############################################################################
+# Use static vals.
+# ############################################################################
+@vals = (
+   # first check, no wait
+   { Threads_connected => 1, },
+   { Threads_running   => 1, },
+);
+
+$sw = new MySQLStatusWaiter(
+   oktorun    => \&oktorun,
+   get_status => \&get_status,
+   sleep      => \&sleep,
+   spec       => [qw(Threads_connected=5 Threads_running=5)],
+);
+
+is_deeply(
+   $sw->max_values(),
+   {
+      Threads_connected => 5,
+      Threads_running   => 5,
+   },
+   "Static max values"
+);
+
+# first check
+@checked = ();
+$slept   = 0;
+$sw->wait();
+
+is_deeply(
+   \@checked,
+   [qw(Threads_connected Threads_running)],
+   "Checked both vars"
+);
+
+is(
+   $slept,
+   0,
+   "Vals not too high, did not sleep"
+);
+
+# ############################################################################
+# No spec, no wait.
+# ############################################################################
+@vals = (
+   # first check, no wait
+   { Threads_connected => 1, },
+   { Threads_running   => 1, },
+);
+
+$sw = new MySQLStatusWaiter(
+   oktorun    => \&oktorun,
+   get_status => \&get_status,
+   sleep      => \&sleep,
+   spec       => [],
+);
+
+is(
+   $sw->max_values(),
+   undef,
+   "No spec, no max values"
+);
+
+# first check
+@checked = ();
+$slept   = 0;
+$sw->wait();
+
+is_deeply(
+   \@checked,
+   [],
+   "No spec, no vars checked"
+);
+
+is(
+   $slept,
+   0,
+   "No spec, no sleep"
+);
+
+# #############################################################################
+# Done.
+# #############################################################################
+my $output = '';
+{
+   local *STDERR;
+   open STDERR, '>', \$output;
+   $sw->_d('Complete test coverage');
+}
+like(
+   $output,
+   qr/Complete test coverage/,
+   '_d() works'
+);
+exit;