@@ -, +, @@ $ kshell k$ prove t/db_dependent/Koha/ArticleRequests.t --- Koha/ArticleRequests.pm | 45 ++++++++- t/db_dependent/Koha/ArticleRequests.t | 134 ++++++++++++++++++++++++++ 2 files changed, 178 insertions(+), 1 deletion(-) create mode 100755 t/db_dependent/Koha/ArticleRequests.t --- a/Koha/ArticleRequests.pm +++ a/Koha/ArticleRequests.pm @@ -33,7 +33,7 @@ Koha::ArticleRequests - Koha ArticleRequests Object class =head1 API -=head2 Class Methods +=head2 Class methods =cut @@ -60,6 +60,49 @@ sub search_limited { return $self->search( $params, $attributes ); } +=head3 filter_by_current + + my $current_article_requests = $article_requests->filter_by_current; + +Returns a new resultset, filtering out finished article requests. + +=cut + +sub filter_by_current { + my ($self) = @_; + + return $self->search( + { + status => [ + Koha::ArticleRequest::Status::Requested, + Koha::ArticleRequest::Status::Pending, + Koha::ArticleRequest::Status::Processing, + ] + } + ); +} + +=head3 filter_by_finished + + my $finished_article_requests = $article_requests->filter_by_finished; + +Returns a new resultset, filtering out current article requests. + +=cut + +sub filter_by_finished { + my ($self) = @_; + + return $self->search( + { + status => [ + Koha::ArticleRequest::Status::Completed, + Koha::ArticleRequest::Status::Canceled, + ] + } + ); +} + =head3 requested =cut --- a/t/db_dependent/Koha/ArticleRequests.t +++ a/t/db_dependent/Koha/ArticleRequests.t @@ -0,0 +1,134 @@ +#!/usr/bin/perl + +# This file is part of Koha +# +# Koha is free software; you can redistribute it and/or modify it +# under the terms of the GNU General Public License as published by +# the Free Software Foundation; either version 3 of the License, or +# (at your option) any later version. +# +# Koha is distributed in the hope that it will be useful, but +# WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with Koha; if not, see . + +use Modern::Perl; + +use Test::More tests => 2; + +use Koha::ArticleRequest::Status; +use Koha::ArticleRequests; + +use t::lib::TestBuilder; + +my $schema = Koha::Database->new->schema; +my $builder = t::lib::TestBuilder->new; + +subtest 'filter_by_current() tests' => sub { + + plan tests => 1; + + $schema->storage->txn_begin; + + my $ar_requested = $builder->build_object( + { + class => 'Koha::ArticleRequests', + value => { status => Koha::ArticleRequest::Status::Requested } + } + ); + my $ar_pending = $builder->build_object( + { + class => 'Koha::ArticleRequests', + value => { status => Koha::ArticleRequest::Status::Pending } + } + ); + my $ar_processing = $builder->build_object( + { + class => 'Koha::ArticleRequests', + value => { status => Koha::ArticleRequest::Status::Processing } + } + ); + my $ar_completed = $builder->build_object( + { + class => 'Koha::ArticleRequests', + value => { status => Koha::ArticleRequest::Status::Completed } + } + ); + my $ar_cancelled = $builder->build_object( + { + class => 'Koha::ArticleRequests', + value => { status => Koha::ArticleRequest::Status::Canceled } + } + ); + + my $article_requests = Koha::ArticleRequests->search( + { + id => [ + $ar_requested->id, $ar_pending->id, $ar_processing->id, + $ar_completed->id, $ar_cancelled->id + ] + } + ); + + my $current_article_requests = $article_requests->filter_by_current; + + is( $current_article_requests->count, 3, 'Count is correct' ); + + $schema->storage->txn_rollback; +}; + +subtest 'filter_by_current() tests' => sub { + + plan tests => 1; + + $schema->storage->txn_begin; + + my $ar_requested = $builder->build_object( + { + class => 'Koha::ArticleRequests', + value => { status => Koha::ArticleRequest::Status::Requested } + } + ); + my $ar_pending = $builder->build_object( + { + class => 'Koha::ArticleRequests', + value => { status => Koha::ArticleRequest::Status::Pending } + } + ); + my $ar_processing = $builder->build_object( + { + class => 'Koha::ArticleRequests', + value => { status => Koha::ArticleRequest::Status::Processing } + } + ); + my $ar_completed = $builder->build_object( + { + class => 'Koha::ArticleRequests', + value => { status => Koha::ArticleRequest::Status::Completed } + } + ); + my $ar_cancelled = $builder->build_object( + { + class => 'Koha::ArticleRequests', + value => { status => Koha::ArticleRequest::Status::Canceled } + } + ); + + my $article_requests = Koha::ArticleRequests->search( + { + id => [ + $ar_requested->id, $ar_pending->id, $ar_processing->id, + $ar_completed->id, $ar_cancelled->id + ] + } + ); + + my $finished_article_requests = $article_requests->filter_by_finished; + + is( $finished_article_requests->count, 2, 'Count is correct' ); + + $schema->storage->txn_rollback; +}; --