t/prefetch/multiple_hasmany.t

   1 use strict;
   2 use warnings;
   3
   4 use Test::More;
   5 use lib qw(t/lib);
   6 use DBICTest;
   7
   8 my $schema = DBICTest->init_schema();
   9 my $sdebug = $schema->storage->debug;
  10
  11 # once the following TODO is complete, remove the 2 warning tests immediately
  12 # after the TODO block
  13 # (the TODO block itself contains tests ensuring that the warns are removed)
  14 TODO: {
  15     local $TODO = 'Prefetch of multiple has_many rels at the same level (currently warn to protect the clueless git)';
  16
  17     #( 1 -> M + M )
  18     my $cd_rs = $schema->resultset('CD')->search ({ 'me.title' => 'Forkful of bees' });
  19     my $pr_cd_rs = $cd_rs->search ({}, {
  20         prefetch => [qw/tracks tags/],
  21     });
  22
  23     my $tracks_rs = $cd_rs->first->tracks;
  24     my $tracks_count = $tracks_rs->count;
  25
  26     my ($pr_tracks_rs, $pr_tracks_count);
  27
  28     my $queries = 0;
  29     $schema->storage->debugcb(sub { $queries++ });
  30     $schema->storage->debug(1);
  31
  32     my $o_mm_warn;
  33     {
  34         local $SIG{__WARN__} = sub { $o_mm_warn = shift };
  35         $pr_tracks_rs = $pr_cd_rs->first->tracks;
  36     };
  37     $pr_tracks_count = $pr_tracks_rs->count;
  38
  39     ok(! $o_mm_warn, 'no warning on attempt to prefetch several same level has_many\'s (1 -> M + M)');
  40
  41   {
  42     local $TODO;
  43     is($queries, 1, 'prefetch one->(has_many,has_many) ran exactly 1 query');
  44   }
  45     $schema->storage->debugcb (undef);
  46     $schema->storage->debug ($sdebug);
  47
  48     is($pr_tracks_count, $tracks_count, 'equal count of prefetched relations over several same level has_many\'s (1 -> M + M)');
  49     is ($pr_tracks_rs->all, $tracks_rs->all, 'equal amount of objects returned with and without prefetch over several same level has_many\'s (1 -> M + M)');
  50
  51     #( M -> 1 -> M + M )
  52     my $note_rs = $schema->resultset('LinerNotes')->search ({ notes => 'Buy Whiskey!' });
  53     my $pr_note_rs = $note_rs->search ({}, {
  54         prefetch => {
  55             cd => [qw/tracks tags/]
  56         },
  57     });
  58
  59     my $tags_rs = $note_rs->first->cd->tags;
  60     my $tags_count = $tags_rs->count;
  61
  62     my ($pr_tags_rs, $pr_tags_count);
  63
  64     $queries = 0;
  65     $schema->storage->debugcb(sub { $queries++ });
  66     $schema->storage->debug(1);
  67
  68     my $m_o_mm_warn;
  69     {
  70         local $SIG{__WARN__} = sub { $m_o_mm_warn = shift };
  71         $pr_tags_rs = $pr_note_rs->first->cd->tags;
  72     };
  73     $pr_tags_count = $pr_tags_rs->count;
  74
  75     ok(! $m_o_mm_warn, 'no warning on attempt to prefetch several same level has_many\'s (M -> 1 -> M + M)');
  76
  77   {
  78     local $TODO;
  79
  80     is($queries, 1, 'prefetch one->(has_many,has_many) ran exactly 1 query');
  81     $schema->storage->debugcb (undef);
  82     $schema->storage->debug ($sdebug);
  83
  84     is($pr_tags_count, $tags_count, 'equal count of prefetched relations over several same level has_many\'s (M -> 1 -> M + M)');
  85     is($pr_tags_rs->all, $tags_rs->all, 'equal amount of objects with and without prefetch over several same level has_many\'s (M -> 1 -> M + M)');
  86   }
  87 }
  88
  89 # remove this closure once the TODO above is working
  90 {
  91     my $warn_re = qr/will explode the number of row objects retrievable via/;
  92
  93     my (@w, @dummy);
  94     local $SIG{__WARN__} = sub { $_[0] =~ $warn_re ? push @w, @_ : warn @_ };
  95
  96     my $rs = $schema->resultset('CD')->search ({ 'me.title' => 'Forkful of bees' }, { prefetch => [qw/tracks tags/] });
  97     @w = ();
  98     @dummy = $rs->first;
  99     is (@w, 1, 'warning on attempt prefetching several same level has_manys (1 -> M + M)');
 100
 101     my $rs2 = $schema->resultset('LinerNotes')->search ({ notes => 'Buy Whiskey!' }, { prefetch => { cd => [qw/tags tracks/] } });
 102     @w = ();
 103     @dummy = $rs2->first;
 104     is (@w, 1, 'warning on attempt prefetching several same level has_manys (M -> 1 -> M + M)');
 105 }
 106
 107 done_testing;