Files
percona-toolkit/t/lib/QueryIterator.t

133 lines
3.1 KiB
Perl

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