t/db_dependent/Virtualshelves.t

   1 #!/usr/bin/perl
   2
   3 use Modern::Perl;
   4 use Test::More tests => 6;
   5 use DateTime::Duration;
   6
   7 use C4::Context;
   8 use Koha::DateUtils;
   9 use Koha::Virtualshelves;
  10 use Koha::Virtualshelfshares;
  11 use Koha::Virtualshelfcontents;
  12
  13 use t::lib::TestBuilder;
  14
  15 my $builder = t::lib::TestBuilder->new;
  16
  17 my $dbh = C4::Context->dbh;
  18 $dbh->{AutoCommit} = 0;
  19 teardown();
  20
  21 subtest 'CRUD' => sub {
  22     plan tests => 13;
  23     my $patron = $builder->build({
  24         source => 'Borrower',
  25     });
  26
  27     my $number_of_shelves = Koha::Virtualshelves->search->count;
  28
  29     is( $number_of_shelves, 0, 'No shelves should exist' );
  30
  31     my $shelf = Koha::Virtualshelf->new({
  32             shelfname => "my first shelf",
  33             owner => $patron->{borrowernumber},
  34             category => 1,
  35         }
  36     )->store;
  37
  38     is( ref( $shelf ), 'Koha::Virtualshelf', 'The constructor should return a valid object' );
  39
  40     $number_of_shelves = Koha::Virtualshelves->search->count;
  41     is( $number_of_shelves, 1, '1 shelf should have been inserted' );
  42     is( $shelf->allow_add, 0, 'The default value for allow_add should be 1' );
  43     is( $shelf->allow_delete_own, 1, 'The default value for allow_delete_own should be 0' );
  44     is( $shelf->allow_delete_other, 0, 'The default value for allow_delete_other should be 0' );
  45     is( output_pref($shelf->created_on), output_pref(dt_from_string), 'The creation time should have been set to today' );
  46
  47     my $retrieved_shelf = Koha::Virtualshelves->find( $shelf->shelfnumber );
  48
  49     is( $retrieved_shelf->shelfname, $shelf->shelfname, 'Find should correctly return the shelfname' );
  50
  51     # Insert with the same name
  52     eval {
  53         $shelf = Koha::Virtualshelf->new({
  54                 shelfname => "my first shelf",
  55                 owner => $patron->{borrowernumber},
  56                 category => 1,
  57             }
  58         )->store;
  59     };
  60     is( ref($@), 'Koha::Exceptions::Virtualshelves::DuplicateObject',
  61         'Exception on duplicate name' );
  62     $number_of_shelves = Koha::Virtualshelves->search->count;
  63     is( $number_of_shelves, 1, 'To be sure the number of shelves is still 1' );
  64
  65     my $another_patron = $builder->build({
  66         source => 'Borrower',
  67     });
  68
  69     $shelf = Koha::Virtualshelf->new({
  70             shelfname => "my first shelf",
  71             owner => $another_patron->{borrowernumber},
  72             category => 1,
  73         }
  74     )->store;
  75     $number_of_shelves = Koha::Virtualshelves->search->count;
  76     is( $number_of_shelves, 2, 'Another patron should be able to create a shelf with an existing shelfname');
  77
  78     my $is_deleted = Koha::Virtualshelves->find( $shelf->shelfnumber )->delete;
  79     is( $is_deleted, 1, 'The shelf has been deleted correctly' );
  80     $number_of_shelves = Koha::Virtualshelves->search->count;
  81     is( $number_of_shelves, 1, 'To be sure the shelf has been deleted' );
  82
  83     teardown();
  84 };
  85
  86 subtest 'Sharing' => sub {
  87     plan tests => 18;
  88     my $patron_wants_to_share = $builder->build({
  89         source => 'Borrower',
  90     });
  91     my $share_with_me = $builder->build({
  92         source => 'Borrower',
  93     });
  94     my $just_another_patron = $builder->build({
  95         source => 'Borrower',
  96     });
  97
  98     my $number_of_shelves_shared = Koha::Virtualshelfshares->search->count;
  99     is( $number_of_shelves_shared, 0, 'No shelves should exist' );
 100
 101     my $shelf_to_share = Koha::Virtualshelf->new({
 102             shelfname => "my first shelf",
 103             owner => $patron_wants_to_share->{borrowernumber},
 104             category => 1,
 105         }
 106     )->store;
 107
 108     my $shelf_not_to_share = Koha::Virtualshelf->new({
 109             shelfname => "my second shelf",
 110             owner => $patron_wants_to_share->{borrowernumber},
 111             category => 1,
 112         }
 113     )->store;
 114
 115     my $shared_shelf = eval { $shelf_to_share->share };
 116     is ( ref( $@ ), 'Koha::Exceptions::Virtualshelves::InvalidKeyOnSharing', 'Do not share if no key given' );
 117     $shared_shelf = eval { $shelf_to_share->share('this is a valid key') };
 118     is( ref( $shared_shelf ), 'Koha::Virtualshelfshare', 'On sharing, the method should return a valid Koha::Virtualshelfshare object' );
 119
 120     my $another_shared_shelf = eval { $shelf_to_share->share('this is another valid key') }; # Just to have 2 shares in DB
 121
 122     $number_of_shelves_shared = Koha::Virtualshelfshares->search->count;
 123     is( $number_of_shelves_shared, 2, '2 shares should have been inserted' );
 124
 125     my $is_accepted = eval {
 126         $shared_shelf->accept( 'this is an invalid key', $share_with_me->{borrowernumber} );
 127     };
 128     is( $is_accepted, undef, 'The share should have not been accepted if the key is invalid' );
 129     is( ref( $@ ), 'Koha::Exceptions::Virtualshelves::InvalidInviteKey', 'accept with an invalid key should raise an exception' );
 130
 131     $is_accepted = $shared_shelf->accept( 'this is a valid key', $share_with_me->{borrowernumber} );
 132     ok( defined($is_accepted), 'The share should have been accepted if the key valid' );
 133
 134     is( $shelf_to_share->is_shared, 1, 'first shelf is shared' );
 135     is( $shelf_not_to_share->is_shared, 0, 'second shelf is not shared' );
 136
 137     is( $shelf_to_share->is_shared_with( $patron_wants_to_share->{borrowernumber} ), 0 , "The shelf should not be shared with the owner" );
 138     is( $shelf_to_share->is_shared_with( $share_with_me->{borrowernumber} ), 1 , "The shelf should be shared with share_with_me" );
 139     is( $shelf_to_share->is_shared_with( $just_another_patron->{borrowernumber} ), 0, "The shelf should not be shared with just_another_patron" );
 140
 141     is( $shelf_to_share->remove_share( $just_another_patron->{borrowernumber} ), 0, 'No share should be removed if the share has not been done with this patron' );
 142     $number_of_shelves_shared = Koha::Virtualshelfshares->search->count;
 143     is( $number_of_shelves_shared, 2, 'To be sure no shares have been removed' );
 144
 145     is( $shelf_not_to_share->remove_share( $share_with_me->{borrowernumber} ), 0, '0 share should have been removed if the shelf is not share' );
 146     $number_of_shelves_shared = Koha::Virtualshelfshares->search->count;
 147     is( $number_of_shelves_shared, 2, 'To be sure no shares have been removed' );
 148
 149     is( $shelf_to_share->remove_share( $share_with_me->{borrowernumber} ), 1, '1 share should have been removed if the shelf was shared with this patron' );
 150     $number_of_shelves_shared = Koha::Virtualshelfshares->search->count;
 151     is( $number_of_shelves_shared, 1, 'To be sure the share has been removed' );
 152
 153     teardown();
 154 };
 155
 156 subtest 'Shelf content' => sub {
 157
 158     plan tests => 18;
 159     my $patron1 = $builder->build( { source => 'Borrower', } );
 160     my $patron2 = $builder->build( { source => 'Borrower', } );
 161     my $biblio1 = $builder->build( { source => 'Biblio', } );
 162     my $biblio2 = $builder->build( { source => 'Biblio', } );
 163     my $biblio3 = $builder->build( { source => 'Biblio', } );
 164     my $biblio4 = $builder->build( { source => 'Biblio', } );
 165     my $number_of_contents = Koha::Virtualshelfcontents->search->count;
 166
 167     is( $number_of_contents, 0, 'No content should exist' );
 168
 169     my $dt_yesterday = dt_from_string->subtract_duration( DateTime::Duration->new( days => 1 ) );
 170     my $shelf = Koha::Virtualshelf->new(
 171         {   shelfname    => "my first shelf",
 172             owner        => $patron1->{borrowernumber},
 173             category     => 1,
 174             lastmodified => $dt_yesterday,
 175         }
 176     )->store;
 177
 178     $shelf = Koha::Virtualshelves->find( $shelf->shelfnumber );
 179     is( output_pref( dt_from_string $shelf->lastmodified ), output_pref($dt_yesterday), 'The lastmodified has been set to yesterday, will be useful for another test later' );
 180     my $content1 = $shelf->add_biblio( $biblio1->{biblionumber}, $patron1->{borrowernumber} );
 181     is( ref($content1), 'Koha::Virtualshelfcontent', 'add_biblio to a shelf should return a Koha::Virtualshelfcontent object if inserted' );
 182     $shelf = Koha::Virtualshelves->find( $shelf->shelfnumber );
 183     is( output_pref( dt_from_string( $shelf->lastmodified ) ), output_pref(dt_from_string), 'Adding a biblio to a shelf should update the lastmodified for the shelf' );
 184     my $content2 = $shelf->add_biblio( $biblio2->{biblionumber}, $patron1->{borrowernumber} );
 185     $number_of_contents = Koha::Virtualshelfcontents->search->count;
 186     is( $number_of_contents, 2, '2 biblio should have been inserted' );
 187
 188     my $content1_bis = $shelf->add_biblio( $biblio1->{biblionumber}, $patron1->{borrowernumber} );
 189     is( $content1_bis, undef, 'add_biblio should return undef on duplicate' );    # Or an exception ?
 190     $number_of_contents = Koha::Virtualshelfcontents->search->count;
 191     is( $number_of_contents, 2, 'The biblio should not have been duplicated' );
 192
 193     $shelf = Koha::Virtualshelves->find( $shelf->shelfnumber );
 194     my $contents = $shelf->get_contents;
 195     is( $contents->count, 2, 'There are 2 biblios on this shelf' );
 196
 197     # Patron 2 will try to remove a content
 198     # allow_add = 0, allow_delete_own = 1, allow_delete_other = 0 => Default values
 199     my $number_of_deleted_biblios = $shelf->remove_biblios( { biblionumbers => [ $biblio1->{biblionumber} ], borrowernumber => $patron2->{borrowernumber} } );
 200     is( $number_of_deleted_biblios, 0, 'Patron 2 removed nothing' );
 201     # Now try with patron 1
 202     $number_of_deleted_biblios = $shelf->remove_biblios( { biblionumbers => [ $biblio1->{biblionumber} ], borrowernumber => $patron1->{borrowernumber} } );
 203     is( $number_of_deleted_biblios, 1, 'Patron 1 removed biblio' );
 204
 205     $number_of_contents = Koha::Virtualshelfcontents->search->count;
 206     is( $number_of_contents, 1, 'To be sure the content has been deleted' );
 207
 208     # allow_delete_own = 0
 209     $shelf->allow_delete_own(0);
 210     $shelf->store;
 211     $number_of_deleted_biblios = $shelf->remove_biblios( { biblionumbers => [ $biblio2->{biblionumber} ], borrowernumber => $patron1->{borrowernumber} } );
 212     is( $number_of_deleted_biblios, 1, 'Patron 1 removed another biblio' );
 213     $number_of_contents = Koha::Virtualshelfcontents->search->count;
 214     is( $number_of_contents, 0, 'The biblio should have been deleted to the shelf by the patron, it is his own content (allow_delete_own=0)' );
 215     $shelf->add_biblio( $biblio2->{biblionumber}, $patron1->{borrowernumber} );
 216
 217     # allow_add = 1, allow_delete_own = 1
 218     $shelf->allow_add(1);
 219     $shelf->allow_delete_own(1);
 220     $shelf->store;
 221
 222     my $content3 = $shelf->add_biblio( $biblio3->{biblionumber}, $patron2->{borrowernumber} );
 223     my $content4 = $shelf->add_biblio( $biblio4->{biblionumber}, $patron2->{borrowernumber} );
 224
 225     $number_of_contents = Koha::Virtualshelfcontents->search->count;
 226     is( $number_of_contents, 3, 'The biblio should have been added to the shelf by the patron 2' );
 227
 228     $number_of_deleted_biblios = $shelf->remove_biblios( { biblionumbers => [ $biblio3->{biblionumber} ], borrowernumber => $patron2->{borrowernumber} } );
 229     is( $number_of_deleted_biblios, 1, 'Biblio 3 deleted by patron 2' );
 230     $number_of_contents = Koha::Virtualshelfcontents->search->count;
 231     is( $number_of_contents, 2, 'The biblio should have been deleted to the shelf by the patron, it is his own content (allow_delete_own=1) ' );
 232
 233     # allow_add = 1, allow_delete_own = 1, allow_delete_other = 1
 234     $shelf->allow_delete_other(1);
 235     $shelf->store;
 236
 237     $number_of_deleted_biblios = $shelf->remove_biblios( { biblionumbers => [ $biblio2->{biblionumber} ], borrowernumber => $patron2->{borrowernumber} } );
 238     is( $number_of_deleted_biblios, 1, 'Biblio 2 deleted by patron 2' );
 239     $number_of_contents = Koha::Virtualshelfcontents->search->count;
 240     is( $number_of_contents, 1, 'The biblio should have been deleted to the shelf by the patron 2, even if it is not his own content (allow_delete_other=1)' );
 241
 242     teardown();
 243 };
 244
 245 subtest 'Shelf permissions' => sub {
 246
 247     plan tests => 40;
 248     my $patron1 = $builder->build( { source => 'Borrower', value => { flags => '2096766' } } ); # 2096766 is everything checked but not superlibrarian
 249     my $patron2 = $builder->build( { source => 'Borrower', value => { flags => '1048190' } } ); # 1048190 is everything checked but not superlibrarian and delete_public_lists
 250     my $biblio1 = $builder->build( { source => 'Biblio', } );
 251     my $biblio2 = $builder->build( { source => 'Biblio', } );
 252     my $biblio3 = $builder->build( { source => 'Biblio', } );
 253     my $biblio4 = $builder->build( { source => 'Biblio', } );
 254
 255
 256     my $public_shelf = Koha::Virtualshelf->new(
 257         {   shelfname    => "my first shelf",
 258             owner        => $patron1->{borrowernumber},
 259             category     => 2,
 260             allow_add          => 0,
 261             allow_delete_own   => 0,
 262             allow_delete_other => 0,
 263         }
 264     )->store;
 265
 266     is( $public_shelf->can_be_viewed( $patron1->{borrowernumber} ), 1, 'The owner should be able to view his public list' );
 267     is( $public_shelf->can_be_viewed( $patron2->{borrowernumber} ), 1, 'Public list should be viewed by someone else' );
 268
 269     is( $public_shelf->can_be_deleted( $patron1->{borrowernumber} ), 1, 'The owner should be able to delete his list' );
 270     is( $public_shelf->can_be_deleted( $patron2->{borrowernumber} ), 0, 'Public list should not be deleted by someone else' );
 271
 272     is( $public_shelf->can_be_managed( $patron1->{borrowernumber} ), 1, 'The owner should be able to manage his list' );
 273     is( $public_shelf->can_be_managed( $patron2->{borrowernumber} ), 0, 'Public list should not be managed by someone else' );
 274
 275     is( $public_shelf->can_biblios_be_added( $patron1->{borrowernumber} ), 1, 'The owner should be able to add biblios to his list' );
 276     is( $public_shelf->can_biblios_be_added( $patron2->{borrowernumber} ), 0, 'Public list should not be modified (add) by someone else' );
 277
 278     is( $public_shelf->can_biblios_be_removed( $patron1->{borrowernumber} ), 1, 'The owner should be able to remove biblios to his list' );
 279     is( $public_shelf->can_biblios_be_removed( $patron2->{borrowernumber} ), 0, 'Public list should not be modified (remove) by someone else' );
 280
 281
 282     $public_shelf->allow_add(1);
 283     $public_shelf->allow_delete_own(1);
 284     $public_shelf->allow_delete_other(1);
 285     $public_shelf->store;
 286
 287     is( $public_shelf->can_be_viewed( $patron1->{borrowernumber} ), 1, 'The owner should be able to view his public list' );
 288     is( $public_shelf->can_be_viewed( $patron2->{borrowernumber} ), 1, 'Public list should be viewed by someone else' );
 289
 290     is( $public_shelf->can_be_deleted( $patron1->{borrowernumber} ), 1, 'The owner should be able to delete his list' );
 291     is( $public_shelf->can_be_deleted( $patron2->{borrowernumber} ), 0, 'Public list should not be deleted by someone else' );
 292
 293     is( $public_shelf->can_be_managed( $patron1->{borrowernumber} ), 1, 'The owner should be able to manage his list' );
 294     is( $public_shelf->can_be_managed( $patron2->{borrowernumber} ), 0, 'Public list should not be managed by someone else' );
 295
 296     is( $public_shelf->can_biblios_be_added( $patron1->{borrowernumber} ), 1, 'The owner should be able to add biblios to his list' );
 297     is( $public_shelf->can_biblios_be_added( $patron2->{borrowernumber} ), 1, 'Public list should not be modified (add) by someone else' );
 298
 299     is( $public_shelf->can_biblios_be_removed( $patron1->{borrowernumber} ), 1, 'The owner should be able to remove biblios to his list' );
 300     is( $public_shelf->can_biblios_be_removed( $patron2->{borrowernumber} ), 1, 'Public list should not be modified (remove) by someone else' );
 301
 302
 303     my $private_shelf = Koha::Virtualshelf->new(
 304         {   shelfname    => "my first shelf",
 305             owner        => $patron1->{borrowernumber},
 306             category     => 1,
 307             allow_add          => 0,
 308             allow_delete_own   => 0,
 309             allow_delete_other => 0,
 310         }
 311     )->store;
 312
 313     is( $private_shelf->can_be_viewed( $patron1->{borrowernumber} ), 1, 'The owner should be able to view his list' );
 314     is( $private_shelf->can_be_viewed( $patron2->{borrowernumber} ), 0, 'Private list should not be viewed by someone else' );
 315
 316     is( $private_shelf->can_be_deleted( $patron1->{borrowernumber} ), 1, 'The owner should be able to delete his list' );
 317     is( $private_shelf->can_be_deleted( $patron2->{borrowernumber} ), 0, 'Private list should not be deleted by someone else' );
 318
 319     is( $private_shelf->can_be_managed( $patron1->{borrowernumber} ), 1, 'The owner should be able to manage his list' );
 320     is( $private_shelf->can_be_managed( $patron2->{borrowernumber} ), 0, 'Private list should not be managed by someone else' );
 321
 322     is( $private_shelf->can_biblios_be_added( $patron1->{borrowernumber} ), 1, 'The owner should be able to add biblios to his list' );
 323     is( $private_shelf->can_biblios_be_added( $patron2->{borrowernumber} ), 0, 'Private list should not be modified (add) by someone else' );
 324
 325     is( $private_shelf->can_biblios_be_removed( $patron1->{borrowernumber} ), 1, 'The owner should be able to remove biblios to his list' );
 326     is( $private_shelf->can_biblios_be_removed( $patron2->{borrowernumber} ), 0, 'Private list should not be modified (remove) by someone else' );
 327
 328
 329     $private_shelf->allow_add(1);
 330     $private_shelf->allow_delete_own(1);
 331     $private_shelf->allow_delete_other(1);
 332     $private_shelf->store;
 333
 334     is( $private_shelf->can_be_viewed( $patron1->{borrowernumber} ), 1, 'The owner should be able to view his list' );
 335     is( $private_shelf->can_be_viewed( $patron2->{borrowernumber} ), 0, 'Private list should not be viewed by someone else' );
 336
 337     is( $private_shelf->can_be_deleted( $patron1->{borrowernumber} ), 1, 'The owner should be able to delete his list' );
 338     is( $private_shelf->can_be_deleted( $patron2->{borrowernumber} ), 0, 'Private list should not be deleted by someone else' );
 339
 340     is( $private_shelf->can_be_managed( $patron1->{borrowernumber} ), 1, 'The owner should be able to manage his list' );
 341     is( $private_shelf->can_be_managed( $patron2->{borrowernumber} ), 0, 'Private list should not be managed by someone else' );
 342
 343     is( $private_shelf->can_biblios_be_added( $patron1->{borrowernumber} ), 1, 'The owner should be able to add biblios to his list' );
 344     is( $private_shelf->can_biblios_be_added( $patron2->{borrowernumber} ), 1, 'Private list could be modified (add) by someone else # individual check done later' );
 345
 346     is( $private_shelf->can_biblios_be_removed( $patron1->{borrowernumber} ), 1, 'The owner should be able to remove biblios to his list' );
 347     is( $private_shelf->can_biblios_be_removed( $patron2->{borrowernumber} ), 1, 'Private list could be modified (remove) by someone else # individual check done later' );
 348
 349     teardown();
 350 };
 351
 352 subtest 'Get shelves' => sub {
 353     plan tests => 4;
 354     my $patron1 = $builder->build({
 355         source => 'Borrower',
 356     });
 357     my $patron2 = $builder->build({
 358         source => 'Borrower',
 359     });
 360
 361     my $private_shelf1_1 = Koha::Virtualshelf->new({
 362             shelfname => "private shelf 1 for patron 1",
 363             owner => $patron1->{borrowernumber},
 364             category => 1,
 365         }
 366     )->store;
 367     my $private_shelf1_2 = Koha::Virtualshelf->new({
 368             shelfname => "private shelf 2 for patron 1",
 369             owner => $patron1->{borrowernumber},
 370             category => 1,
 371         }
 372     )->store;
 373     my $private_shelf2_1 = Koha::Virtualshelf->new({
 374             shelfname => "private shelf 1 for patron 2",
 375             owner => $patron2->{borrowernumber},
 376             category => 1,
 377         }
 378     )->store;
 379     my $public_shelf1_1 = Koha::Virtualshelf->new({
 380             shelfname => "public shelf 1 for patron 1",
 381             owner => $patron1->{borrowernumber},
 382             category => 2,
 383         }
 384     )->store;
 385     my $public_shelf1_2 = Koha::Virtualshelf->new({
 386             shelfname => "public shelf 2 for patron 1",
 387             owner => $patron1->{borrowernumber},
 388             category => 2,
 389         }
 390     )->store;
 391
 392     my $private_shelves = Koha::Virtualshelves->get_private_shelves;
 393     is( $private_shelves->count, 0, 'Without borrowernumber given, get_private_shelves should not return any shelf' );
 394     $private_shelves = Koha::Virtualshelves->get_private_shelves({ borrowernumber => $patron1->{borrowernumber} });
 395     is( $private_shelves->count, 2, 'get_private_shelves should return all shelves for a given patron' );
 396
 397     $private_shelf2_1->share('a key')->accept('a key', $patron1->{borrowernumber});
 398     $private_shelves = Koha::Virtualshelves->get_private_shelves({ borrowernumber => $patron1->{borrowernumber} });
 399     is( $private_shelves->count, 3, 'get_private_shelves should return all shelves for a given patron, even the shared ones' );
 400
 401     my $public_shelves = Koha::Virtualshelves->get_public_shelves;
 402     is( $public_shelves->count, 2, 'get_public_shelves should return all public shelves, no matter who is the owner' );
 403
 404     teardown();
 405 };
 406
 407 subtest 'Get shelves containing biblios' => sub {
 408
 409     plan tests => 9;
 410     my $patron1 = $builder->build( { source => 'Borrower', } );
 411     my $patron2 = $builder->build( { source => 'Borrower', } );
 412     my $biblio1 = $builder->build( { source => 'Biblio', } );
 413     my $biblio2 = $builder->build( { source => 'Biblio', } );
 414     my $biblio3 = $builder->build( { source => 'Biblio', } );
 415     my $biblio4 = $builder->build( { source => 'Biblio', } );
 416
 417     my $shelf1 = Koha::Virtualshelf->new(
 418         {   shelfname    => "my first shelf",
 419             owner        => $patron1->{borrowernumber},
 420             category     => 1,
 421         }
 422     )->store;
 423     my $shelf2 = Koha::Virtualshelf->new(
 424         {   shelfname    => "my x second shelf", # 'x' to make it sorted after 'third'
 425             owner        => $patron2->{borrowernumber},
 426             category     => 1,
 427         }
 428     )->store;
 429     my $shelf3 = Koha::Virtualshelf->new(
 430         {   shelfname    => "my third shelf",
 431             owner        => $patron1->{borrowernumber},
 432             category     => 2,
 433         }
 434     )->store;
 435
 436     my $content1 = $shelf1->add_biblio( $biblio1->{biblionumber}, $patron1->{borrowernumber} );
 437     my $content2 = $shelf1->add_biblio( $biblio2->{biblionumber}, $patron1->{borrowernumber} );
 438     my $content3 = $shelf2->add_biblio( $biblio2->{biblionumber}, $patron2->{borrowernumber} );
 439     my $content4 = $shelf2->add_biblio( $biblio3->{biblionumber}, $patron2->{borrowernumber} );
 440     my $content5 = $shelf2->add_biblio( $biblio4->{biblionumber}, $patron2->{borrowernumber} );
 441     my $content6 = $shelf3->add_biblio( $biblio4->{biblionumber}, $patron2->{borrowernumber} );
 442
 443     my $shelves_with_biblio1_for_any_patrons = Koha::Virtualshelves->get_shelves_containing_record(
 444         {
 445             biblionumber => $biblio1->{biblionumber},
 446         }
 447     );
 448     is ( $shelves_with_biblio1_for_any_patrons->count, 0, 'shelf1 is private and should not be displayed if patron is not logged in' );
 449
 450     my $shelves_with_biblio4_for_any_patrons = Koha::Virtualshelves->get_shelves_containing_record(
 451         {
 452             biblionumber => $biblio4->{biblionumber},
 453         }
 454     );
 455     is ( $shelves_with_biblio4_for_any_patrons->count, 1, 'shelf3 is public and should be displayed for any patrons' );
 456     is ( $shelves_with_biblio4_for_any_patrons->next->shelfname, $shelf3->shelfname, 'The correct shelf (3) should be displayed' );
 457
 458     my $shelves_with_biblio1_for_other_patrons = Koha::Virtualshelves->get_shelves_containing_record(
 459         {
 460             biblionumber => $biblio1->{biblionumber},
 461             borrowernumber => $patron2->{borrowernumber},
 462         }
 463     );
 464     is ( $shelves_with_biblio1_for_other_patrons->count, 0, 'shelf1 is private and should not be displayed for other patrons' );
 465
 466     my $shelves_with_biblio1_for_owner = Koha::Virtualshelves->get_shelves_containing_record(
 467         {
 468             biblionumber => $biblio1->{biblionumber},
 469             borrowernumber => $patron1->{borrowernumber},
 470         }
 471     );
 472     is ( $shelves_with_biblio1_for_owner->count, 1, 'shelf1 is private and should be displayed for the owner' );
 473
 474     my $shelves_with_biblio2_for_patron1 = Koha::Virtualshelves->get_shelves_containing_record(
 475         {
 476             biblionumber => $biblio2->{biblionumber},
 477             borrowernumber => $patron1->{borrowernumber},
 478         }
 479     );
 480     is ( $shelves_with_biblio2_for_patron1->count, 1, 'Only shelf1 should be displayed for patron 1 and biblio 1' );
 481     is ( $shelves_with_biblio2_for_patron1->next->shelfname, $shelf1->shelfname, 'The correct shelf (1) should be displayed for patron 1' );
 482
 483     my $shelves_with_biblio4_for_patron2 = Koha::Virtualshelves->get_shelves_containing_record(
 484         {
 485             biblionumber => $biblio4->{biblionumber},
 486             borrowernumber => $patron2->{borrowernumber},
 487         }
 488     );
 489     is ( $shelves_with_biblio4_for_patron2->count, 2, 'Patron should shown private and public lists for a given biblio' );
 490     is ( $shelves_with_biblio4_for_patron2->next->shelfname, $shelf3->shelfname, 'The shelves should be sorted by shelfname' );
 491
 492     teardown();
 493 };
 494
 495 sub teardown {
 496     $dbh->do(q|DELETE FROM virtualshelfshares|);
 497     $dbh->do(q|DELETE FROM virtualshelfcontents|);
 498     $dbh->do(q|DELETE FROM virtualshelves|);
 499 }