Add test for --action=wal_files --lsfunc

bucardo · martinmarques · Oct 14, 2014 · Oct 14, 2014 · Jun 2, 2015 · Jun 23, 2015
commit 85e7da645dd611f9948d1e2bfbaaf09c71fd52b3
diff --git a/check_postgres.pl b/check_postgres.pl
@@ -10178,6 +10178,10 @@ =head1 HISTORY
   Add Spanish message translations
     (Luis Vazquez)
 
+  Allow a wrapper function to run wal_files and archive_ready actions as
+  non-superuser
+    (Joshua Elsasser)
+
 =item B<Version 2.22.0> June 30, 2015
 
   Add xact timestamp support to hot_standby_delay.

diff --git a/t/02_wal_files.t b/t/02_wal_files.t
@@ -6,7 +6,7 @@ use 5.006;
 use strict;
 use warnings;
 use Data::Dumper;
-use Test::More tests => 11;
+use Test::More tests => 12;
 use lib 't','.';
 use CP_Testing;
 
@@ -64,6 +64,18 @@ is ($cp->run('--critical=1 --output=mrtg'), "99\n0\n\n\n", $t);
 $t=qq{$S returns correct MRTG information};
 is ($cp->run('--critical=101 --output=mrtg'), "99\n0\n\n\n", $t);
 
+# test --lsfunc
+$dbh->do(q{CREATE FUNCTION ls_xlog_dir()
+      RETURNS SETOF TEXT
+      AS $$ SELECT pg_ls_dir('pg_xlog') $$
+      LANGUAGE SQL
+      SECURITY DEFINER});
+$cp->create_fake_pg_table('ls_xlog_dir', ' ');
+$dbh->do(q{INSERT INTO cptest.ls_xlog_dir SELECT 'ABCDEF123456ABCDEF123456' FROM generate_series(1,55)});
+$dbh->commit();
+$t=qq{$S returns correct number of files};
+like ($cp->run('--critical=1 --lsfunc=ls_xlog_dir'), qr{^$label CRITICAL.+ 55 \|}, $t);
+
 $cp->drop_schema_if_exists();
 
 exit;