t/db_dependent/Koha/BackgroundJob.t

   1 #!/usr/bin/perl
   2
   3 # This file is part of Koha
   4 #
   5 # Koha is free software; you can redistribute it and/or modify it
   6 # under the terms of the GNU General Public License as published by
   7 # the Free Software Foundation; either version 3 of the License, or
   8 # (at your option) any later version.
   9 #
  10 # Koha is distributed in the hope that it will be useful, but
  11 # WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13 # GNU General Public License for more details.
  14 #
  15 # You should have received a copy of the GNU General Public License
  16 # along with Koha; if not, see <http://www.gnu.org/licenses>.
  17
  18 use Modern::Perl;
  19
  20 use Test::More tests => 2;
  21
  22 use Koha::Database;
  23 use Koha::BackgroundJobs;
  24 use Koha::BackgroundJob::BatchUpdateItem;
  25
  26 use t::lib::Mocks;
  27 use t::lib::TestBuilder;
  28
  29 my $schema  = Koha::Database->new->schema;
  30 my $builder = t::lib::TestBuilder->new;
  31
  32 subtest '_derived_class() tests' => sub {
  33
  34     plan tests => 3;
  35
  36     $schema->storage->txn_begin;
  37
  38     my $mapping = Koha::BackgroundJob::type_to_class_mapping;
  39
  40     # pick the first
  41     my $type = ( keys %{$mapping} )[0];
  42
  43     my $job = $builder->build_object(
  44         {   class => 'Koha::BackgroundJobs',
  45             value => { type => $type, data => 'Foo' }
  46         }
  47     );
  48
  49     my $derived = $job->_derived_class;
  50
  51     is( ref($derived), $mapping->{$type}, 'Job object class is correct' );
  52     ok( $derived->in_storage, 'The object is correctly marked as in storage' );
  53
  54     $derived->data('Bar')->store->discard_changes;
  55     $job->discard_changes;
  56
  57     is_deeply( $job->unblessed, $derived->unblessed, '_derived_class object refers to the same DB object and can be manipulated as expected' );
  58
  59     $schema->storage->txn_rollback;
  60 };
  61
  62 subtest 'enqueue() tests' => sub {
  63
  64     plan tests => 6;
  65
  66     $schema->storage->txn_begin;
  67
  68     # FIXME: This all feels we need to do it better...
  69     my $job_id = Koha::BackgroundJob::BatchUpdateItem->new->enqueue( { record_ids => [ 1, 2 ] } );
  70     my $job    = Koha::BackgroundJobs->find($job_id)->_derived_class;
  71
  72     is( $job->size,           2,     'Two steps' );
  73     is( $job->status,         'new', 'Initial status set correctly' );
  74     is( $job->borrowernumber, undef, 'No userenv, borrowernumber undef' );
  75
  76     my $patron = $builder->build_object( { class => 'Koha::Patrons' } );
  77     t::lib::Mocks::mock_userenv( { patron => $patron } );
  78
  79     $job_id = Koha::BackgroundJob::BatchUpdateItem->new->enqueue( { record_ids => [ 1, 2, 3 ] } );
  80     $job    = Koha::BackgroundJobs->find($job_id)->_derived_class;
  81
  82     is( $job->size,           3,           'Three steps' );
  83     is( $job->status,         'new',       'Initial status set correctly' );
  84     is( $job->borrowernumber, $patron->id, 'No userenv, borrowernumber undef' );
  85
  86     $schema->storage->txn_rollback;
  87 };