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     [ [ position => 1 ], [ position => 2 ] ],
  58     'count_rs db-side limit applied',
  59   );
  60 }
  61
  62 # has_many prefetch with limit
  63 {
  64   my $rs = $schema->resultset("Artist")
  65             ->search_related('cds',
  66                 { 'tracks.position' => [1,2] },
  67                 { prefetch => [qw/tracks artist/], rows => 3, offset => 4 },
  68             );
  69   is ($rs->all, 1, 'Correct number of objects');
  70
  71   my ($sql, @bind);
  72   $schema->storage->debugobj(DBIC::DebugObj->new(\$sql, \@bind));
  73   $schema->storage->debug(1);
  74
  75   is ($rs->count, 1, 'Correct count via count()');
  76
  77   is_same_sql_bind (
  78     $sql,
  79     \@bind,
  80     'SELECT COUNT( * )
  81       FROM (
  82         SELECT cds.cdid
  83           FROM artist me
  84           JOIN cd cds ON cds.artist = me.artistid
  85           LEFT JOIN track tracks ON tracks.cd = cds.cdid
  86           JOIN artist artist ON artist.artistid = cds.artist
  87         WHERE tracks.position = ? OR tracks.position = ?
  88         GROUP BY cds.cdid
  89       ) cds
  90     ',
  91     [ qw/'1' '2'/ ],
  92     'count softlimit applied',
  93   );
  94
  95   my $crs = $rs->count_rs;
  96   is ($crs->next, 1, 'Correct count via count_rs()');
  97
  98   is_same_sql_bind (
  99     $crs->as_query,
 100     '(SELECT COUNT( * )
 101       FROM (
 102         SELECT cds.cdid
 103           FROM artist me
 104           JOIN cd cds ON cds.artist = me.artistid
 105           LEFT JOIN track tracks ON tracks.cd = cds.cdid
 106           JOIN artist artist ON artist.artistid = cds.artist
 107         WHERE tracks.position = ? OR tracks.position = ?
 108         GROUP BY cds.cdid
 109         LIMIT 3 OFFSET 4
 110       ) cds
 111     )',
 112     [ [ 'tracks.position' => 1 ], [ 'tracks.position' => 2 ] ],
 113     'count_rs db-side limit applied',
 114   );
 115 }
 116
 117 # count with a having clause
 118 {
 119   my $rs = $schema->resultset("Artist")->search(
 120     {},
 121     {
 122       join      => 'cds',
 123       group_by  => 'me.artistid',
 124       '+select' => [ { max => 'cds.year', -as => 'newest_cd_year' } ],
 125       '+as'     => ['newest_cd_year'],
 126       having    => { 'newest_cd_year' => '2001' }
 127     }
 128   );
 129
 130   my $crs = $rs->count_rs;
 131
 132   is_same_sql_bind (
 133     $crs->as_query,
 134     '(SELECT COUNT( * )
 135       FROM (
 136         SELECT MAX( cds.year ) AS newest_cd_year, me.artistid
 137           FROM artist me
 138           LEFT JOIN cd cds ON cds.artist = me.artistid
 139         GROUP BY me.artistid
 140         HAVING newest_cd_year = ?
 141       ) me
 142     )',
 143     [ [ 'newest_cd_year' => '2001' ],],
 144     'count with having clause keeps sql as alias',
 145   );
 146
 147   is ($crs->next, 2, 'Correct artist count (each with one 2001 cd)');
 148 }
 149
 150 done_testing;