koha-dev.rot13.org Git - srvgit/blob - t/lib/Koha/BackgroundJob/BatchTest.pm

   1 package t::lib::Koha::BackgroundJob::BatchTest;
   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 use JSON qw( encode_json decode_json );
  20
  21 use Koha::BackgroundJobs;
  22 use Koha::DateUtils qw( dt_from_string );
  23
  24 use base 'Koha::BackgroundJob';
  25
  26 sub job_type {
  27     return 'batch_test';
  28 }
  29
  30 sub process {
  31     my ( $self, $args ) = @_;
  32
  33     my $job = Koha::BackgroundJobs->find( $args->{job_id} );
  34
  35     if ( !exists $args->{job_id} || !$job || $job->status eq 'cancelled' ) {
  36         return;
  37     }
  38
  39     my $job_progress = 0;
  40     $job->started_on(dt_from_string)
  41         ->progress($job_progress)
  42         ->status('started')
  43         ->store;
  44
  45     my $report = {
  46         total_records => $job->size,
  47         total_success => 0,
  48     };
  49     my @messages;
  50     for my $i ( 0 .. $job->size - 1 ) {
  51
  52         last if $job->get_from_storage->status eq 'cancelled';
  53
  54         push @messages, {
  55             type => 'success',
  56             i => $i,
  57         };
  58         $report->{total_success}++;
  59         $job->progress( ++$job_progress )->store;
  60     }
  61
  62     my $job_data = decode_json $job->data;
  63     $job_data->{messages} = \@messages;
  64     $job_data->{report} = $report;
  65
  66     $job->ended_on(dt_from_string)
  67         ->data(encode_json $job_data);
  68     $job->status('finished') if $job->status ne 'cancelled';
  69     $job->store;
  70 }
  71
  72 sub enqueue {
  73     my ( $self, $args) = @_;
  74
  75     $self->SUPER::enqueue({
  76         job_size => $args->{size},
  77         job_args => {a => $args->{a}, b => $args->{b}}
  78     });
  79 }
  80
  81 1;