mirror of
https://github.com/pgbackrest/pgbackrest.git
synced 2024-12-16 10:20:02 +02:00
d211c2b8b5
The file write object destructors called close() and finalized the file even if it was not completely written. This was an issue in both the C and Perl code. Rewrite the destructors to simply free resources (like file handles) rather than calling the close() method. This leaves the temp file in place for filesystems that use temp files. Add unit tests to prevent regression. Reported by blogh.
220 lines
12 KiB
Perl
220 lines
12 KiB
Perl
####################################################################################################################################
|
|
# S3 Storage Tests
|
|
####################################################################################################################################
|
|
package pgBackRestTest::Module::Storage::StorageS3PerlTest;
|
|
use parent 'pgBackRestTest::Env::S3EnvTest';
|
|
|
|
####################################################################################################################################
|
|
# Perl includes
|
|
####################################################################################################################################
|
|
use strict;
|
|
use warnings FATAL => qw(all);
|
|
use Carp qw(confess);
|
|
use English '-no_match_vars';
|
|
|
|
use pgBackRest::Common::Log;
|
|
use pgBackRest::Common::String;
|
|
use pgBackRest::LibC qw(:crypto);
|
|
use pgBackRest::Storage::S3::Driver;
|
|
|
|
use pgBackRestTest::Common::ExecuteTest;
|
|
use pgBackRestTest::Common::RunTest;
|
|
|
|
####################################################################################################################################
|
|
# initTest
|
|
####################################################################################################################################
|
|
sub initTest
|
|
{
|
|
my $self = shift;
|
|
|
|
executeTest("$self->{strS3Command} rm --recursive s3://pgbackrest-dev");
|
|
}
|
|
|
|
####################################################################################################################################
|
|
# run
|
|
####################################################################################################################################
|
|
sub run
|
|
{
|
|
my $self = shift;
|
|
|
|
# Initialize the driver
|
|
my $oS3 = $self->initS3();
|
|
my $oStorage = new pgBackRest::Storage::Local('', $oS3);
|
|
|
|
# Test variables
|
|
my $strFile = 'file.txt';
|
|
my $strFileContent = 'TESTDATA';
|
|
my $iFileLength = length($strFileContent);
|
|
|
|
################################################################################################################################
|
|
if ($self->begin('exists()'))
|
|
{
|
|
#---------------------------------------------------------------------------------------------------------------------------
|
|
$self->testResult(sub {$oStorage->exists($strFile)}, false, 'root file does not exist');
|
|
|
|
#---------------------------------------------------------------------------------------------------------------------------
|
|
storageTest()->put($strFile, $strFileContent);
|
|
executeTest("$self->{strS3Command} cp " . $self->testPath() . "/${strFile} s3://pgbackrest-dev");
|
|
|
|
$self->testResult(sub {$oStorage->exists($strFile)}, true, 'root file exists');
|
|
|
|
#---------------------------------------------------------------------------------------------------------------------------
|
|
$self->testResult(sub {$oStorage->pathExists('/path/to')}, false, 'sub path does not exist');
|
|
$self->testResult(sub {$oStorage->exists("/path/to/${strFile}")}, false, 'sub file does not exist');
|
|
|
|
#---------------------------------------------------------------------------------------------------------------------------
|
|
executeTest("$self->{strS3Command} cp " . $self->testPath() . "/${strFile} s3://pgbackrest-dev/path/to/${strFile}");
|
|
|
|
$self->testResult(sub {$oStorage->pathExists('/path/to')}, true, 'sub path exists');
|
|
# $oStorage->pathExists('/path/to');
|
|
$self->testResult(sub {$oStorage->exists("/path/to/${strFile}")}, true, 'sub file exists');
|
|
}
|
|
|
|
################################################################################################################################
|
|
if ($self->begin('manifest()'))
|
|
{
|
|
#---------------------------------------------------------------------------------------------------------------------------
|
|
$self->testResult(sub {$oStorage->manifest('')}, '{. => {type => d}}', 'no files');
|
|
|
|
#---------------------------------------------------------------------------------------------------------------------------
|
|
storageTest()->put($strFile, $strFileContent);
|
|
storageTest()->put("${strFile}2", $strFileContent . '2');
|
|
|
|
executeTest("$self->{strS3Command} cp " . $self->testPath() . "/${strFile} s3://pgbackrest-dev");
|
|
executeTest("$self->{strS3Command} cp " . $self->testPath() . "/${strFile}2 s3://pgbackrest-dev/path/to/${strFile}2");
|
|
|
|
$self->testResult(
|
|
sub {$oStorage->manifest('')},
|
|
'{. => {type => d}, file.txt => {size => 8, type => f}, path => {type => d}, path/to => {type => d},' .
|
|
' path/to/file.txt2 => {size => 9, type => f}}',
|
|
'root path');
|
|
$self->testResult(
|
|
sub {$oStorage->manifest('/path/to')}, '{. => {type => d}, file.txt2 => {size => 9, type => f}}', 'sub path');
|
|
}
|
|
|
|
################################################################################################################################
|
|
if ($self->begin('list()'))
|
|
{
|
|
#---------------------------------------------------------------------------------------------------------------------------
|
|
$self->testResult(
|
|
sub {$oStorage->list('')}, '[undef]', 'no files');
|
|
|
|
#---------------------------------------------------------------------------------------------------------------------------
|
|
storageTest()->put($strFile, $strFileContent);
|
|
storageTest()->put("${strFile}2", $strFileContent . '2');
|
|
|
|
executeTest("$self->{strS3Command} cp " . $self->testPath() . "/${strFile} s3://pgbackrest-dev");
|
|
executeTest("$self->{strS3Command} cp " . $self->testPath() . "/${strFile}2 s3://pgbackrest-dev/path/to/${strFile}2");
|
|
|
|
$self->testResult(sub {$oStorage->list('')}, '(file.txt, path)', 'root path');
|
|
$self->testResult(sub {$oStorage->list('/path/to')}, 'file.txt2', 'sub path');
|
|
}
|
|
|
|
################################################################################################################################
|
|
if ($self->begin('remove()'))
|
|
{
|
|
#---------------------------------------------------------------------------------------------------------------------------
|
|
$oStorage->put($strFile, $strFileContent);
|
|
$oStorage->put("/path/to/${strFile}2", $strFileContent);
|
|
$oStorage->put("/path/to/${strFile}3", $strFileContent);
|
|
$oStorage->put("/path/to/${strFile}4 \@+", $strFileContent);
|
|
|
|
$self->testResult(
|
|
sub {$oStorage->manifest('/')},
|
|
'{. => {type => d}, file.txt => {size => 8, type => f}, path => {type => d}, path/to => {type => d},' .
|
|
' path/to/file.txt2 => {size => 8, type => f}, path/to/file.txt3 => {size => 8, type => f},' .
|
|
' path/to/file.txt4 @+ => {size => 8, type => f}}',
|
|
'check manifest');
|
|
|
|
#---------------------------------------------------------------------------------------------------------------------------
|
|
$self->testResult(sub {$oStorage->remove('/path/to', {bRecurse => true})}, true, 'remove subpath');
|
|
|
|
$self->testResult(
|
|
sub {$oStorage->manifest('/')},
|
|
'{. => {type => d}, file.txt => {size => 8, type => f}}', 'check manifest');
|
|
|
|
#---------------------------------------------------------------------------------------------------------------------------
|
|
$self->testResult(sub {$oStorage->remove($strFile)}, true, 'remove file');
|
|
|
|
$self->testResult(sub {$oStorage->manifest('/')}, '{. => {type => d}}', 'check manifest');
|
|
}
|
|
|
|
################################################################################################################################
|
|
if ($self->begin('info()'))
|
|
{
|
|
#---------------------------------------------------------------------------------------------------------------------------
|
|
storageTest()->put($strFile, $strFileContent);
|
|
storageTest()->put("${strFile}2", $strFileContent . '2');
|
|
|
|
executeTest("$self->{strS3Command} cp " . $self->testPath() . "/${strFile} s3://pgbackrest-dev");
|
|
executeTest("$self->{strS3Command} cp " . $self->testPath() . "/${strFile}2 s3://pgbackrest-dev");
|
|
executeTest("$self->{strS3Command} cp " . $self->testPath() . "/${strFile}2 s3://pgbackrest-dev/path/to/${strFile}2");
|
|
|
|
$self->testResult(sub {$oStorage->info($strFile)->size()}, 8, 'file size');
|
|
$self->testResult(sub {$oStorage->info("/path/to/${strFile}2")->size()}, 9, 'file 2 size');
|
|
}
|
|
|
|
################################################################################################################################
|
|
if ($self->begin('openRead() && S3::FileRead'))
|
|
{
|
|
# Create a random 1mb file
|
|
my $strRandomFile = $self->testPath() . '/random@1mb.bin';
|
|
executeTest("dd if=/dev/urandom of=${strRandomFile} bs=1024k count=1", {bSuppressStdErr => true});
|
|
my $strRandom = ${storageTest()->get($strRandomFile)};
|
|
|
|
executeTest("$self->{strS3Command} cp ${strRandomFile} s3://pgbackrest-dev/path/to/${strFile}");
|
|
|
|
#---------------------------------------------------------------------------------------------------------------------------
|
|
my $tBuffer;
|
|
my $oFileRead = $self->testResult(sub {$oS3->openRead("/path/to/${strFile}")}, '[object]', 'open read');
|
|
$self->testResult(sub {$oFileRead->read(\$tBuffer, 524288)}, 524288, ' read half');
|
|
$self->testResult(sub {$oFileRead->read(\$tBuffer, 524288)}, 524288, ' read half');
|
|
$self->testResult(sub {$oFileRead->read(\$tBuffer, 512)}, 0, ' read 0');
|
|
$self->testResult(length($tBuffer), 1048576, ' check length');
|
|
$self->testResult(cryptoHashOne('sha1', $tBuffer), cryptoHashOne('sha1', $strRandom), ' check hash');
|
|
}
|
|
|
|
################################################################################################################################
|
|
if ($self->begin('openWrite() && S3::FileWrite'))
|
|
{
|
|
# Create a random 1mb file
|
|
my $strRandomFile = $self->testPath() . '/random1mb.bin';
|
|
executeTest("dd if=/dev/urandom of=${strRandomFile} bs=1024k count=1", {bSuppressStdErr => true});
|
|
my $strRandom = ${storageTest()->get($strRandomFile)};
|
|
|
|
#---------------------------------------------------------------------------------------------------------------------------
|
|
my $oFileWrite = $self->testResult(sub {$oS3->openWrite("/path/to/${strFile}")}, '[object]', 'open write');
|
|
$self->testResult(sub {$oFileWrite->name()}, "/path/to/${strFile}", ' check filename');
|
|
$self->testResult(sub {$oFileWrite->close()}, true, ' close without writing');
|
|
|
|
#---------------------------------------------------------------------------------------------------------------------------
|
|
$oFileWrite = $self->testResult(sub {$oS3->openWrite("/path/to/${strFile}" . '.@')}, '[object]', 'open write');
|
|
$self->testResult(sub {$oFileWrite->write()}, 0, ' write undef');
|
|
$self->testResult(sub {$oFileWrite->write(\$strFileContent)}, $iFileLength, ' write');
|
|
$oFileWrite->close();
|
|
|
|
$self->testResult(sub {$oS3->exists("/path/to/${strFile}" . '.@')}, true, 'destination file exists');
|
|
|
|
# Test that a premature destroy (from error or otherwise) does not rename the file
|
|
#---------------------------------------------------------------------------------------------------------------------------
|
|
$oFileWrite = $self->testResult(sub {$oS3->openWrite("/path/to/abort.file" . '.@')}, '[object]', 'open write');
|
|
$self->testResult(sub {$oFileWrite->write()}, 0, ' write undef');
|
|
$self->testResult(sub {$oFileWrite->write(\$strFileContent)}, $iFileLength, ' write');
|
|
|
|
undef($oFileWrite);
|
|
$self->testResult(sub {$oS3->exists("/path/to/abort.file")}, false, 'destination file does not exist');
|
|
|
|
#---------------------------------------------------------------------------------------------------------------------------
|
|
$oFileWrite = $self->testResult(sub {$oS3->openWrite("/path/to/${strFile}")}, '[object]', 'open write');
|
|
|
|
for (my $iIndex = 1; $iIndex <= 17; $iIndex++)
|
|
{
|
|
$self->testResult(sub {$oFileWrite->write(\$strRandom)}, 1024 * 1024, ' write 1mb');
|
|
}
|
|
|
|
$self->testResult(sub {$oFileWrite->close()}, true, ' close');
|
|
}
|
|
}
|
|
|
|
1;
|