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 use DBIC::DebugObj;
  10
  11 plan tests => 9;
  12
  13 my $schema = DBICTest->init_schema();
  14
  15 # collapsing prefetch
  16 {
  17   my $rs = $schema->resultset("Artist")
  18             ->search_related('cds',
  19                 { 'tracks.position' => [1,2] },
  20                 { prefetch => [qw/tracks artist/] },
  21             );
  22   is ($rs->all, 5, 'Correct number of objects');
  23
  24
  25   my ($sql, @bind);
  26   $schema->storage->debugobj(DBIC::DebugObj->new(\$sql, \@bind));
  27   $schema->storage->debug(1);
  28
  29
  30   is ($rs->count, 5, 'Correct count');
  31
  32   is_same_sql_bind (
  33     $sql,
  34     \@bind,
  35     'SELECT COUNT( * ) FROM (SELECT cds.cdid FROM artist me JOIN cd cds ON cds.artist = me.artistid LEFT JOIN track tracks ON tracks.cd = cds.cdid JOIN artist artist ON artist.artistid = cds.artist WHERE tracks.position = ? OR tracks.position = ? GROUP BY cds.cdid) count_subq',
  36     [ qw/'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' => 'foo' },
  45                 { prefetch => q(cds) },
  46             );
  47   is ($rs->all, 5, 'Correct number of objects');
  48
  49
  50   my ($sql, @bind);
  51   $schema->storage->debugobj(DBIC::DebugObj->new(\$sql, \@bind));
  52   $schema->storage->debug(1);
  53
  54
  55   is ($rs->count, 5, 'Correct count');
  56
  57   is_same_sql_bind (
  58     $sql,
  59     \@bind,
  60     'SELECT COUNT( * ) FROM (SELECT cds.cdid FROM artist me JOIN cd cds ON cds.artist = me.artistid LEFT JOIN track tracks ON tracks.cd = cds.cdid JOIN artist artist ON artist.artistid = cds.artist WHERE tracks.position = ? OR tracks.position = ? GROUP BY cds.cdid) count_subq',
  61     [ qw/'1' '2'/ ],
  62   );
  63 }
  64
  65 # non-collapsing prefetch (no multi prefetches)
  66 {
  67   my $rs = $schema->resultset("CD")
  68             ->search_related('tracks',
  69                 { position => [1,2] },
  70                 { prefetch => [qw/disc lyrics/] },
  71             );
  72   is ($rs->all, 10, 'Correct number of objects');
  73
  74
  75   my ($sql, @bind);
  76   $schema->storage->debugobj(DBIC::DebugObj->new(\$sql, \@bind));
  77   $schema->storage->debug(1);
  78
  79
  80   is ($rs->count, 10, 'Correct count');
  81
  82   is_same_sql_bind (
  83     $sql,
  84     \@bind,
  85     'SELECT COUNT( * ) FROM cd me JOIN track tracks ON tracks.cd = me.cdid JOIN cd disc ON disc.cdid = tracks.cd LEFT JOIN lyrics lyrics ON lyrics.track_id = tracks.trackid WHERE ( ( position = ? OR position = ? ) )',
  86     [ qw/'1' '2'/ ],
  87   );
  88 }