t/prefetch/count.t

   1 use strict;
   2 use warnings;
   3
   4 use Test::More;
   5 use lib qw(t/lib);
   6 use DBICTest ':DiffSQL';
   7
   8 my $schema = DBICTest->init_schema();
   9
  10 my $cd_rs = $schema->resultset('CD')->search (
  11   { 'tracks.cd' => { '!=', undef } },
  12   { prefetch => ['tracks', 'artist'] },
  13 );
  14
  15 is($cd_rs->count, 5, 'CDs with tracks count');
  16 is($cd_rs->search_related('tracks')->count, 15, 'Tracks associated with CDs count (before SELECT()ing)');
  17
  18 is($cd_rs->all, 5, 'Amount of CD objects with tracks');
  19 is($cd_rs->search_related('tracks')->count, 15, 'Tracks associated with CDs count (after SELECT()ing)');
  20
  21 is($cd_rs->search_related ('tracks')->all, 15, 'Track objects associated with CDs (after SELECT()ing)');
  22
  23 my $artist = $schema->resultset('Artist')->create({name => 'xxx'});
  24
  25 my $artist_rs = $schema->resultset('Artist')->search(
  26   {artistid => $artist->id},
  27   {prefetch=>'cds', join => 'twokeys' }
  28 );
  29
  30 is($artist_rs->count, 1, "New artist found with prefetch turned on");
  31 is(scalar($artist_rs->all), 1, "New artist fetched with prefetch turned on");
  32 is($artist_rs->related_resultset('cds')->count, 0, "No CDs counted on a brand new artist");
  33 is(scalar($artist_rs->related_resultset('cds')->all), 0, "No CDs fetched on a brand new artist (count == fetch)");
  34
  35 # create a cd, and make sure the non-existing join does not skew the count
  36 $artist->create_related ('cds', { title => 'yyy', year => '1999' });
  37 is($artist_rs->related_resultset('cds')->count, 1, "1 CDs counted on a brand new artist");
  38 is(scalar($artist_rs->related_resultset('cds')->all), 1, "1 CDs prefetched on a brand new artist (count == fetch)");
  39
  40 # Really fuck shit up with one more cd and some insanity
  41 # this doesn't quite work as there are the prefetch gets lost
  42 # on search_related. This however is too esoteric to fix right
  43 # now
  44
  45 my $cd2 = $artist->create_related ('cds', {
  46     title => 'zzz',
  47     year => '1999',
  48     tracks => [{ title => 'ping' }, { title => 'pong' }],
  49 });
  50
  51 my $cds = $cd2->search_related ('artist', {}, { join => 'twokeys' })
  52                   ->search_related ('cds');
  53 my $tracks = $cds->search_related ('tracks');
  54
  55 is($tracks->count, 2, "2 Tracks counted on cd via artist via one of the cds");
  56 is(scalar($tracks->all), 2, "2 Track objects on cd via artist via one of the cds");
  57
  58 is($cds->count, 2, "2 CDs counted on artist via one of the cds");
  59 is(scalar($cds->all), 2, "2 CD objectson artist via one of the cds");
  60
  61 # make sure the join collapses all the way
  62 is_same_sql_bind (
  63   $tracks->count_rs->as_query,
  64   '(
  65     SELECT COUNT( * )
  66       FROM artist me
  67       LEFT JOIN twokeys twokeys ON twokeys.artist = me.artistid
  68       JOIN cd cds ON cds.artist = me.artistid
  69       JOIN track tracks ON tracks.cd = cds.cdid
  70     WHERE ( me.artistid = ? )
  71   )',
  72   [ [ { sqlt_datatype => 'integer', dbic_colname => 'me.artistid' }
  73       => 4 ] ],
  74 );
  75
  76 {
  77   local $TODO = "Chaining with prefetch is fundamentally broken";
  78   $schema->is_executed_querycount( sub {
  79
  80     my $cds = $cd2->search_related ('artist', {}, { prefetch => { cds => 'tracks' }, join => 'twokeys' })
  81                   ->search_related ('cds');
  82
  83     my $tracks = $cds->search_related ('tracks');
  84
  85     is($tracks->count, 2, "2 Tracks counted on cd via artist via one of the cds");
  86     is(scalar($tracks->all), 2, "2 Tracks prefetched on cd via artist via one of the cds");
  87     is($tracks->count, 2, "Cached 2 Tracks counted on cd via artist via one of the cds");
  88
  89     is($cds->count, 2, "2 CDs counted on artist via one of the cds");
  90     is(scalar($cds->all), 2, "2 CDs prefetched on artist via one of the cds");
  91     is($cds->count, 2, "Cached 2 CDs counted on artist via one of the cds");
  92   }, 3, '2 counts + 1 prefetch?' );
  93 }
  94
  95 done_testing;