Bug 15030: Add tests
[koha.git] / Koha / Objects.pm
blob92312066f83c5b516afa70571669d597c7e9d379
1 package Koha::Objects;
3 # Copyright ByWater Solutions 2014
5 # This file is part of Koha.
7 # Koha is free software; you can redistribute it and/or modify it under the
8 # terms of the GNU General Public License as published by the Free Software
9 # Foundation; either version 3 of the License, or (at your option) any later
10 # version.
12 # Koha is distributed in the hope that it will be useful, but WITHOUT ANY
13 # WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
14 # A PARTICULAR PURPOSE. See the GNU General Public License for more details.
16 # You should have received a copy of the GNU General Public License along
17 # with Koha; if not, write to the Free Software Foundation, Inc.,
18 # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
20 use Modern::Perl;
22 use Carp;
24 use Koha::Database;
26 =head1 NAME
28 Koha::Objects - Koha Object set base class
30 =head1 SYNOPSIS
32 use Koha::Objects;
33 my @objects = Koha::Objects->search({ borrowernumber => $borrowernumber});
35 =head1 DESCRIPTION
37 This class must be subclassed.
39 =head1 API
41 =head2 Class Methods
43 =cut
45 =head3 Koha::Objects->new();
47 my $object = Koha::Objects->new();
49 =cut
51 sub new {
52 my ($class) = @_;
53 my $self = {};
55 bless( $self, $class );
58 =head3 Koha::Objects->_new_from_dbic();
60 my $object = Koha::Objects->_new_from_dbic( $resultset );
62 =cut
64 sub _new_from_dbic {
65 my ( $class, $resultset ) = @_;
66 my $self = { _resultset => $resultset };
68 bless( $self, $class );
71 =head3 Koha::Objects->find();
73 my $object = Koha::Objects->find($id);
74 my $object = Koha::Objects->find( { keypart1 => $keypart1, keypart2 => $keypart2 } );
76 =cut
78 sub find {
79 my ( $self, $id ) = @_;
81 return unless defined($id);
83 my $result = $self->_resultset()->find($id);
85 return unless $result;
87 my $object = $self->object_class()->_new_from_dbic( $result );
89 return $object;
92 =head3 Koha::Objects->find_or_create();
94 my $object = Koha::Objects->find_or_create( $attrs );
96 =cut
98 sub find_or_create {
99 my ( $self, $params ) = @_;
101 my $result = $self->_resultset->find_or_create($params);
103 return unless $result;
105 my $object = $self->object_class->_new_from_dbic($result);
107 return $object;
110 =head3 Koha::Objects->search();
112 my @objects = Koha::Objects->search($params);
114 =cut
116 sub search {
117 my ( $self, $params, $attributes ) = @_;
119 if (wantarray) {
120 my @dbic_rows = $self->_resultset()->search($params, $attributes);
122 return $self->_wrap(@dbic_rows);
125 else {
126 my $class = ref($self) ? ref($self) : $self;
127 my $rs = $self->_resultset()->search($params, $attributes);
129 return $class->_new_from_dbic($rs);
133 =head3 search_related
135 my @objects = Koha::Objects->search_related( $rel_name, $cond?, \%attrs? );
136 my $objects = Koha::Objects->search_related( $rel_name, $cond?, \%attrs? );
138 Searches the specified relationship, optionally specifying a condition and attributes for matching records.
140 =cut
142 sub search_related {
143 my ( $self, $rel_name, @params ) = @_;
145 return if !$rel_name;
146 if (wantarray) {
147 my @dbic_rows = $self->_resultset()->search_related($rel_name, @params);
148 return if !@dbic_rows;
149 my $object_class = _get_objects_class( $dbic_rows[0]->result_class );
151 eval "require $object_class";
152 return _wrap( $object_class, @dbic_rows );
154 } else {
155 my $rs = $self->_resultset()->search_related($rel_name, @params);
156 return if !$rs;
157 my $object_class = _get_objects_class( $rs->result_class );
159 eval "require $object_class";
160 return _new_from_dbic( $object_class, $rs );
164 =head3 single
166 my $object = Koha::Objects->search({}, { rows => 1 })->single
168 Returns one and only one object that is part of this set.
169 Returns undef if there are no objects found.
171 This is optimal as it will grab the first returned result without instantiating
172 a cursor.
174 See:
175 http://search.cpan.org/dist/DBIx-Class/lib/DBIx/Class/Manual/Cookbook.pod#Retrieve_one_and_only_one_row_from_a_resultset
177 =cut
179 sub single {
180 my ($self) = @_;
182 my $single = $self->_resultset()->single;
183 return unless $single;
185 return $self->object_class()->_new_from_dbic($single);
188 =head3 Koha::Objects->next();
190 my $object = Koha::Objects->next();
192 Returns the next object that is part of this set.
193 Returns undef if there are no more objects to return.
195 =cut
197 sub next {
198 my ( $self ) = @_;
200 my $result = $self->_resultset()->next();
201 return unless $result;
203 my $object = $self->object_class()->_new_from_dbic( $result );
205 return $object;
208 =head3 Koha::Objects->reset();
210 Koha::Objects->reset();
212 resets iteration so the next call to next() will start agein
213 with the first object in a set.
215 =cut
217 sub reset {
218 my ( $self ) = @_;
220 $self->_resultset()->reset();
222 return $self;
225 =head3 Koha::Objects->as_list();
227 Koha::Objects->as_list();
229 Returns an arrayref of the objects in this set.
231 =cut
233 sub as_list {
234 my ( $self ) = @_;
236 my @dbic_rows = $self->_resultset()->all();
238 my @objects = $self->_wrap(@dbic_rows);
240 return wantarray ? @objects : \@objects;
243 =head3 Koha::Objects->unblessed
245 Returns an unblessed representation of objects.
247 =cut
249 sub unblessed {
250 my ($self) = @_;
252 return [ map { $_->unblessed } $self->as_list ];
255 =head3 Koha::Objects->_wrap
257 wraps the DBIC object in a corresponding Koha object
259 =cut
261 sub _wrap {
262 my ( $self, @dbic_rows ) = @_;
264 my @objects = map { $self->object_class->_new_from_dbic( $_ ) } @dbic_rows;
266 return @objects;
269 =head3 Koha::Objects->_resultset
271 Returns the internal resultset or creates it if undefined
273 =cut
275 sub _resultset {
276 my ($self) = @_;
278 if ( ref($self) ) {
279 $self->{_resultset} ||=
280 Koha::Database->new()->schema()->resultset( $self->_type() );
282 return $self->{_resultset};
284 else {
285 return Koha::Database->new()->schema()->resultset( $self->_type() );
289 sub _get_objects_class {
290 my ( $type ) = @_;
291 return unless $type;
293 if( $type->can('koha_objects_class') ) {
294 return $type->koha_objects_class;
296 $type =~ s|Schema::Result::||;
297 return "${type}s";
300 =head3 columns
302 my @columns = Koha::Objects->columns
304 Return the table columns
306 =cut
308 sub columns {
309 my ( $class ) = @_;
310 return Koha::Database->new->schema->resultset( $class->_type )->result_source->columns;
313 =head3 AUTOLOAD
315 The autoload method is used call DBIx::Class method on a resultset.
317 Important: If you plan to use one of the DBIx::Class methods you must provide
318 relevant tests in t/db_dependent/Koha/Objects.t
319 Currently count, pager, update and delete are covered.
321 =cut
323 sub AUTOLOAD {
324 my ( $self, @params ) = @_;
326 my @known_methods = qw( count pager update delete result_class single );
327 my $method = our $AUTOLOAD;
328 $method =~ s/.*:://;
330 carp "The method $method is not covered by tests" and return unless grep {/^$method$/} @known_methods;
331 my $r = eval { $self->_resultset->$method(@params) };
332 if ( $@ ) {
333 carp "No method $method found for " . ref($self) . " " . $@;
334 return
336 return $r;
339 =head3 _type
341 The _type method must be set for all child classes.
342 The value returned by it should be the DBIC resultset name.
343 For example, for holds, _type should return 'Reserve'.
345 =cut
347 sub _type { }
349 =head3 object_class
351 This method must be set for all child classes.
352 The value returned by it should be the name of the Koha
353 object class that is returned by this class.
354 For example, for holds, object_class should return 'Koha::Hold'.
356 =cut
358 sub object_class { }
360 sub DESTROY { }
362 =head1 AUTHOR
364 Kyle M Hall <kyle@bywatersolutions.com>
366 =cut