1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
#!perl
## Test the "custom_query" action
use 5.006;
use strict;
use warnings;
use Data::Dumper;
use Test::More tests => 12;
use lib 't','.';
use CP_Testing;
use vars qw/$dbh $result $t $host $dbname/;
my $good_query = q{SELECT count(*) AS result FROM pg_database};
my $bad_query = q{THIS IS NOT A QUERY};
my $cp = CP_Testing->new( {default_action => 'custom_query'} );
$dbh = $cp->test_database_handle();
$dbh->{AutoCommit} = 1;
$dbname = $cp->get_dbname;
$host = $cp->get_host();
my $S = q{Action 'custom_query'};
my $label = 'POSTGRES_CUSTOM_QUERY';
$t = qq{$S self-identifies correctly};
$result = $cp->run(qq{-w 0 --query="$good_query"});
like ($result, qr{^$label}, $t);
$t = qq{$S identifies host};
like ($result, qr{host:$host}, $t);
$t = qq{$S fails when called with an invalid option};
like ($cp->run('foobar=12'), qr{Usage:}, $t);
$t = qq{$S handles 'string' type for non-match};
like ($cp->run(qq{--query="$good_query" --valtype=string --warning=abc}),
qr{$label OK}, $t);
$t = qq{$S handles 'string' type for match};
like ($cp->run(q{--query="SELECT 'abc' AS result" --valtype=string --warning=abc}),
qr{$label WARNING}, $t);
$t = qq{$S handles 'time' type};
like ($cp->run(qq{--query="$good_query" --valtype=time --warning='1 second'}),
qr{$label WARNING}, $t);
$t = qq{$S handles invalid 'time' arg};
like ($cp->run(qq{--query="$good_query" --valtype=time --warning=foobar}),
qr{ERROR: Value for 'warning' must be a valid time.}, $t);
$t = qq{$S handles 'size' type};
like ($cp->run(qq{--query="$good_query" --valtype=size --warning=1c}),
qr{$label WARNING}, $t);
$t = qq{$S handles invalid 'size' arg};
like ($cp->run(qq{--query="$good_query" --valtype=size --warning=foobar}),
qr{ERROR: Invalid size for 'warning' option}, $t);
$t = qq{$S handles 'integer' type};
like ($cp->run(qq{--query="$good_query" --valtype=integer --warning=1}),
qr{$label WARNING}, $t);
$t = qq{$S fails when called with an invalid --warning};
like ($cp->run(qq{--query="$good_query" --valtype=integer --warning=a}),
qr{ERROR: Invalid argument for 'warning' option: must be an integer}, $t);
$t = qq{$S fails when called with an invalid query};
like ($cp->run(qq{--query="$bad_query" --warning=0}),
qr{\Q$bad_query\E}, $t);
exit;
|
