t/prefetch/multiple_hasmany.t

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