t/count/count_rs.t

   1 use strict;
   2 use warnings;
   3
   4 use lib qw(t/lib);
   5
   6 use Test::More;
   7 use DBICTest;
   8 use DBIC::SqlMakerTest;
   9 use DBIC::DebugObj;
  10
  11 my $schema = DBICTest->init_schema();
  12
  13 # non-collapsing prefetch (no multi prefetches)
  14 {
  15   my $rs = $schema->resultset("CD")
  16             ->search_related('tracks',
  17                 { position => [1,2] },
  18                 { prefetch => [qw/disc lyrics/], rows => 3, offset => 8 },
  19             );
  20   is ($rs->all, 2, 'Correct number of objects');
  21
  22
  23   my ($sql, @bind);
  24   $schema->storage->debugobj(DBIC::DebugObj->new(\$sql, \@bind));
  25   $schema->storage->debug(1);
  26
  27   is ($rs->count, 2, 'Correct count via count()');
  28
  29   is_same_sql_bind (
  30     $sql,
  31     \@bind,
  32     'SELECT COUNT( * )
  33       FROM cd me
  34       JOIN track tracks ON tracks.cd = me.cdid
  35       JOIN cd disc ON disc.cdid = tracks.cd
  36      WHERE ( ( position = ? OR position = ? ) )
  37     ',
  38     [ qw/'1' '2'/ ],
  39     'count softlimit applied',
  40   );
  41
  42   my $crs = $rs->count_rs;
  43   is ($crs->next, 2, 'Correct count via count_rs()');
  44
  45   is_same_sql_bind (
  46     $crs->as_query,
  47     '(SELECT COUNT( * )
  48        FROM (
  49         SELECT tracks.trackid
  50           FROM cd me
  51           JOIN track tracks ON tracks.cd = me.cdid
  52           JOIN cd disc ON disc.cdid = tracks.cd
  53         WHERE ( ( position = ? OR position = ? ) )
  54         LIMIT 3 OFFSET 8
  55        ) tracks
  56     )',
  57     [
  58       [ { sqlt_datatype => 'int', dbic_colname => 'position' }
  59         => 1 ],
  60       [ { sqlt_datatype => 'int', dbic_colname => 'position' }
  61         => 2 ],
  62     ],
  63     'count_rs db-side limit applied',
  64   );
  65 }
  66
  67 # has_many prefetch with limit
  68 {
  69   my $rs = $schema->resultset("Artist")
  70             ->search_related('cds',
  71                 { 'tracks.position' => [1,2] },
  72                 { prefetch => [qw/tracks artist/], rows => 3, offset => 4 },
  73             );
  74   is ($rs->all, 1, 'Correct number of objects');
  75
  76   my ($sql, @bind);
  77   $schema->storage->debugobj(DBIC::DebugObj->new(\$sql, \@bind));
  78   $schema->storage->debug(1);
  79
  80   is ($rs->count, 1, 'Correct count via count()');
  81
  82   is_same_sql_bind (
  83     $sql,
  84     \@bind,
  85     'SELECT COUNT( * )
  86       FROM (
  87         SELECT cds.cdid
  88           FROM artist me
  89           JOIN cd cds ON cds.artist = me.artistid
  90           LEFT JOIN track tracks ON tracks.cd = cds.cdid
  91           JOIN artist artist ON artist.artistid = cds.artist
  92         WHERE tracks.position = ? OR tracks.position = ?
  93         GROUP BY cds.cdid
  94       ) cds
  95     ',
  96     [ qw/'1' '2'/ ],
  97     'count softlimit applied',
  98   );
  99
 100   my $crs = $rs->count_rs;
 101   is ($crs->next, 1, 'Correct count via count_rs()');
 102
 103   is_same_sql_bind (
 104     $crs->as_query,
 105     '(SELECT COUNT( * )
 106       FROM (
 107         SELECT cds.cdid
 108           FROM artist me
 109           JOIN cd cds ON cds.artist = me.artistid
 110           LEFT JOIN track tracks ON tracks.cd = cds.cdid
 111           JOIN artist artist ON artist.artistid = cds.artist
 112         WHERE tracks.position = ? OR tracks.position = ?
 113         GROUP BY cds.cdid
 114         LIMIT 3 OFFSET 4
 115       ) cds
 116     )',
 117     [
 118       [ { sqlt_datatype => 'int', dbic_colname => 'tracks.position' }
 119         => 1 ],
 120       [ { sqlt_datatype => 'int', dbic_colname => 'tracks.position' }
 121         => 2 ],
 122     ],
 123     'count_rs db-side limit applied',
 124   );
 125 }
 126
 127 # count with a having clause
 128 {
 129   my $rs = $schema->resultset("Artist")->search(
 130     {},
 131     {
 132       join      => 'cds',
 133       group_by  => 'me.artistid',
 134       '+select' => [ { max => 'cds.year', -as => 'newest_cd_year' } ],
 135       '+as'     => ['newest_cd_year'],
 136       having    => { 'newest_cd_year' => '2001' }
 137     }
 138   );
 139
 140   my $crs = $rs->count_rs;
 141
 142   is_same_sql_bind (
 143     $crs->as_query,
 144     '(SELECT COUNT( * )
 145       FROM (
 146         SELECT me.artistid, MAX( cds.year ) AS newest_cd_year,
 147           FROM artist me
 148           LEFT JOIN cd cds ON cds.artist = me.artistid
 149         GROUP BY me.artistid
 150         HAVING newest_cd_year = ?
 151       ) me
 152     )',
 153     [ [ { dbic_colname => 'newest_cd_year' }
 154           => '2001' ] ],
 155     'count with having clause keeps sql as alias',
 156   );
 157
 158   is ($crs->next, 2, 'Correct artist count (each with one 2001 cd)');
 159 }
 160
 161 done_testing;