t/relationship/custom.t

   1 BEGIN { do "./t/lib/ANFANG.pm" or die ( $@ || $! ) }
   2
   3 use strict;
   4 use warnings;
   5
   6 use Test::More;
   7 use Test::Exception;
   8 use Test::Warn;
   9
  10 use DBICTest ':DiffSQL';
  11
  12 my $schema = DBICTest->init_schema();
  13
  14 $schema->resultset('Artist')->delete;
  15 $schema->resultset('CD')->delete;
  16
  17 my $artist  = $schema->resultset("Artist")->create({ artistid => 21, name => 'Michael Jackson', rank => 20 });
  18 my $artist2 = $schema->resultset("Artist")->create({ artistid => 22, name => 'Chico Buarque', rank => 1 }) ;
  19 my $artist3 = $schema->resultset("Artist")->create({ artistid => 23, name => 'Ziraldo', rank => 1 });
  20 my $artist4 = $schema->resultset("Artist")->create({ artistid => 24, name => 'Paulo Caruso', rank => 20 });
  21
  22 my @artworks;
  23
  24 foreach my $year (1975..1985) {
  25   my $cd = $artist->create_related('cds', { year => $year, title => 'Compilation from ' . $year });
  26   push @artworks, $cd->create_related('artwork', {});
  27 }
  28
  29 foreach my $year (1975..1995) {
  30   my $cd = $artist2->create_related('cds', { year => $year, title => 'Compilation from ' . $year });
  31   push @artworks, $cd->create_related('artwork', {});
  32 }
  33
  34 foreach my $artwork (@artworks) {
  35   $artwork->create_related('artwork_to_artist', { artist => $_ }) for ($artist3, $artist4);
  36 }
  37
  38
  39 my $cds_80s_rs = $artist->cds_80s;
  40 is_same_sql_bind(
  41   $cds_80s_rs->as_query,
  42   '(
  43     SELECT me.cdid, me.artist, me.title, me.year, me.genreid, me.single_track
  44       FROM cd me
  45     WHERE ( ( me.artist = ? AND ( me.year < ? AND me.year > ? ) ) )
  46   )',
  47   [
  48     [
  49       {}
  50         => 21
  51     ],
  52     [
  53       { sqlt_datatype => 'varchar', sqlt_size => 100, dbic_colname => 'me.year' }
  54         => 1990
  55     ],
  56     [
  57       { sqlt_datatype => 'varchar', sqlt_size => 100, dbic_colname => 'me.year' }
  58         => 1979
  59     ],
  60   ],
  61 );
  62 my @cds_80s = $cds_80s_rs->all;
  63 is(@cds_80s, 6, '6 80s cds found (1980 - 1985)');
  64 map { ok($_->year < 1990 && $_->year > 1979) } @cds_80s;
  65
  66
  67 my $cds_90s_rs = $artist2->cds_90s;
  68 is_same_sql_bind(
  69   $cds_90s_rs->as_query,
  70   '(
  71     SELECT me.cdid, me.artist, me.title, me.year, me.genreid, me.single_track
  72       FROM artist artist__row
  73       JOIN cd me
  74         ON ( me.artist = artist__row.artistid AND ( me.year < ? AND me.year > ? ) )
  75       WHERE ( artist__row.artistid = ? )
  76   )',
  77   [
  78     [
  79       { sqlt_datatype => 'varchar', sqlt_size => 100, dbic_colname => 'me.year' }
  80         => 2000
  81     ],
  82     [
  83       { sqlt_datatype => 'varchar', sqlt_size => 100, dbic_colname => 'me.year' }
  84         => 1989
  85     ],
  86     [ { sqlt_datatype => 'integer', dbic_colname => 'artist__row.artistid' }
  87         => 22
  88     ],
  89   ]
  90 );
  91
  92 # re-test with ::-containing moniker name
  93 # (we don't have any currently, so fudge it with lots of local() )
  94 {
  95   local $schema->source('Artist')->{source_name} = 'Ar::Tist';
  96   local $artist2->{related_resultsets};
  97
  98   is_same_sql_bind(
  99     $artist2->cds_90s->as_query,
 100     '(
 101       SELECT me.cdid, me.artist, me.title, me.year, me.genreid, me.single_track
 102         FROM artist ar_tist__row
 103         JOIN cd me
 104           ON ( me.artist = ar_tist__row.artistid AND ( me.year < ? AND me.year > ? ) )
 105         WHERE ( ar_tist__row.artistid = ? )
 106     )',
 107     [
 108       [
 109         { sqlt_datatype => 'varchar', sqlt_size => 100, dbic_colname => 'me.year' }
 110           => 2000
 111       ],
 112       [
 113       { sqlt_datatype => 'varchar', sqlt_size => 100, dbic_colname => 'me.year' }
 114           => 1989
 115       ],
 116       [ { sqlt_datatype => 'integer', dbic_colname => 'ar_tist__row.artistid' }
 117           => 22
 118       ],
 119     ]
 120   );
 121 }
 122
 123
 124 my @cds_90s = $cds_90s_rs->all;
 125 is(@cds_90s, 6, '6 90s cds found (1990 - 1995) even with non-optimized search');
 126 map { ok($_->year < 2000 && $_->year > 1989) } @cds_90s;
 127
 128 lives_ok {
 129   my @cds_90s_95 = $artist2->cds_90s->search({ 'me.year' => 1995 });
 130   is(@cds_90s_95, 1, '1 90s (95) cds found even with non-optimized search');
 131   map { ok($_->year == 1995) } @cds_90s_95;
 132 } 'should preserve chain-head "me" alias (API-consistency)';
 133
 134 # search for all artists prefetching published cds in the 80s...
 135 my @all_artists_with_80_cds = $schema->resultset("Artist")->search
 136   ({ 'cds_80s.cdid' => { '!=' => undef } }, { join => 'cds_80s', distinct => 1 });
 137
 138 is_deeply(
 139   [ sort ( map { $_->year } map { $_->cds_80s->all } @all_artists_with_80_cds ) ],
 140   [ sort (1980..1989, 1980..1985) ],
 141   '16 correct cds found'
 142 );
 143
 144 lives_ok {
 145
 146 my @all_artists_with_80_cds_pref = $schema->resultset("Artist")->search
 147   ({ 'cds_80s.cdid' => { '!=' => undef } }, { prefetch => 'cds_80s' });
 148
 149 is_deeply(
 150   [ sort ( map { $_->year } map { $_->cds_80s->all } @all_artists_with_80_cds_pref ) ],
 151   [ sort (1980..1989, 1980..1985) ],
 152   '16 correct cds found'
 153 );
 154
 155 } 'prefetchy-fetchy-fetch';
 156
 157 # create_related a plain cd via the equoivalent coderef cond, with no extra conditions
 158 lives_ok {
 159   $artist->create_related('cds_cref_cond', { title => 'related creation via coderef cond', year => '2010' } );
 160 } 'created_related with simple condition works';
 161
 162 # try to create_related a 80s cd
 163 throws_ok {
 164   $artist->create_related('cds_80s', { title => 'related creation 1' });
 165 } qr/\QUnable to complete value inferrence - custom relationship 'cds_80s' on source 'Artist' returns conditions instead of values for column(s): 'year'/,
 166 'Create failed - complex cond';
 167
 168 # now supply an explicit arg overwriting the ambiguous cond
 169 my $cd_2020 = $artist->create_related('cds_80s', { title => 'related creation 2', year => '2020' });
 170 my $id_2020 = $cd_2020->id;
 171 is(
 172   $schema->resultset('CD')->find($id_2020)->title,
 173   'related creation 2',
 174   '2020 CD created correctly'
 175 );
 176
 177 # try a default year from a specific rel
 178 my $id_1984 = $artist->create_related('cds_84', { title => 'related creation 3' })->id;
 179 is(
 180   $schema->resultset('CD')->find($id_1984)->title,
 181   'related creation 3',
 182   '1984 CD created correctly'
 183 );
 184
 185 # try a specific everything via a non-simplified rel
 186 throws_ok {
 187   $artist->create_related('cds_90s', { title => 'related_creation 4', year => '2038' });
 188 } qr/\QRelationship 'cds_90s' on source 'Artist' does not resolve to a join-free condition fragment/,
 189 'Create failed - non-simplified rel';
 190
 191 # Do a self-join last-entry search
 192 my @last_tracks;
 193 for my $cd ($schema->resultset('CD')->search ({}, { order_by => 'cdid'})->all) {
 194   push @last_tracks, $cd->tracks
 195                          ->search ({}, { order_by => { -desc => 'position'} })
 196                           ->next || ();
 197 }
 198
 199 my $last_tracks_rs = $schema->resultset('Track')->search (
 200   {'next_tracks.trackid' => undef},
 201   { join => 'next_tracks', order_by => 'me.cd' },
 202 );
 203
 204 is_deeply (
 205   [$last_tracks_rs->get_column ('trackid')->all],
 206   [ map { $_->trackid } @last_tracks ],
 207   'last group-entry via self-join works',
 208 );
 209
 210 is_deeply (
 211   [map { $_->last_track->id } grep { $_->last_track } $schema->resultset('CD')->search ({}, { order_by => 'cdid', prefetch => 'last_track'})->all],
 212   [ map { $_->trackid } @last_tracks ],
 213   'last_track via insane subquery condition works',
 214 );
 215
 216 is_deeply (
 217   [map { $_->last_track->id } grep { $_->last_track } $schema->resultset('CD')->search ({}, { order_by => 'cdid'})->all],
 218   [ map { $_->trackid } @last_tracks ],
 219   'last_track via insane subquery condition works, even without prefetch',
 220 );
 221
 222
 223 my $artwork = $schema->resultset('Artwork')->search({},{ order_by => 'cd_id' })->first;
 224 my @artists = $artwork->artists->search({}, { order_by => 'artistid' } );
 225 is(scalar @artists, 2, 'the two artists are associated');
 226
 227 my @artwork_artists = $artwork->artwork_to_artist->all;
 228 foreach (@artwork_artists) {
 229   lives_ok {
 230     my $artista = $_->artist;
 231     my $artistb = $_->artist_limited_rank;
 232     ok($artista->rank < 10 ? $artistb : 1, 'belongs_to with custom rel works.');
 233     my $artistc = $_->artist_limited_rank_opaque;
 234     ok($artista->rank < 10 ? $artistc : 1, 'belongs_to with custom rel works even in non-simplified.');
 235   } 'belongs_to works with custom rels';
 236 }
 237
 238 is(
 239   $schema->resultset('Artwork')
 240           ->related_resultset( 'artwork_to_artist_via_opaque_customcond' )
 241            ->related_resultset( 'artist' )
 242             ->search({}, { collapse => 1 })
 243              ->count,
 244   2,
 245   'Custom rel works correctly',
 246 );
 247
 248 is (
 249   scalar $artwork->all_artists_via_opaque_customcond->all,
 250   2,
 251   'Expected two m2m associated artist objects via opaque costom cond'
 252 );
 253
 254 for (qw( artist_limited_rank artist_limited_rank_opaque )) {
 255   is(
 256     $schema->resultset('Artwork')
 257             ->related_resultset( 'artwork_to_artist_via_opaque_customcond' )
 258              ->related_resultset( $_ )
 259               ->search({}, { collapse => 1 })
 260                ->count,
 261     1,
 262     'Condition over double custom rel works correctly',
 263   );
 264
 265   is (
 266     scalar $artwork->$_->all,
 267     1,
 268     'Expected one m2m associated artist object via opaque custom cond + conditional far cond'
 269   );
 270
 271   cmp_ok(
 272     $artwork->${\"remove_from_$_"} ( $artists[1] ),
 273     '==',
 274     0,
 275     'deletion action reports 0'
 276   );
 277
 278   is (
 279     scalar $artwork->all_artists_via_opaque_customcond->all,
 280     2,
 281     'Indeed nothing was removed'
 282   );
 283
 284   cmp_ok(
 285     $artwork->${\"remove_from_$_"} ( $artists[0] ),
 286     '==',
 287     1,
 288     'Removal reports correct count'
 289   );
 290
 291   is (
 292     scalar $artwork->all_artists_via_opaque_customcond->all,
 293     1,
 294     'Indeed removed the matching artist'
 295   );
 296
 297   $artwork->${\"set_$_"}([]);
 298
 299   is (
 300     scalar $artwork->all_artists_via_opaque_customcond->all,
 301     0,
 302     'Everything removed via limited far cond'
 303   );
 304
 305   # can't use the opaque one - need set_from_related to work
 306   $artwork->set_artist_limited_rank( \@artists );
 307
 308   {
 309     local $TODO = 'Taking into account the relationship bridge condition is not likely to ever work... unless we get DQ hooked somehow';
 310
 311     is (
 312       scalar $artwork->all_artists_via_opaque_customcond->all,
 313       1,
 314       'Re-Addition passed through only one of the artists'
 315     );
 316   }
 317
 318   throws_ok { $artwork->set_all_artists_via_opaque_customcond( \@artists ) }
 319     qr/\QRelationship 'artwork_to_artist_via_opaque_customcond' on source 'Artwork' does not resolve to a join-free condition fragment/;
 320
 321   is (
 322     scalar $artwork->all_artists_via_opaque_customcond->all,
 323     2,
 324     'Everything still there as expected'
 325   );
 326 }
 327
 328
 329 # Make a single for each last_track
 330 my @singles = map {
 331   $_->create_related('cd_single', {
 332     title => $_->title . ' (the single)',
 333     artist => $artist,
 334     year => 1999,
 335   }) } @last_tracks
 336 ;
 337
 338 # See if chaining works
 339 is_deeply (
 340   [ map { $_->title } $last_tracks_rs->search_related('cd_single')->all ],
 341   [ map { $_->title } @singles ],
 342   'Retrieved singles in proper order'
 343 );
 344
 345 # See if prefetch works
 346 is_deeply (
 347   [ map { $_->cd_single->title } $last_tracks_rs->search({}, { prefetch => 'cd_single' })->all ],
 348   [ map { $_->title } @singles ],
 349   'Prefetched singles in proper order'
 350 );
 351
 352 # test set_from_related/find_related with a belongs_to custom condition
 353 my $preexisting_cd = $schema->resultset('CD')->find(1);
 354
 355 my $cd_single_track = $schema->resultset('CD')->create({
 356   artist => $artist,
 357   title => 'one one one',
 358   year => 2001,
 359   tracks => [{ title => 'uno uno uno' }]
 360 });
 361
 362 my $single_track = $cd_single_track->tracks->next;
 363
 364 is(
 365   $single_track->cd_cref_cond->title,
 366   $cd_single_track->title,
 367   'Got back the expected single-track cd title',
 368 );
 369
 370 is_deeply
 371   { $schema->resultset('Track')->find({ cd_cref_cond => { cdid => $cd_single_track->id } })->get_columns },
 372   { $single_track->get_columns },
 373   'Proper find with related via coderef cond',
 374 ;
 375
 376 warnings_exist {
 377   is_same_sql_bind(
 378     $single_track->deliberately_broken_all_cd_tracks->as_query,
 379     '(
 380       SELECT me.trackid, me.cd, me.position, me.title, me.last_updated_on, me.last_updated_at
 381         FROM track track__row
 382         JOIN track me
 383           ON me.cd = ?
 384       WHERE track__row.trackid = ?
 385     )',
 386     [
 387       [{ dbic_colname => "me.cd", sqlt_datatype => "integer" }
 388         => "track__row.cd" ],
 389       [{ dbic_colname => "track__row.trackid", sqlt_datatype => "integer" }
 390         => 19 ],
 391     ],
 392     'Expected nonsensical JOIN cond',
 393   ),
 394 } qr/\Qrelationship 'deliberately_broken_all_cd_tracks' on source 'Track' specifies equality of column 'cd' and the *VALUE* 'cd' (you did not use the { -ident => ... } operator)/,
 395   'Warning on 99.9999% malformed custom cond'
 396 ;
 397
 398 $single_track->set_from_related( cd_cref_cond => undef );
 399 ok $single_track->is_column_changed('cd');
 400 is $single_track->get_column('cd'), undef, 'UNset from related via coderef cond';
 401 is $single_track->cd, undef, 'UNset related object via coderef cond';
 402
 403 $single_track->discard_changes;
 404
 405 $single_track->set_from_related( cd_cref_cond => $preexisting_cd );
 406 ok $single_track->is_column_changed('cd');
 407 is $single_track->get_column('cd'), 1, 'set from related via coderef cond';
 408 is_deeply
 409   { $single_track->cd->get_columns },
 410   { $preexisting_cd->get_columns },
 411   'set from related via coderef cond inflates properly',
 412 ;
 413
 414 throws_ok {
 415   local $schema->source('Track')->relationship_info('cd_cref_cond')->{cond} = sub { 1,2,3 };
 416   $schema->resultset('Track')->find({ cd_cref_cond => {} });
 417 } qr/\QA custom condition coderef can return at most 2 conditions, but relationship 'cd_cref_cond' on source 'Track' returned extra values: 3/;
 418
 419 done_testing;