Bug 20727: Move temporary_directory() to C4::Context
[koha.git] / t / db_dependent / Upload.t
index 49ec239..ccb879a 100644 (file)
@@ -2,7 +2,8 @@
 
 use Modern::Perl;
 use File::Temp qw/ tempdir /;
-use Test::More tests => 9;
+use Test::More tests => 13;
+use Test::Warn;
 
 use Test::MockModule;
 use t::lib::Mocks;
@@ -10,12 +11,14 @@ use t::lib::TestBuilder;
 
 use C4::Context;
 use Koha::Database;
+use Koha::DateUtils;
 use Koha::UploadedFile;
 use Koha::UploadedFiles;
 use Koha::Uploader;
 
 my $schema  = Koha::Database->new->schema;
 $schema->storage->txn_begin;
+our $builder = t::lib::TestBuilder->new;
 
 our $current_upload = 0;
 our $uploads = [
@@ -39,6 +42,10 @@ our $uploads = [
     [
         { name => 'file5', cat => undef, size => 7000 },
     ],
+    [
+        { name => 'file6', cat => undef, size => 6500 },
+        { name => 'file7', cat => undef, size => 6501 },
+    ],
 ];
 
 # Redirect upload dir structure and mock File::Spec and CGI
@@ -50,50 +57,29 @@ my $cgimod = Test::MockModule->new( 'CGI' );
 $cgimod->mock( 'new' => \&newCGI );
 
 # Start testing
-subtest 'Test01' => sub {
-    plan tests => 11;
-    test01();
-};
-subtest 'Test02' => sub {
-    plan tests => 5;
-    test02();
-};
-subtest 'Test03' => sub {
-    plan tests => 2;
-    test03();
-};
-subtest 'Test04' => sub {
-    plan tests => 3;
-    test04();
-};
-subtest 'Test05' => sub {
-    plan tests => 6;
-    test05();
-};
-subtest 'Test06' => sub {
-    plan tests => 3;
-    test06();
-};
-subtest 'Test07' => sub {
-    plan tests => 2;
-    test07();
-};
-subtest 'Test08: allows_add_by' => sub {
-    plan tests => 4;
-    test08();
-};
-$schema->storage->txn_rollback;
+subtest 'Make a fresh start' => sub {
+    plan tests => 1;
 
-sub test01 {
     # Delete existing records (for later tests)
-    # Passing keep_file suppresses warnings
-    Koha::UploadedFiles->new->delete({ keep_file => 1 });
+    # Passing keep_file suppresses warnings (and does not delete files)
+    # Note that your files are not in danger, since we redirected
+    # all files to a new empty temp folder
+    Koha::UploadedFiles->delete({ keep_file => 1 });
+    is( Koha::UploadedFiles->count, 0, 'No records left' );
+};
+
+subtest 'permanent_directory and temporary_directory' => sub {
+    plan tests => 2;
 
     # Check mocked directories
     is( Koha::UploadedFile->permanent_directory, $tempdir,
         'Check permanent directory' );
-    is( Koha::UploadedFile->temporary_directory, $tempdir,
+    is( C4::Context::temporary_directory, $tempdir,
         'Check temporary directory' );
+};
+
+subtest 'Add two uploads in category A' => sub {
+    plan tests => 9;
 
     my $upl = Koha::Uploader->new({
         category => $uploads->[$current_upload]->[0]->{cat},
@@ -115,9 +101,11 @@ sub test01 {
     is( $rec->filename, 'file2', 'Check file name 2' );
     is( $rec->filesize, 8000, 'Check size of file2' );
     is( $rec->public, undef, 'Check public undefined' );
-}
+};
+
+subtest 'Add another upload, check file_handle' => sub {
+    plan tests => 5;
 
-sub test02 {
     my $upl = Koha::Uploader->new({
         category => $uploads->[$current_upload]->[0]->{cat},
         public => 1,
@@ -135,17 +123,21 @@ sub test02 {
     $rec->filename( 'doesprobablynotexist' )->store;
     is( $rec->file_handle, undef, 'Sabotage with file handle' );
     $rec->filename( $orgname )->store;
-}
+};
+
+subtest 'Add temporary upload' => sub {
+    plan tests => 2;
 
-sub test03 {
     my $upl = Koha::Uploader->new({ tmp => 1 }); #temporary
     my $cgi= $upl->cgi;
     is( $upl->count, 1, 'Upload 3 includes one temporary file' );
     my $rec = Koha::UploadedFiles->find( $upl->result );
     is( $rec->uploadcategorycode =~ /_upload$/, 1, 'Check category temp file' );
-}
+};
+
+subtest 'Add same file in same category' => sub {
+    plan tests => 3;
 
-sub test04 { # Fail on a file already there
     my $upl = Koha::Uploader->new({
         category => $uploads->[$current_upload]->[0]->{cat},
     });
@@ -153,19 +145,22 @@ sub test04 { # Fail on a file already there
     is( $upl->count, 0, 'Upload 4 failed as expected' );
     is( $upl->result, undef, 'Result is undefined' );
     my $e = $upl->err;
-    is( $e->{file2}, 1, "Errcode 1 [already exists] reported" );
-}
+    is( $e->{file2}->{code}, 1, "Errcode 1 [already exists] reported" );
+};
+
+subtest 'Test delete via UploadedFile as well as UploadedFiles' => sub {
+    plan tests => 10;
 
-sub test05 { # add temporary file with same name and contents, delete it
+    # add temporary file with same name and contents (file4)
     my $upl = Koha::Uploader->new({ tmp => 1 });
     my $cgi= $upl->cgi;
-    is( $upl->count, 1, 'Upload 5 adds duplicate temporary file' );
+    is( $upl->count, 1, 'Add duplicate temporary file (file4)' );
     my $id = $upl->result;
     my $path = Koha::UploadedFiles->find( $id )->full_path;
 
     # testing delete via UploadedFiles (plural)
     my $delete = Koha::UploadedFiles->search({ id => $id })->delete;
-    is( $delete, 1, 'Delete successful' );
+    isnt( $delete, "0E0", 'Delete successful' );
     isnt( -e $path, 1, 'File no longer found after delete' );
     is( Koha::UploadedFiles->find( $id ), undef, 'Record also gone' );
 
@@ -174,14 +169,52 @@ sub test05 { # add temporary file with same name and contents, delete it
     $upl = Koha::Uploader->new({ tmp => 1 });
     $upl->cgi;
     my $kohaobj = Koha::UploadedFiles->find( $upl->result );
-    my $name = $kohaobj->filename;
     $path = $kohaobj->full_path;
     $delete = $kohaobj->delete;
-    is( $delete, $name, 'Delete successful' );
+    ok( $delete=~/^-?1$/, 'Delete successful' );
     isnt( -e $path, 1, 'File no longer found after delete' );
-}
 
-sub test06 { #search_term with[out] private flag
+    # add another record with TestBuilder, so file does not exist
+    # catch warning
+    my $upload01 = $builder->build({ source => 'UploadedFile' });
+    warning_like { $delete = Koha::UploadedFiles->find( $upload01->{id} )->delete; }
+        qr/file was missing/,
+        'delete warns when file is missing';
+    ok( $delete=~/^-?1$/, 'Deleting record was successful' );
+    is( Koha::UploadedFiles->count, 4, 'Back to four uploads now' );
+
+    # add another one with TestBuilder and delete twice (file does not exist)
+    $upload01 = $builder->build({ source => 'UploadedFile' });
+    $kohaobj = Koha::UploadedFiles->find( $upload01->{id} );
+    $delete = $kohaobj->delete({ keep_file => 1 });
+    $delete = $kohaobj->delete({ keep_file => 1 });
+    ok( $delete =~ /^(0E0|-1)$/, 'Repeated delete unsuccessful' );
+    # NOTE: Koha::Object->delete does not return 0E0 (yet?)
+};
+
+subtest 'Test delete_missing' => sub {
+    plan tests => 5;
+
+    # If we add files via TestBuilder, they do not exist
+    my $upload01 = $builder->build({ source => 'UploadedFile' });
+    my $upload02 = $builder->build({ source => 'UploadedFile' });
+    # dry run first
+    my $deleted = Koha::UploadedFiles->delete_missing({ keep_record => 1 });
+    is( $deleted, 2, 'Expect two records with missing files' );
+    isnt( Koha::UploadedFiles->find( $upload01->{id} ), undef, 'Not deleted' );
+    $deleted = Koha::UploadedFiles->delete_missing;
+    ok( $deleted =~ /^(2|-1)$/, 'Deleted two records with missing files' );
+    is( Koha::UploadedFiles->search({
+        id => [ $upload01->{id}, $upload02->{id} ],
+    })->count, 0, 'Records are gone' );
+    # Repeat it
+    $deleted = Koha::UploadedFiles->delete_missing;
+    is( $deleted, "0E0", "Return value of 0E0 expected" );
+};
+
+subtest 'Call search_term with[out] private flag' => sub {
+    plan tests => 3;
+
     my @recs = Koha::UploadedFiles->search_term({ term => 'file' });
     is( @recs, 1, 'Returns only one public result' );
     is( $recs[0]->filename, 'file3', 'Should be file3' );
@@ -189,20 +222,22 @@ sub test06 { #search_term with[out] private flag
     is( Koha::UploadedFiles->search_term({
         term => 'file', include_private => 1,
     })->count, 4, 'Returns now four results' );
-}
+};
+
+subtest 'Simple tests for httpheaders and getCategories' => sub {
+    plan tests => 2;
 
-sub test07 { #simple test for httpheaders and getCategories
     my $rec = Koha::UploadedFiles->search_term({ term => 'file' })->next;
     my @hdrs = $rec->httpheaders;
     is( @hdrs == 4 && $hdrs[1] =~ /application\/octet-stream/, 1, 'Simple test for httpheaders');
-    my $builder = t::lib::TestBuilder->new;
     $builder->build({ source => 'AuthorisedValue', value => { category => 'UPLOAD', authorised_value => 'HAVE_AT_LEAST_ONE', lib => 'Hi there' } });
-    my $cat = Koha::UploadedFile->getCategories;
+    my $cat = Koha::UploadedFiles->getCategories;
     is( @$cat >= 1, 1, 'getCategories returned at least one category' );
-}
+};
+
+subtest 'Testing allows_add_by' => sub {
+    plan tests => 4;
 
-sub test08 { # allows_add_by
-    my $builder = t::lib::TestBuilder->new;
     my $patron = $builder->build({
         source => 'Borrower',
         value  => { flags => 0 }, #no permissions
@@ -236,25 +271,66 @@ sub test08 { # allows_add_by
     });
     is( Koha::Uploader->allows_add_by( $patron->{userid} ),
         1, 'Patron is still allowed to add uploaded files' );
-}
+};
 
-# Additional tests for Koha::UploadedFiles
-# TODO Rearrange the tests after this migration
-subtest 'Some basic CRUD testing' => sub {
-    plan tests => 2;
+subtest 'Testing delete_temporary' => sub {
+    plan tests => 9;
 
-    # Test find and attribute id, delete and search
-    my $builder = t::lib::TestBuilder->new;
-    my $upload01 = $builder->build({ source => 'UploadedFile' });
-    my $found = Koha::UploadedFiles->find( $upload01->{id} );
-    is( $found->id, $upload01->{id}, 'Koha::Object returns id' );
-    $found->delete({ keep_file => 1 }); #note that it does not exist
-    $found = Koha::UploadedFiles->search(
-        { id => $upload01->{id} },
-    );
-    is( $found->count, 0, 'Delete seems successful' );
+    # Add two temporary files: result should be 3 + 3
+    Koha::Uploader->new({ tmp => 1 })->cgi; # add file6 and file7
+    is( Koha::UploadedFiles->search->count, 6, 'Test starting count' );
+    is( Koha::UploadedFiles->search({ permanent => 1 })->count, 3,
+        'Includes 3 permanent' );
+
+    # Move all permanents to today - 1
+    # Move temp 1 to today - 3, and temp 2,3 to today - 5
+    my $today = dt_from_string;
+    $today->subtract( minutes => 2 ); # should be enough :)
+    my $dt = $today->clone->subtract( days => 1 );
+    foreach my $rec ( Koha::UploadedFiles->search({ permanent => 1 }) ) {
+        $rec->dtcreated($dt)->store;
+    }
+    my @recs = Koha::UploadedFiles->search({ permanent => 0 });
+    $dt = $today->clone->subtract( days => 3 );
+    $recs[0]->dtcreated($dt)->store;
+    $dt = $today->clone->subtract( days => 5 );
+    $recs[1]->dtcreated($dt)->store;
+    $recs[2]->dtcreated($dt)->store;
+
+    # Now call delete_temporary with 6, 5 and 0
+    t::lib::Mocks::mock_preference('UploadPurgeTemporaryFilesDays', 6 );
+    my $delete = Koha::UploadedFiles->delete_temporary;
+    ok( $delete =~ /^(-1|0E0)$/, 'Check return value with 6' );
+    is( Koha::UploadedFiles->search->count, 6, 'Delete with pref==6' );
+
+    # use override parameter
+    $delete = Koha::UploadedFiles->delete_temporary({ override_pref => 5 });
+    ok( $delete =~ /^(2|-1)$/, 'Check return value with 5' );
+    is( Koha::UploadedFiles->search->count, 4, 'Delete with override==5' );
+
+    t::lib::Mocks::mock_preference('UploadPurgeTemporaryFilesDays', 0 );
+    $delete = Koha::UploadedFiles->delete_temporary;
+    ok( $delete =~ /^(-1|1)$/, 'Check return value with 0' );
+    is( Koha::UploadedFiles->search->count, 3, 'Delete with pref==0 makes 3' );
+    is( Koha::UploadedFiles->search({ permanent => 1 })->count, 3,
+        'Still 3 permanent uploads' );
 };
 
+subtest 'Testing download headers' => sub {
+    plan tests => 2;
+    my $test_pdf = Koha::UploadedFile->new({ filename => 'pdf.pdf', uploadcategorycode => 'B', filesize => 1000 });
+    my $test_not = Koha::UploadedFile->new({ filename => 'pdf.not', uploadcategorycode => 'B', filesize => 1000 });
+    my @pdf_expect = ( '-type'=>'application/pdf','Content-Disposition'=>'inline; filename=pdf.pdf' );
+    my @not_expect = ( '-type'=>'application/octet-stream','-attachment'=>'pdf.not' );
+    my @pdf_head = $test_pdf->httpheaders;
+    my @not_head = $test_not->httpheaders;
+    is_deeply(\@pdf_head, \@pdf_expect,"Get inline pdf headers for pdf");
+    is_deeply(\@not_head, \@not_expect,"Get download headers for non pdf");
+};
+# The end
+$schema->storage->txn_rollback;
+
+# Helper routine
 sub newCGI {
     my ( $class, $hook ) = @_;
     my $read = 0;