t/cdbi/12-filter.t

   1 use strict;
   2 use Test::More;
   3
   4 BEGIN {
   5   eval "use DBIx::Class::CDBICompat;";
   6   if ($@) {
   7     plan (skip_all => 'Class::Trigger and DBIx::ContextualFetch required');
   8     next;
   9   }
  10   eval "use DBD::SQLite";
  11   plan $@ ? (skip_all => 'needs DBD::SQLite for testing') : (tests => 50);
  12 }
  13
  14 use lib 't/cdbi/testlib';
  15 use Actor;
  16 use Film;
  17 Film->has_many(actors                => 'Actor');
  18 Actor->has_a('film'                  => 'Film');
  19 Actor->add_constructor(double_search => 'name = ? AND salary = ?');
  20
  21 my $film  = Film->create({ Title => 'MY Film' });
  22 my $film2 = Film->create({ Title => 'Another Film' });
  23
  24 my @act = (
  25   Actor->create(
  26     {
  27       name   => 'Actor 1',
  28       film   => $film,
  29       salary => 10,
  30     }
  31   ),
  32   Actor->create(
  33     {
  34       name   => 'Actor 2',
  35       film   => $film,
  36       salary => 20,
  37     }
  38   ),
  39   Actor->create(
  40     {
  41       name   => 'Actor 3',
  42       film   => $film,
  43       salary => 30,
  44     }
  45   ),
  46   Actor->create(
  47     {
  48       name   => 'Actor 4',
  49       film   => $film2,
  50       salary => 50,
  51     }
  52   ),
  53 );
  54
  55 eval {
  56   my @actors = $film->actors(name => 'Actor 1');
  57   is @actors, 1, "Got one actor from restricted has_many";
  58   is $actors[0]->name, "Actor 1", "Correct name";
  59 };
  60 is $@, '', "No errors";
  61
  62 {
  63   my @actors = Actor->double_search("Actor 1", 10);
  64   is @actors, 1, "Got one actor";
  65   is $actors[0]->name, "Actor 1", "Correct name";
  66 }
  67
  68 {
  69   ok my @actors = Actor->salary_between(0, 100), "Range 0 - 100";
  70   is @actors, 4, "Got all";
  71 }
  72
  73 {
  74   my @actors = Actor->salary_between(100, 200);
  75   is @actors, 0, "None in Range 100 - 200";
  76 }
  77
  78 {
  79   ok my @actors = Actor->salary_between(0, 10), "Range 0 - 10";
  80   is @actors, 1, "Got 1";
  81   is $actors[0]->name, $act[0]->name, "Actor 1";
  82 }
  83
  84 {
  85   ok my @actors = Actor->salary_between(20, 30), "Range 20 - 20";
  86   @actors = sort { $a->salary <=> $b->salary } @actors;
  87   is @actors, 2, "Got 2";
  88   is $actors[0]->name, $act[1]->name, "Actor 2";
  89   is $actors[1]->name, $act[2]->name, "and Actor 3";
  90 }
  91
  92 {
  93   ok my @actors = Actor->search(Film => $film), "Search by object";
  94   is @actors, 3, "3 actors in film 1";
  95 }
  96
  97 #----------------------------------------------------------------------
  98 # Iterators
  99 #----------------------------------------------------------------------
 100
 101 my $it_class = 'DBIx::Class::ResultSet';
 102
 103 sub test_normal_iterator {
 104   my $it = $film->actors;
 105   isa_ok $it, $it_class;
 106   is $it->count, 3, " - with 3 elements";
 107   my $i = 0;
 108   while (my $film = $it->next) {
 109     is $film->name, $act[ $i++ ]->name, "Get $i";
 110   }
 111   ok !$it->next, "No more";
 112   is $it->first->name, $act[0]->name, "Get first";
 113 }
 114
 115 test_normal_iterator;
 116 {
 117   Film->has_many(actor_ids => [ Actor => 'id' ]);
 118   my $it = $film->actor_ids;
 119   isa_ok $it, $it_class;
 120   is $it->count, 3, " - with 3 elements";
 121   my $i = 0;
 122   while (my $film_id = $it->next) {
 123     is $film_id, $act[ $i++ ]->id, "Get id $i";
 124   }
 125   ok !$it->next, "No more";
 126   is $it->first, $act[0]->id, "Get first";
 127 }
 128
 129 # make sure nothing gets clobbered;
 130 test_normal_iterator;
 131
 132 SKIP: {
 133   #skip "dbic iterators don't support slice yet", 12;
 134
 135
 136 {
 137   my @acts = $film->actors->slice(1, 2);
 138   is @acts, 2, "Slice gives 2 actor";
 139   is $acts[0]->name, "Actor 2", "Actor 2";
 140   is $acts[1]->name, "Actor 3", "and actor 3";
 141 }
 142
 143 {
 144   my @acts = $film->actors->slice(1);
 145   is @acts, 1, "Slice of 1 actor";
 146   is $acts[0]->name, "Actor 2", "Actor 2";
 147 }
 148
 149 {
 150   my @acts = $film->actors->slice(2, 8);
 151   is @acts, 1, "Slice off the end";
 152   is $acts[0]->name, "Actor 3", "Gets last actor only";
 153 }
 154
 155 package Class::DBI::My::Iterator;
 156
 157 use vars qw/@ISA/;
 158
 159 @ISA = ($it_class);
 160
 161 sub slice { qw/fred barney/ }
 162
 163 package main;
 164
 165 Actor->iterator_class('Class::DBI::My::Iterator');
 166
 167 delete $film->{related_resultsets};
 168
 169 {
 170   my @acts = $film->actors->slice(1, 2);
 171   is @acts, 2, "Slice gives 2 results";
 172   ok eq_set(\@acts, [qw/fred barney/]), "Fred and Barney";
 173
 174   ok $film->actors->delete_all, "Can delete via iterator";
 175   is $film->actors, 0, "no actors left";
 176
 177   eval { $film->actors->delete_all };
 178   is $@, '', "Deleting again does no harm";
 179 }
 180
 181 } # end SKIP block