Bug 8015: (follow-up) move from the Koha to the C4 namespace

[koha.git] / t / ItemType.t

diff --git a/t/ItemType.t b/t/ItemType.t
index 476ca19..558bba9 100755 (executable)
--- a/t/ItemType.t
+++ b/t/ItemType.t
@@ -1,12 +1,9 @@
 #!/usr/bin/perl
-#
-# This Koha test module is a stub!
-# Add more tests here!!!
 
 use strict;
 use warnings;
 use DBI;
-use Test::More tests => 15;
+use Test::More tests => 26;
 use Test::MockModule;
 
 BEGIN {
@@ -27,13 +24,21 @@ $module->mock(
 my $itemtypes = [
     [
         'itemtype', 'description', 'rentalcharge', 'notforloan',
-        'imageurl', 'summary'
+        'imageurl', 'summary', 'checkinmsg'
+    ],
+    [ 'BK', 'Books', 0, 0, '', '', 'foo' ],
+    [ 'CD', 'CDRom', 0, 0, '', '', 'bar' ]
+];
+
+my $itemtypes_empty = [
+    [
+        'itemtype', 'description', 'rentalcharge', 'notforloan',
+        'imageurl', 'summary', 'checkinmsg'
     ],
-    [ 'BK', 'Books', 0, 0, '', '' ],
-    [ 'CD', 'CDRom', 0, 0, '', '' ]
 ];
 
 my $dbh = C4::Context->dbh();
+$dbh->{mock_add_resultset} = $itemtypes_empty;
 
 my @itemtypes = C4::ItemType->all();
 is( @itemtypes, 0, 'Testing all itemtypes is empty' );
@@ -46,6 +51,10 @@ is( scalar( @{$history} ), 1, 'Correct number of statements executed' );
 $dbh->{mock_add_resultset} = $itemtypes;
 
 @itemtypes = C4::ItemType->all();
+
+$history = $dbh->{mock_all_history};
+is( scalar( @{$history} ), 2, 'Correct number of statements executed' );
+
 is( @itemtypes, 2, 'ItemType->all should return an array with 2 elements' );
 
 is( $itemtypes[0]->fish, undef, 'Calling a bad descriptor gives undef' );
@@ -66,6 +75,33 @@ is( $itemtypes[0]->notforloan, '0', 'first not for loan is 0' );
 
 is( $itemtypes[1]->notforloan, '0', 'second not for loan is 0' );
 
-is( $itemtypes[0]->imageurl, '', 'first not for loan is undef' );
+is( $itemtypes[0]->imageurl, '', 'first imageurl is undef' );
+
+is( $itemtypes[1]->imageurl, '', 'second imageurl is undef' );
+
+is( $itemtypes[0]->checkinmsg, 'foo', 'first checkinmsg is foo' );
+
+is( $itemtypes[1]->checkinmsg, 'bar', 'second checkinmsg is bar' );
+
+# Mock the data again
+$dbh->{mock_add_resultset} = $itemtypes;
+
+# Test get(), which should return one itemtype
+my $itemtype = C4::ItemType->get( 'BK' );
+
+$history = $dbh->{mock_all_history};
+is( scalar( @{$history} ), 3, 'Correct number of statements executed' );
+
+is( $itemtype->fish, undef, 'Calling a bad descriptor gives undef' );
+
+is( $itemtype->itemtype, 'BK', 'itemtype is bk' );
+
+is( $itemtype->description, 'Books', 'description is books' );
+
+is( $itemtype->rentalcharge, '0', 'rental charge is 0' );
+
+is( $itemtype->notforloan, '0', 'not for loan is 0' );
+
+is( $itemtype->imageurl, '', ' not for loan is undef' );
 
-is( $itemtypes[1]->imageurl, '', 'second not for loan is undef' );
+is( $itemtype->checkinmsg, 'foo', 'checkinmsg is foo' );