First working scaffolding and fondation of host-to-host comparison.

2025-09-29 01:21:37 +00:00 · 2013-02-14 19:33:58 -07:00
parent e677ae8483
commit 550a97b9c2
4 changed files with 1833 additions and 130 deletions
--- a/t/lib/QueryIterator.t
+++ b/t/lib/QueryIterator.t
@@ -0,0 +1,132 @@
+#!/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;
+use Data::Dumper;
+
+use FileIterator;
+use QueryRewriter;
+use QueryIterator;
+use SlowLogParser;
+use PerconaTest;
+
+my $parser    = SlowLogParser->new();
+my $qr        = QueryRewriter->new();
+my $file_iter = FileIterator->new();
+
+my $oktorun = 1;
+my $sample  = "t/lib/samples/slowlogs";
+
+sub test_query_iter {
+   my (%args) = @_;
+
+   my $files = $file_iter->get_file_itr(
+      @{$args{files}}
+   );
+
+   my $query_iter = QueryIterator->new(
+      file_iter   => $files,
+      parser      => $args{parser} || $parser,
+      fingerprint => sub { return $qr->fingerprint(@_) },
+      oktorun     => sub { return $oktorun },
+      # Optional args
+      default_database => $args{default_database},
+      ($args{filter}       ? (filter       => $args{filter})       : ()),
+      ($args{read_only}    ? (read_only    => $args{read_only})    : ()),
+      ($args{read_timeout} ? (read_timeout => $args{read_timeout}) : ()),
+   );
+
+   my @events;
+   while ( my $event = $query_iter->next() ) {
+      push @events, $event;
+   }
+
+   is_deeply(
+      \@events,
+      $args{expect},
+      $args{name}
+   ) or diag(Dumper(\@events));
+}
+
+my $slow001_events = [
+   {
+      Lock_time => '0',
+      Query_time => '2',
+      Rows_examined => '0',
+      Rows_sent => '1',
+      arg => 'select sleep(2) from n',
+      bytes => 22,
+      cmd => 'Query',
+      db => 'test',
+      fingerprint => 'select sleep(?) from n',
+      host => 'localhost',
+      ip => '',
+      pos_in_log => 0,
+      ts => '071015 21:43:52',
+      user => 'root',
+   },
+   {
+      Lock_time => '0',
+      Query_time => '2',
+      Rows_examined => '0',
+      Rows_sent => '1',
+      arg => 'select sleep(2) from test.n',
+      bytes => 27,
+      cmd => 'Query',
+      db => 'sakila',
+      fingerprint => 'select sleep(?) from test.n',
+      host => 'localhost',
+      ip => '',
+      pos_in_log => 359,
+      ts => '071015 21:45:10',
+      user => 'root',
+   }
+];
+
+test_query_iter(
+   name   => "slow001.txt, defaults",
+   files  => [
+      "$trunk/$sample/slow001.txt"     
+   ],
+   expect => $slow001_events,
+);
+
+test_query_iter(
+   name         => "slow001.txt, read_timeout=5",
+   read_timeout => 5,
+   files        => [
+      "$trunk/$sample/slow001.txt"     
+   ],
+   expect => $slow001_events,
+);
+
+test_query_iter(
+   name      => "slow001.txt, read_only",
+   read_only => 1,
+   files     => [
+      "$trunk/$sample/slow001.txt"     
+   ],
+   expect => $slow001_events,
+);
+
+test_query_iter(
+   name   => "slow001.txt, in-line filter",
+   filter => '$event->{db} eq "test"',
+   files  => [
+      "$trunk/$sample/slow001.txt"     
+   ],
+   expect => [ $slow001_events->[0] ],
+);
+
+# #############################################################################
+# Done.
+# #############################################################################
+done_testing;