From a5cd8e512ac0ab9e0e6b4d12abcc2c876cae7400 Mon Sep 17 00:00:00 2001
From: Lari Taskula <lari.taskula@jns.fi>
Date: Tue, 7 Feb 2017 19:40:38 +0200
Subject: [PATCH] Bug 7534: Add method
 Koha::Template::Plugin::Branches->pickup_locations

This patch adds a new method Koha::Template::Plugin::Branches->pickup_locations.

Unit tests included.

To test:
1. prove t/db_dependent/Template/Plugin/Branches.t
---
 Koha/Template/Plugin/Branches.pm          | 12 ++++++++++--
 t/db_dependent/Template/Plugin/Branches.t | 16 +++++++++++++++-
 2 files changed, 25 insertions(+), 3 deletions(-)

diff --git a/Koha/Template/Plugin/Branches.pm b/Koha/Template/Plugin/Branches.pm
index 3d17edc..75ed8f7 100644
--- a/Koha/Template/Plugin/Branches.pm
+++ b/Koha/Template/Plugin/Branches.pm
@@ -59,10 +59,11 @@ sub all {
     my ( $self, $params ) = @_;
     my $selected = $params->{selected};
     my $unfiltered = $params->{unfiltered} || 0;
+    my $search_params = $params->{search_params} || {};
 
     my $libraries = $unfiltered
-      ? Koha::Libraries->search( {}, { order_by => ['branchname'] } )->unblessed
-      : Koha::Libraries->search_filtered( {}, { order_by => ['branchname'] } )->unblessed;
+      ? Koha::Libraries->search( $search_params, { order_by => ['branchname'] } )->unblessed
+      : Koha::Libraries->search_filtered( $search_params, { order_by => ['branchname'] } )->unblessed;
 
     for my $l ( @$libraries ) {
         if (       defined $selected and $l->{branchcode} eq $selected
@@ -75,4 +76,11 @@ sub all {
     return $libraries;
 }
 
+sub pickup_locations {
+    my ($self, $params) = @_;
+    $params->{search_params} ||= {};
+    $params->{search_params}->{pickup_location} = 1;
+    return $self->all($params);
+}
+
 1;
diff --git a/t/db_dependent/Template/Plugin/Branches.t b/t/db_dependent/Template/Plugin/Branches.t
index 507cea9..00bb768 100644
--- a/t/db_dependent/Template/Plugin/Branches.t
+++ b/t/db_dependent/Template/Plugin/Branches.t
@@ -16,7 +16,7 @@
 
 use Modern::Perl;
 
-use Test::More tests => 15;
+use Test::More tests => 17;
 
 use C4::Context;
 use Koha::Database;
@@ -75,6 +75,20 @@ is( grep ( { $_->{branchcode} eq 'ANOTHERLIB' and $_->{selected} == 1 } @$librar
 $libraries = $plugin->all( { selected => '' } );
 is( grep ( { exists $_->{selected} } @$libraries ), 0, 'With selected parameter set to an empty string, no library should be preselected' );
 
+my $total = @{$plugin->all};
+my $pickupable = @{$plugin->pickup_locations};
+my $yet_another_library = $builder->build({
+    source => 'Branch',
+    value => {
+        branchcode => 'CANTPICKUP',
+        pickup_location => 0,
+    }
+});
+is(@{$plugin->pickup_locations}, $pickupable, 'Adding a new library with pickups'
+   .' disabled does not increase the amount returned by ->pickup_locations');
+is(@{$plugin->all}, $total+1, 'However, adding a new library increases'
+   .' the total amount gotten with ->all');
+
 t::lib::Mocks::mock_preference( 'IndependentBranches', 1 );
 $libraries = $plugin->all();
 is( scalar(@$libraries), 1, 'If IndependentBranches is set, only 1 library should be returned' );
-- 
1.9.1