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