← Back to branch summary

~percona-toolkit-dev/percona-toolkit/pt-table-checksum-fails-on-BINARY-field-in-PK-1381280

« back to all changes in this revision

Viewing changes to t/pt-agent/schedule_services.t

  • Committer: Daniel Nichter
  • Date: 2014-05-30 01:09:13 UTC
  • mfrom: (598.5.6 release-2.2.8)
  • Revision ID: daniel@percona.com-20140530010913-4wep0en37aa4vvok
Merge release-2.2.8.

Show diffs side-by-side

added added

removed removed

Lines of Context:
t/pt-agent/schedule_services.t
1
 
#!/usr/bin/env perl
2
 
 
3
 
BEGIN {
4
 
   die "The PERCONA_TOOLKIT_BRANCH environment variable is not set.\n"
5
 
      unless $ENV{PERCONA_TOOLKIT_BRANCH} && -d $ENV{PERCONA_TOOLKIT_BRANCH};
6
 
   unshift @INC, "$ENV{PERCONA_TOOLKIT_BRANCH}/lib";
7
 
};
8
 
 
9
 
use strict;
10
 
use warnings FATAL => 'all';
11
 
use English qw(-no_match_vars);
12
 
use Test::More;
13
 
use JSON;
14
 
use File::Temp qw(tempfile tempdir);
15
 
 
16
 
use Percona::Test;
17
 
use Percona::Test::Mock::AgentLogger;
18
 
require "$trunk/bin/pt-agent";
19
 
 
20
 
my $crontab = `crontab -l 2>/dev/null`;
21
 
if ( $crontab ) {
22
 
   plan skip_all => 'Crontab is not empty';
23
 
}
24
 
 
25
 
Percona::Toolkit->import(qw(have_required_args Dumper));
26
 
 
27
 
my $sample = "t/pt-agent/samples";
28
 
my $tmpdir = tempdir("/tmp/pt-agent.$PID.XXXXXX", CLEANUP => 1);
29
 
 
30
 
my @log;
31
 
my $logger = Percona::Test::Mock::AgentLogger->new(log => \@log);
32
 
pt_agent::_logger($logger);
33
 
 
34
 
# #############################################################################
35
 
# Schedule a good crontab.
36
 
# #############################################################################
37
 
 
38
 
my $run0 = Percona::WebAPI::Resource::Task->new(
39
 
   name    => 'query-history',
40
 
   number  => '0',
41
 
   program => 'pt-query-digest',
42
 
   options => '--output json',
43
 
   output  => 'spool',
44
 
);
45
 
 
46
 
my $svc0 = Percona::WebAPI::Resource::Service->new(
47
 
   ts             => 100,
48
 
   name           => 'query-history',
49
 
   run_schedule   => '* 8 * * 1,2,3,4,5',
50
 
   spool_schedule => '* 9 * * 1,2,3,4,5',
51
 
   tasks          => [ $run0 ],
52
 
);
53
 
 
54
 
# First add a fake line so we can know that the real, existing
55
 
# crontab is used and not clobbered.
56
 
my ($fh, $file) = tempfile();
57
 
print {$fh} "* 0  *  *  *  date > /dev/null\n";
58
 
close $fh or warn "Cannot close $file: $OS_ERROR";
59
 
my $output = `crontab $file 2>&1`;
60
 
 
61
 
$crontab = `crontab -l 2>&1`;
62
 
 
63
 
is(
64
 
   $crontab,
65
 
   "* 0  *  *  *  date > /dev/null\n",
66
 
   "Set other crontab line"
67
 
) or diag($output);
68
 
 
69
 
unlink $file or warn "Cannot remove $file: $OS_ERROR";
70
 
 
71
 
eval {
72
 
   $output = output(
73
 
      sub {
74
 
         pt_agent::schedule_services(
75
 
            services => [ $svc0 ],
76
 
            lib_dir  => $tmpdir,
77
 
         )
78
 
      },
79
 
      stderr => 1,
80
 
   );
81
 
};
82
 
 
83
 
is(
84
 
   $EVAL_ERROR,
85
 
   "",
86
 
   "No error"
87
 
) or diag($output);
88
 
 
89
 
$crontab = `crontab -l 2>/dev/null`;
90
 
 
91
 
# pt-agent uses $FindBin::Bin/pt-agent for the path to pt-agent,
92
 
# which in testing will be $trunk/t/pt-agent/ because that's where
93
 
# this file is located.  However, if $FindBin::Bin resovles sym
94
 
# links where as $trunk does not, so to make things simple we just
95
 
# cut out the full path. 
96
 
if ( $crontab ) {
97
 
   $crontab =~ s! /.+?/pt-agent --! pt-agent --!g;
98
 
}
99
 
is(
100
 
   $crontab,
101
 
   "* 0  *  *  *  date > /dev/null
102
 
* 8 * * 1,2,3,4,5 pt-agent --run-service query-history
103
 
* 9 * * 1,2,3,4,5 pt-agent --send-data query-history
104
 
",
105
 
   "schedule_services()"
106
 
);
107
 
 
108
 
ok(
109
 
   -f "$tmpdir/crontab",
110
 
   "Wrote crontab to --lib/crontab"
111
 
) or diag(`ls -l $tmpdir`);
112
 
 
113
 
ok(
114
 
   -f "$tmpdir/crontab.err",
115
 
   "Write --lib/crontab.err",
116
 
) or diag(`ls -l $tmpdir`);
117
 
 
118
 
my $err = -f "$tmpdir/crontab.err" ? `cat $tmpdir/crontab.err` : '';
119
 
is(
120
 
   $err,
121
 
   "",
122
 
   "No crontab error"
123
 
);
124
 
 
125
 
system("crontab -r 2>/dev/null");
126
 
$crontab = `crontab -l 2>/dev/null`;
127
 
is(
128
 
   $crontab,
129
 
   "",
130
 
   "Removed crontab"
131
 
);
132
 
 
133
 
# #############################################################################
134
 
# Handle bad crontab lines.
135
 
# #############################################################################
136
 
 
137
 
$svc0 = Percona::WebAPI::Resource::Service->new(
138
 
   ts             => 100,
139
 
   name           => 'query-history',
140
 
   run_schedule   => '* * * * Foo',  # "foo":0: bad day-of-week
141
 
   spool_schedule => '* 8 * * Mon',
142
 
   tasks          => [ $run0 ],
143
 
);
144
 
 
145
 
eval {
146
 
   $output = output(
147
 
      sub {
148
 
         pt_agent::schedule_services(
149
 
            services => [ $svc0 ],
150
 
            lib_dir  => $tmpdir,
151
 
         ),
152
 
      },
153
 
      stderr => 1,
154
 
      die    => 1,
155
 
   );
156
 
};
157
 
 
158
 
like(
159
 
   $EVAL_ERROR,
160
 
   qr/Error setting new crontab/,
161
 
   "Throws errors"
162
 
) or diag($output);
163
 
 
164
 
$crontab = `crontab -l 2>/dev/null`;
165
 
is(
166
 
   $crontab,
167
 
   "",
168
 
   "Bad schedule_services()"
169
 
);
170
 
 
171
 
ok(
172
 
   -f "$tmpdir/crontab",
173
 
   "Wrote crontab to --lib/crontab"
174
 
) or diag(`ls -l $tmpdir`);
175
 
 
176
 
ok(
177
 
   -f "$tmpdir/crontab.err",
178
 
   "Write --lib/crontab.err",
179
 
) or diag(`ls -l $tmpdir`);
180
 
 
181
 
$err = -f "$tmpdir/crontab.err" ? `cat $tmpdir/crontab.err` : '';
182
 
like(
183
 
   $err,
184
 
   qr/bad/,
185
 
   "Crontab error"
186
 
);
187
 
 
188
 
system("crontab -r 2>/dev/null");
189
 
$crontab = `crontab -l 2>/dev/null`;
190
 
is(
191
 
   $crontab,
192
 
   "",
193
 
   "Removed crontab"
194
 
);
195
 
 
196
 
 
197
 
# #############################################################################
198
 
# Done.
199
 
# #############################################################################
200
 
done_testing;