# Copyright (c) 2021-2025, 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',
  '--pgdata' => $backup_path,
  '--manifest-checksums' => $algorithm,
  '--no-sync',
  '--checkpoint' => 'fast');
 my @verify = ('pg_verifybackup', '--exit-on-error', $backup_path);

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

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

 # 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();