# Copyright (c) 2021-2024, PostgreSQL Global Development Group

# Verify that we can take and verify backups with various checksum types.

use strict;
use warnings FATAL => 'all';
use File::Path qw(rmtree);
use PostgreSQL::Test::Cluster;
use PostgreSQL::Test::Utils;
use Test::More;

my $primary = PostgreSQL::Test::Cluster->new('primary');
$primary->init(allows_streaming => 1);
$primary->start;

sub test_checksums
{
	my ($format, $algorithm) = @_;
	my $backup_path = $primary->backup_dir . '/' . $format . '/' . $algorithm;
	my @backup = (
		'pg_basebackup', '-D', $backup_path,
		'--manifest-checksums', $algorithm, '--no-sync', '-cfast');
	my @verify = ('pg_verifybackup', '-e', $backup_path);

	if ($format eq 'tar')
	{
		# Add switch to get a tar-format backup
		push @backup, ('-F', 't');

		# Add switch to skip WAL verification, which is not yet supported for
		# tar-format backups
		push @verify, ('-n');
	}

	# A backup with a bogus algorithm should fail.
	if ($algorithm eq 'bogus')
	{
		$primary->command_fails(\@backup,
			"$format format backup fails with algorithm \"$algorithm\"");
		return;
	}

	# A backup with a valid algorithm should work.
	$primary->command_ok(\@backup, "$format format backup ok with algorithm \"$algorithm\"");

	# We expect each real checksum algorithm to be mentioned on every line of
	# the backup manifest file except the first and last; for simplicity, we
	# just check that it shows up lots of times. When the checksum algorithm
	# is none, we just check that the manifest exists.
	if ($algorithm eq 'none')
	{
		ok(-f "$backup_path/backup_manifest", "$format format backup manifest exists");
	}
	else
	{
		my $manifest = slurp_file("$backup_path/backup_manifest");
		my $count_of_algorithm_in_manifest =
		  (() = $manifest =~ /$algorithm/mig);
		cmp_ok($count_of_algorithm_in_manifest,
			'>', 100, "$algorithm is mentioned many times in the manifest");
	}

	# Make sure that it verifies OK.
	$primary->command_ok(\@verify,
		"verify $format format backup with algorithm \"$algorithm\"");

	# Remove backup immediately to save disk space.
	rmtree($backup_path);
}

# Do the check
for my $format (qw(plain tar))
{
	for my $algorithm (qw(bogus none crc32c sha224 sha256 sha384 sha512))
	{
		test_checksums($format, $algorithm);
	}
}

done_testing();