t/count/prefetch.t

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