Bug 18014: Add test to AuthoritiesMarc.t to expose problem in AddAuthority
[srvgit] / Koha / Objects.pm
index 6b3100a..9231206 100644 (file)
@@ -23,8 +23,6 @@ use Carp;
 
 use Koha::Database;
 
-our $type;
-
 =head1 NAME
 
 Koha::Objects - Koha Object set base class
@@ -80,15 +78,35 @@ my $object = Koha::Objects->find( { keypart1 => $keypart1, keypart2 => $keypart2
 sub find {
     my ( $self, $id ) = @_;
 
-    return unless $id;
+    return unless defined($id);
 
     my $result = $self->_resultset()->find($id);
 
+    return unless $result;
+
     my $object = $self->object_class()->_new_from_dbic( $result );
 
     return $object;
 }
 
+=head3 Koha::Objects->find_or_create();
+
+my $object = Koha::Objects->find_or_create( $attrs );
+
+=cut
+
+sub find_or_create {
+    my ( $self, $params ) = @_;
+
+    my $result = $self->_resultset->find_or_create($params);
+
+    return unless $result;
+
+    my $object = $self->object_class->_new_from_dbic($result);
+
+    return $object;
+}
+
 =head3 Koha::Objects->search();
 
 my @objects = Koha::Objects->search($params);
@@ -96,32 +114,75 @@ my @objects = Koha::Objects->search($params);
 =cut
 
 sub search {
-    my ( $self, $params ) = @_;
+    my ( $self, $params, $attributes ) = @_;
 
     if (wantarray) {
-        my @dbic_rows = $self->_resultset()->search($params);
+        my @dbic_rows = $self->_resultset()->search($params, $attributes);
 
         return $self->_wrap(@dbic_rows);
 
     }
     else {
         my $class = ref($self) ? ref($self) : $self;
-        my $rs = $self->_resultset()->search($params);
+        my $rs = $self->_resultset()->search($params, $attributes);
 
         return $class->_new_from_dbic($rs);
     }
 }
 
-=head3 Koha::Objects->count();
+=head3 search_related
 
-my @objects = Koha::Objects->count($params);
+    my @objects = Koha::Objects->search_related( $rel_name, $cond?, \%attrs? );
+    my $objects = Koha::Objects->search_related( $rel_name, $cond?, \%attrs? );
+
+Searches the specified relationship, optionally specifying a condition and attributes for matching records.
 
 =cut
 
-sub count {
-    my ( $self, $params ) = @_;
+sub search_related {
+    my ( $self, $rel_name, @params ) = @_;
+
+    return if !$rel_name;
+    if (wantarray) {
+        my @dbic_rows = $self->_resultset()->search_related($rel_name, @params);
+        return if !@dbic_rows;
+        my $object_class = _get_objects_class( $dbic_rows[0]->result_class );
 
-    return $self->_resultset()->count($params);
+        eval "require $object_class";
+        return _wrap( $object_class, @dbic_rows );
+
+    } else {
+        my $rs = $self->_resultset()->search_related($rel_name, @params);
+        return if !$rs;
+        my $object_class = _get_objects_class( $rs->result_class );
+
+        eval "require $object_class";
+        return _new_from_dbic( $object_class, $rs );
+    }
+}
+
+=head3 single
+
+my $object = Koha::Objects->search({}, { rows => 1 })->single
+
+Returns one and only one object that is part of this set.
+Returns undef if there are no objects found.
+
+This is optimal as it will grab the first returned result without instantiating
+a cursor.
+
+See:
+http://search.cpan.org/dist/DBIx-Class/lib/DBIx/Class/Manual/Cookbook.pod#Retrieve_one_and_only_one_row_from_a_resultset
+
+=cut
+
+sub single {
+    my ($self) = @_;
+
+    my $single = $self->_resultset()->single;
+    return unless $single;
+
+    return $self->object_class()->_new_from_dbic($single);
 }
 
 =head3 Koha::Objects->next();
@@ -179,6 +240,18 @@ sub as_list {
     return wantarray ? @objects : \@objects;
 }
 
+=head3 Koha::Objects->unblessed
+
+Returns an unblessed representation of objects.
+
+=cut
+
+sub unblessed {
+    my ($self) = @_;
+
+    return [ map { $_->unblessed } $self->as_list ];
+}
+
 =head3 Koha::Objects->_wrap
 
 wraps the DBIC object in a corresponding Koha object
@@ -188,7 +261,7 @@ wraps the DBIC object in a corresponding Koha object
 sub _wrap {
     my ( $self, @dbic_rows ) = @_;
 
-    my @objects = map { $self->object_class()->_new_from_dbic( $_ ) } @dbic_rows;
+    my @objects = map { $self->object_class->_new_from_dbic( $_ ) } @dbic_rows;
 
     return @objects;
 }
@@ -204,24 +277,74 @@ sub _resultset {
 
     if ( ref($self) ) {
         $self->{_resultset} ||=
-          Koha::Database->new()->schema()->resultset( $self->type() );
+          Koha::Database->new()->schema()->resultset( $self->_type() );
 
         return $self->{_resultset};
     }
     else {
-        return Koha::Database->new()->schema()->resultset( $self->type() );
+        return Koha::Database->new()->schema()->resultset( $self->_type() );
+    }
+}
+
+sub _get_objects_class {
+    my ( $type ) = @_;
+    return unless $type;
+
+    if( $type->can('koha_objects_class') ) {
+        return $type->koha_objects_class;
+    }
+    $type =~ s|Schema::Result::||;
+    return "${type}s";
+}
+
+=head3 columns
+
+my @columns = Koha::Objects->columns
+
+Return the table columns
+
+=cut
+
+sub columns {
+    my ( $class ) = @_;
+    return Koha::Database->new->schema->resultset( $class->_type )->result_source->columns;
+}
+
+=head3 AUTOLOAD
+
+The autoload method is used call DBIx::Class method on a resultset.
+
+Important: If you plan to use one of the DBIx::Class methods you must provide
+relevant tests in t/db_dependent/Koha/Objects.t
+Currently count, pager, update and delete are covered.
+
+=cut
+
+sub AUTOLOAD {
+    my ( $self, @params ) = @_;
+
+    my @known_methods = qw( count pager update delete result_class single );
+    my $method = our $AUTOLOAD;
+    $method =~ s/.*:://;
+
+    carp "The method $method is not covered by tests" and return unless grep {/^$method$/} @known_methods;
+    my $r = eval { $self->_resultset->$method(@params) };
+    if ( $@ ) {
+        carp "No method $method found for " . ref($self) . " " . $@;
+        return
     }
+    return $r;
 }
 
-=head3 type
+=head3 _type
 
-The type method must be set for all child classes.
+The _type method must be set for all child classes.
 The value returned by it should be the DBIC resultset name.
-For example, for holds, type should return 'Reserve'.
+For example, for holds, _type should return 'Reserve'.
 
 =cut
 
-sub type { }
+sub _type { }
 
 =head3 object_class