t/count/prefetch.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
  10 my $schema = DBICTest->init_schema();
  11
  12 # collapsing prefetch
  13 {
  14   my $rs = $schema->resultset("Artist")
  15             ->search_related('cds',
  16                 { 'tracks.position' => [1,2] },
  17                 { prefetch => [qw/tracks artist/] },
  18             );
  19   is ($rs->all, 5, 'Correct number of objects');
  20   is ($rs->count, 5, 'Correct count');
  21
  22   is_same_sql_bind (
  23     $rs->count_rs->as_query,
  24     '(
  25       SELECT COUNT( * )
  26         FROM (
  27           SELECT cds.cdid
  28             FROM artist me
  29             JOIN cd cds ON cds.artist = me.artistid
  30             LEFT JOIN track tracks ON tracks.cd = cds.cdid
  31             JOIN artist artist ON artist.artistid = cds.artist
  32           WHERE tracks.position = ? OR tracks.position = ?
  33           GROUP BY cds.cdid
  34         ) count_subq
  35     )',
  36     [ map { [ 'tracks.position' => $_ ] } (1, 2) ],
  37   );
  38 }
  39
  40 # collapsing prefetch with distinct
  41 {
  42   my $rs = $schema->resultset("Artist")->search(undef, {distinct => 1})
  43             ->search_related('cds')->search_related('genre',
  44                 { 'genre.name' => 'emo' },
  45                 { prefetch => q(cds) },
  46             );
  47   is ($rs->all, 1, 'Correct number of objects');
  48   is ($rs->count, 1, 'Correct count');
  49
  50   is_same_sql_bind (
  51     $rs->count_rs->as_query,
  52     '(
  53       SELECT COUNT( * )
  54         FROM (
  55           SELECT genre.genreid
  56             FROM (
  57               SELECT cds.cdid, cds.artist, cds.title, cds.year, cds.genreid, cds.single_track
  58                 FROM (
  59                   SELECT me.artistid, me.name, me.rank, me.charfield
  60                     FROM artist me GROUP BY me.artistid, me.name, me.rank, me.charfield
  61                 ) me
  62                 LEFT JOIN cd cds ON cds.artist = me.artistid
  63               GROUP BY cds.cdid, cds.artist, cds.title, cds.year, cds.genreid, cds.single_track
  64             ) cds
  65             JOIN genre genre ON genre.genreid = cds.genreid
  66             LEFT JOIN cd cds_2 ON cds_2.genreid = genre.genreid
  67           WHERE ( genre.name = ? )
  68           GROUP BY genre.genreid
  69         ) count_subq
  70     )',
  71     [ [ 'genre.name' => 'emo' ] ],
  72   );
  73 }
  74
  75 # non-collapsing prefetch (no multi prefetches)
  76 {
  77   my $rs = $schema->resultset("CD")
  78             ->search_related('tracks',
  79                 { position => [1,2] },
  80                 { prefetch => [qw/disc lyrics/] },
  81             );
  82   is ($rs->all, 10, 'Correct number of objects');
  83
  84
  85   is ($rs->count, 10, 'Correct count');
  86
  87   is_same_sql_bind (
  88     $rs->count_rs->as_query,
  89     '(
  90       SELECT COUNT( * )
  91         FROM cd me
  92         JOIN track tracks ON tracks.cd = me.cdid
  93         JOIN cd disc ON disc.cdid = tracks.cd
  94         LEFT JOIN lyrics lyrics ON lyrics.track_id = tracks.trackid
  95       WHERE position = ? OR position = ?
  96     )',
  97     [ map { [ position => $_ ] } (1, 2) ],
  98   );
  99 }
 100
 101 done_testing;