From bd87f9b2b2ece43a4cd6e33a55f402ae5576f96f Mon Sep 17 00:00:00 2001
From: Tomas Cohen Arazi <tomascohen@theke.io>
Date: Tue, 3 May 2022 11:18:10 -0300
Subject: [PATCH] Bug 30663: Add Koha::Suggestions helper methods

This patch adds the following helper methods:

* filter_by_pending
* filter_by_suggested_days_range

This methods follow basically what's done in opac-suggestions.pl
I chose 'pending' as opposed to 'open' to follow what we use in the UI
which might be the case because of being more accurate for end users.

To test:
1. Apply this patch
2. Run:
   $ kshell
  k$ prove t/db_dependent/Koha/Suggestions.t
=> SUCCESS: Tests pass!
3. Sign off :-D

Signed-off-by: Tomas Cohen Arazi <tomascohen@theke.io>
---
 Koha/Suggestions.pm               | 32 +++++++++++++++
 t/db_dependent/Koha/Suggestions.t | 65 ++++++++++++++++++++++++++++++-
 2 files changed, 95 insertions(+), 2 deletions(-)

diff --git a/Koha/Suggestions.pm b/Koha/Suggestions.pm
index 5fadfd163b..9b1d266586 100644
--- a/Koha/Suggestions.pm
+++ b/Koha/Suggestions.pm
@@ -21,6 +21,7 @@ use Modern::Perl;
 
 
 use Koha::Database;
+use Koha::DateUtils qw(dt_from_string);
 use Koha::Suggestion;
 
 use base qw(Koha::Objects);
@@ -62,6 +63,37 @@ sub search_limited {
     return $resultset->search( $params, $attributes);
 }
 
+=head3 filter_by_pending
+
+    my $open = $suggestions->filter_by_pending;
+
+Filters the resultset on those that are considered pending (i.e. STATUS = ASKED).
+
+=cut
+
+sub filter_by_pending {
+    my ($self) = @_;
+
+    return $self->search( { STATUS => 'ASKED' } );
+}
+
+=head3 filter_by_suggested_days_range
+
+    my $suggestions = $suggestions->filter_by_suggested_days_range( $days );
+
+Filters the resultset on those placed within some I<$days> range.
+
+=cut
+
+sub filter_by_suggested_days_range {
+    my ( $self, $days ) = @_;
+
+    my $dtf = Koha::Database->new->schema->storage->datetime_parser;
+
+    return $self->search(
+        { suggesteddate => { '>=' => $dtf->format_date( dt_from_string->subtract( days => $days ) ) } } );
+}
+
 =head2 Internal methods
 
 =head3 _type
diff --git a/t/db_dependent/Koha/Suggestions.t b/t/db_dependent/Koha/Suggestions.t
index 8dbb93afd3..ae95ebe274 100755
--- a/t/db_dependent/Koha/Suggestions.t
+++ b/t/db_dependent/Koha/Suggestions.t
@@ -19,10 +19,9 @@
 
 use Modern::Perl;
 
-use Test::More tests => 9;
+use Test::More tests => 11;
 use Test::Exception;
 
-use Koha::Suggestion;
 use Koha::Suggestions;
 use Koha::Database;
 use Koha::DateUtils qw( dt_from_string output_pref );
@@ -329,3 +328,65 @@ subtest 'search_limited() tests' => sub {
 
     $schema->storage->txn_rollback;
 };
+
+subtest 'filter_by_pending() tests' => sub {
+
+    plan tests => 3;
+
+    $schema->storage->txn_begin;
+
+    my $suggestion_1 = $builder->build_object( { class => 'Koha::Suggestions', value => { STATUS => 'ASKED' } } );
+    my $suggestion_2 = $builder->build_object( { class => 'Koha::Suggestions', value => { STATUS => 'ACCEPTED' } } );
+    my $suggestion_3 = $builder->build_object( { class => 'Koha::Suggestions', value => { STATUS => 'ASKED' } } );
+
+    my $suggestions =
+      Koha::Suggestions->search( { suggestionid => [ $suggestion_1->id, $suggestion_2->id, $suggestion_3->id ] },
+        { order_by => ['suggestionid'] } );
+
+    is( $suggestions->count, 3 );
+
+    my $pending     = $suggestions->filter_by_pending;
+    my @pending_ids = $pending->get_column('suggestionid');
+
+    is( $pending->count, 2 );
+    is_deeply( \@pending_ids, [ $suggestion_1->id, $suggestion_3->id ] );
+
+    $schema->storage->txn_rollback;
+};
+
+subtest 'filter_by_suggested_days_range() tests' => sub {
+
+    plan tests => 4;
+
+    $schema->storage->txn_begin;
+
+    my $today             = dt_from_string;
+    my $today_minus_two   = dt_from_string->subtract( days => 2 );
+    my $today_minus_three = dt_from_string->subtract( days => 3 );
+
+    my $dtf = Koha::Database->new->schema->storage->datetime_parser;
+
+    my $suggestion_1 = $builder->build_object(
+        { class => 'Koha::Suggestions', value => { suggesteddate => $dtf->format_date($today) } } );
+    my $suggestion_2 = $builder->build_object(
+        { class => 'Koha::Suggestions', value => { suggesteddate => $dtf->format_date($today_minus_two) } } );
+    my $suggestion_3 = $builder->build_object(
+        { class => 'Koha::Suggestions', value => { suggesteddate => $dtf->format_date($today_minus_three) } } );
+
+    my $suggestions =
+      Koha::Suggestions->search( { suggestionid => [ $suggestion_1->id, $suggestion_2->id, $suggestion_3->id ] },
+        { order_by => ['suggestionid'] } );
+
+    is( $suggestions->count, 3 );
+
+    my $three_days = $suggestions->filter_by_suggested_days_range(3);
+    is( $three_days->count, 3 );
+
+    my $two_days = $suggestions->filter_by_suggested_days_range(2);
+    is( $two_days->count, 2 );
+
+    my $one_days = $suggestions->filter_by_suggested_days_range(1);
+    is( $one_days->count, 1 );
+
+    $schema->storage->txn_rollback;
+};
-- 
2.36.0