t/db_dependent/VirtualShelves.t

   1 #!/usr/bin/perl
   2
   3 #
   4 # This file is a test script for C4::VirtualShelves.pm
   5 # Author : Antoine Farnault, antoine@koha-fr.org
   6 #
   7
   8 use strict;
   9 use warnings;
  10 use C4::Context;
  11
  12 use Test::More tests => 82;
  13
  14 # Getting some borrowers from database.
  15 my $dbh = C4::Context->dbh;
  16 my $query = qq/
  17     SELECT borrowernumber
  18     FROM   borrowers
  19     LIMIT  10
  20 /;
  21 my $sth = $dbh->prepare($query);
  22 $sth->execute;
  23 my @borrowers;
  24 while(my $borrower = $sth->fetchrow){
  25     push @borrowers, $borrower;
  26 }
  27
  28 # Getting some biblionumbers from database
  29 $query = qq/
  30     SELECT biblionumber
  31     FROM   biblio
  32     LIMIT  10
  33 /;
  34 $sth = $dbh->prepare($query);
  35 $sth->execute;
  36 my @biblionumbers;
  37 while(my $biblionumber = $sth->fetchrow){
  38     push @biblionumbers, $biblionumber;
  39 }
  40
  41 # ---
  42 my $delete_virtualshelf = qq/
  43     DELETE FROM  virtualshelves WHERE 1
  44 /;
  45 my $delete_virtualshelfcontent =qq/
  46     DELETE  FROM  virtualshelfcontents WHERE 1
  47 /;
  48
  49 $sth = $dbh->prepare($delete_virtualshelf);
  50 $sth->execute;
  51 $sth = $dbh->prepare($delete_virtualshelfcontent);
  52 $sth->execute;
  53 # ---
  54
  55 #----------------------------------------------------------------------#
  56 #
  57 #           TESTS START HERE
  58 #
  59 #----------------------------------------------------------------------#
  60
  61 use_ok('C4::VirtualShelves');
  62
  63 #-----------------------TEST AddShelf function------------------------#
  64 # usage : $shelfnumber = &AddShelf( $shelfname, $owner, $category);
  65
  66 # creating 10 good shelves.
  67 my @shelves;
  68 for(my $i=0; $i<10;$i++){
  69      my $ShelfNumber = AddShelf(
  70         {shelfname=>"Shelf_".$i, category=>int(rand(2))+1 }, $borrowers[$i] );
  71      die "test Not ok, remove some shelves before" if ($ShelfNumber == -1);
  72      ok($ShelfNumber > -1, "created shelf");   # Shelf creation successful;
  73      push @shelves, $ShelfNumber if $ShelfNumber > -1;
  74 }
  75
  76 ok(10 == scalar @shelves, 'created 10 lists'); # 10 shelves in @shelves;
  77
  78 # try to create some shelf which already exists.
  79 for(my $i=0;$i<10;$i++){
  80     my @shlf=GetShelf($shelves[$i]);
  81     my $badNumShelf = AddShelf(
  82         {shelfname=>"Shelf_".$i, category=>$shlf[3] }, $borrowers[$i]);
  83     ok(-1 == $badNumShelf, 'do not create lists with duplicate names');   # AddShelf returns -1 if name already exist.
  84 }
  85
  86 #-----------TEST AddToShelf & GetShelfContents &  DelFromShelf functions--------------#
  87 # usage : &AddToShelf($biblionumber, $shelfnumber);
  88 # usage : $biblist = &GetShelfContents($shelfnumber);
  89 # usage : $biblist = GetShelfContents($shelfnumber);
  90
  91 my %used = ();
  92 for(my $i=0; $i<10;$i++){
  93     my $bib = $biblionumbers[int(rand(9))];
  94     my $shelfnumber = $shelves[int(rand(9))];
  95
  96     my $key = "$bib\t$shelfnumber";
  97     my $should_fail = exists($used{$key}) ? 1 : 0;
  98
  99     my ($biblistBefore,$countbefore) = GetShelfContents($shelfnumber);
 100     my $status = AddToShelf($bib,$shelfnumber,$borrowers[$i]);
 101     my ($biblistAfter,$countafter) = GetShelfContents($shelfnumber);
 102
 103     if ($should_fail) {
 104         ok(!defined($status), 'failed to add to list when we should');
 105     } else {
 106         ok(defined($status), 'added to list when we should');
 107     }
 108
 109     if (defined $status) {
 110         ok($countbefore == $countafter - 1, 'added bib to list');  # the bib has been successfuly added.
 111     } else {
 112         ok($countbefore == $countafter,     'did not add duplicate bib to list');  # the bib has been successfuly added.
 113     }
 114
 115     $used{$key}++;
 116
 117 }
 118
 119 #-----------------------TEST ModShelf & GetShelf functions------------------------#
 120 # usage : ModShelf($shelfnumber, $shelfname, $owner, $category )
 121 # usage : (shelfnumber,shelfname,owner,category) = GetShelf($shelfnumber);
 122
 123 for(my $i=0; $i<10;$i++){
 124     my $rand = int(rand(9));
 125     my $numA = $shelves[$rand];
 126     my $shelf = { shelfname => "NewName_".$rand,
 127         category =>  int(rand(2))+1 };
 128
 129     ModShelf($numA,$shelf);
 130     my ($numB,$nameB,$ownerB,$categoryB) = GetShelf($numA);
 131
 132     ok($numA == $numB, 'modified shelf');
 133     ok($shelf->{shelfname} eq $nameB,     '... and name change took');
 134     ok($shelf->{category}  eq $categoryB, '... and category change took');
 135 }
 136
 137 #-----------------------TEST DelShelf & DelFromShelf functions------------------------#
 138 # usage : ($status) = &DelShelf($shelfnumber);
 139
 140 for(my $i=0; $i<10;$i++){
 141     my $shelfnumber = $shelves[$i];
 142     my $status = DelShelf($shelfnumber);
 143     ok(1 == $status, "deleted shelf $shelfnumber and its contents");
 144 }