Bug 10403: (follow-up) fix test to use vendor created earlier during test

[koha.git] / t / Members_AttributeTypes.t

diff --git a/t/Members_AttributeTypes.t b/t/Members_AttributeTypes.t
index 9dfc6c3..152fc53 100755 (executable)
--- a/t/Members_AttributeTypes.t
+++ b/t/Members_AttributeTypes.t
@@ -1,14 +1,67 @@
 #!/usr/bin/perl
 #
-# This Koha test module is a stub!  
-# Add more tests here!!!
+# Tests 'fetch', 'fake db data', and 'checks for existant attributes'
 
 use strict;
 use warnings;
-
-use Test::More tests => 1;
+use Test::MockModule;
+use Test::More tests => 9;
 
 BEGIN {
-        use_ok('C4::Members::AttributeTypes');
+    use_ok('C4::Members::AttributeTypes');
 }
 
+my $module = new Test::MockModule('C4::Context');
+$module->mock(
+    '_new_dbh',
+    sub {
+        my $dbh = DBI->connect( 'DBI:Mock:', '', '' )
+          || die "Cannot create handle: $DBI::errstr\n";
+        return $dbh;
+    }
+);
+my $members_attributetypes = [
+    [
+        'code',             'description',
+        'repeatable',       'unique_id',
+        'opac_display',     'password_allowed',
+        'staff_searchable', 'authorised_value_category',
+        'display_checkout', 'catagory_code',
+        'class'
+    ],
+    [ 'one', 'ISBN', '1', '1', '1', '1', '1', 'red',  '1', 'orange', 'green' ],
+    [ 'two', 'ISSN', '0', '0', '0', '0', '0', 'blue', '0', 'yellow', 'silver' ]
+];
+
+my $dbh = C4::Context->dbh();
+
+$dbh->{mock_add_resultset} = $members_attributetypes;
+
+my @members_attributetypes = C4::Members::AttributeTypes::GetAttributeTypes(undef, 1);
+
+is( $members_attributetypes[0]->{'code'}, 'one', 'First code value is one' );
+
+is( $members_attributetypes[1]->{'code'}, 'two', 'Second code value is two' );
+
+is( $members_attributetypes[0]->{'class'},
+    'green', 'First class value is green' );
+
+is( $members_attributetypes[1]->{'class'},
+    'silver', 'Second class value is silver' );
+
+$dbh->{mock_add_resultset} = $members_attributetypes;
+
+ok( C4::Members::AttributeTypes::AttributeTypeExists('one'),
+    'checking an attribute type exists' );
+
+ok(
+    !C4::Members::AttributeTypes::AttributeTypeExists('three'),
+    "checking a attribute that isn't in the code doesn't exist"
+);
+
+$dbh->{mock_add_resultset} = $members_attributetypes;
+
+ok( C4::Members::AttributeTypes->fetch('ISBN'), "testing fetch feature" );
+
+ok( !C4::Members::AttributeTypes->fetch('FAKE'),
+    "testing fetch feature doesn't work if value not in database" );