From 9b815a8ff227e62442259e0fbabc5cf37e433df9 Mon Sep 17 00:00:00 2001
From: Stephen Frost <sfrost@snowman.net>
Date: Fri, 6 Jan 2017 16:29:31 -0500
Subject: [PATCH] Add basic pg_dumpall/pg_restore TAP tests

For reasons unknown, pg_dumpall and pg_restore managed to escape the
basic set of TAP tests that were added for pg_dump in 6bd356c3, so
let's get them added now.  A few minor adjustments are also made to the
dump/restore tests to improve code coverage for pg_restore/pg_dumpall.
---
 src/bin/pg_dump/t/001_basic.pl   | 61 +++++++++++++++++++++++++++++++-
 src/bin/pg_dump/t/002_pg_dump.pl |  8 ++---
 2 files changed, 64 insertions(+), 5 deletions(-)

diff --git a/src/bin/pg_dump/t/001_basic.pl b/src/bin/pg_dump/t/001_basic.pl
index 6d545b29e21..e11f3dff2c9 100644
--- a/src/bin/pg_dump/t/001_basic.pl
+++ b/src/bin/pg_dump/t/001_basic.pl
@@ -4,7 +4,7 @@ use warnings;
 use Config;
 use PostgresNode;
 use TestLib;
-use Test::More tests => 15;
+use Test::More tests => 42;
 
 my $tempdir       = TestLib::tempdir;
 my $tempdir_short = TestLib::tempdir_short;
@@ -16,6 +16,14 @@ program_help_ok('pg_dump');
 program_version_ok('pg_dump');
 program_options_handling_ok('pg_dump');
 
+program_help_ok('pg_restore');
+program_version_ok('pg_restore');
+program_options_handling_ok('pg_restore');
+
+program_help_ok('pg_dumpall');
+program_version_ok('pg_dumpall');
+program_options_handling_ok('pg_dumpall');
+
 #########################################
 # Test various invalid options and disallowed combinations
 # Doesn't require a PG instance to be set up, so do this first.
@@ -23,17 +31,40 @@ program_options_handling_ok('pg_dump');
 command_exit_is([ 'pg_dump', 'qqq', 'abc' ],
 	1, 'pg_dump: too many command-line arguments (first is "asd")');
 
+command_exit_is([ 'pg_restore', 'qqq', 'abc' ],
+	1, 'pg_restore too many command-line arguments (first is "asd")');
+
+command_exit_is([ 'pg_dumpall', 'qqq', 'abc' ],
+	1, 'pg_dumpall: too many command-line arguments (first is "qqq")');
+
 command_exit_is(
 	[ 'pg_dump', '-s', '-a' ],
 	1,
 'pg_dump: options -s/--schema-only and -a/--data-only cannot be used together'
 );
 
+command_exit_is(
+	[ 'pg_restore', '-s', '-a' ],
+	1,
+'pg_restore: options -s/--schema-only and -a/--data-only cannot be used together'
+);
+
+command_exit_is(
+	[ 'pg_restore', '-d', 'xxx', '-f', 'xxx' ],
+	1,
+'pg_restore: options -d/--dbname and -f/--file cannot be used together'
+);
+
 command_exit_is(
 	[ 'pg_dump', '-c', '-a' ],
 	1,
 	'pg_dump: options -c/--clean and -a/--data-only cannot be used together');
 
+command_exit_is(
+	[ 'pg_restore', '-c', '-a' ],
+	1,
+	'pg_restore: options -c/--clean and -a/--data-only cannot be used together');
+
 command_exit_is(
 	[ 'pg_dump', '--inserts', '-o' ],
 	1,
@@ -48,3 +79,31 @@ command_exit_is([ 'pg_dump', '-j' ],
 
 command_exit_is([ 'pg_dump', '-j3' ],
 	1, 'pg_dump: parallel backup only supported by the directory format');
+
+command_exit_is([ 'pg_restore', '--single-transaction', '-j3' ],
+	1, 'pg_restore: cannot specify both --single-transaction and multiple jobs');
+
+command_exit_is([ 'pg_restore', '--if-exists' ],
+	1, 'pg_restore: option --if-exists requires option -c/--clean');
+
+# pg_dumpall command-line argument checks
+command_exit_is(
+	[ 'pg_dumpall', '-g', '-r' ],
+	1,
+'pg_restore: options -g/--globals-only and -r/--roles-only cannot be used together'
+);
+
+command_exit_is(
+	[ 'pg_dumpall', '-g', '-t' ],
+	1,
+'pg_restore: options -g/--globals-only and -t/--tablespaces-only cannot be used together'
+);
+
+command_exit_is(
+	[ 'pg_dumpall', '-r', '-t' ],
+	1,
+'pg_restore: options -r/--roles-only and -t/--tablespaces-only cannot be used together'
+);
+
+command_exit_is([ 'pg_dumpall', '--if-exists' ],
+	1, 'pg_dumpall: option --if-exists requires option -c/--clean');
diff --git a/src/bin/pg_dump/t/002_pg_dump.pl b/src/bin/pg_dump/t/002_pg_dump.pl
index 59191ccecd3..b732627c3a2 100644
--- a/src/bin/pg_dump/t/002_pg_dump.pl
+++ b/src/bin/pg_dump/t/002_pg_dump.pl
@@ -88,7 +88,7 @@ my %pgdump_runs = (
 			"$tempdir/defaults_custom_format.dump", 'postgres', ],
 		restore_cmd => [
 			'pg_restore',
-			'-f',
+			'-Fc', '-f',
 			"$tempdir/defaults_custom_format.sql",
 			"$tempdir/defaults_custom_format.dump", ], },
 	defaults_dir_format => {
@@ -98,7 +98,7 @@ my %pgdump_runs = (
 			'postgres', ],
 		restore_cmd => [
 			'pg_restore',
-			'-f',
+			'-Fd', '-f',
 			"$tempdir/defaults_dir_format.sql",
 			"$tempdir/defaults_dir_format", ], },
 	defaults_parallel => {
@@ -117,7 +117,7 @@ my %pgdump_runs = (
 			'postgres', ],
 		restore_cmd => [
 			'pg_restore',
-			'-f',
+			'-Ft', '-f',
 			"$tempdir/defaults_tar_format.sql",
 			"$tempdir/defaults_tar_format.tar", ], },
 	exclude_dump_test_schema => {
@@ -136,7 +136,7 @@ my %pgdump_runs = (
 			'--exclude-table-data=dump_test.test_table', 'postgres', ], },
 	pg_dumpall_globals => {
 		dump_cmd =>
-		  [ 'pg_dumpall', '-f', "$tempdir/pg_dumpall_globals.sql", '-g', ], },
+		  [ 'pg_dumpall', '-v', '-f', "$tempdir/pg_dumpall_globals.sql", '-g', ], },
 	pg_dumpall_globals_clean => {
 		dump_cmd => [
 			'pg_dumpall', '-f', "$tempdir/pg_dumpall_globals_clean.sql",
-- 
GitLab