a623b4b8f50c9394e337ed306c4799d0c74fd0fa
[dbsrgits/DBIx-Class-Historic.git] / t / relationship / custom.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 ':DiffSQL';
8
9 my $schema = DBICTest->init_schema();
10
11 $schema->resultset('Artist')->delete;
12 $schema->resultset('CD')->delete;
13
14 my $artist  = $schema->resultset("Artist")->create({ artistid => 21, name => 'Michael Jackson', rank => 20 });
15 my $artist2 = $schema->resultset("Artist")->create({ artistid => 22, name => 'Chico Buarque', rank => 1 }) ;
16 my $artist3 = $schema->resultset("Artist")->create({ artistid => 23, name => 'Ziraldo', rank => 1 });
17 my $artist4 = $schema->resultset("Artist")->create({ artistid => 24, name => 'Paulo Caruso', rank => 20 });
18
19 my @artworks;
20
21 foreach my $year (1975..1985) {
22   my $cd = $artist->create_related('cds', { year => $year, title => 'Compilation from ' . $year });
23   push @artworks, $cd->create_related('artwork', {});
24 }
25
26 foreach my $year (1975..1995) {
27   my $cd = $artist2->create_related('cds', { year => $year, title => 'Compilation from ' . $year });
28   push @artworks, $cd->create_related('artwork', {});
29 }
30
31 foreach my $artwork (@artworks) {
32   $artwork->create_related('artwork_to_artist', { artist => $_ }) for ($artist3, $artist4);
33 }
34
35
36 my $cds_80s_rs = $artist->cds_80s;
37 is_same_sql_bind(
38   $cds_80s_rs->as_query,
39   '(
40     SELECT me.cdid, me.artist, me.title, me.year, me.genreid, me.single_track
41       FROM cd me
42     WHERE ( ( me.artist = ? AND ( me.year < ? AND me.year > ? ) ) )
43   )',
44   [
45     [
46       {}
47         => 21
48     ],
49     [
50       { sqlt_datatype => 'varchar', sqlt_size => 100, dbic_colname => 'me.year' }
51         => 1990
52     ],
53     [
54       { sqlt_datatype => 'varchar', sqlt_size => 100, dbic_colname => 'me.year' }
55         => 1979
56     ],
57   ],
58 );
59 my @cds_80s = $cds_80s_rs->all;
60 is(@cds_80s, 6, '6 80s cds found (1980 - 1985)');
61 map { ok($_->year < 1990 && $_->year > 1979) } @cds_80s;
62
63
64 my $cds_90s_rs = $artist2->cds_90s;
65 is_same_sql_bind(
66   $cds_90s_rs->as_query,
67   '(
68     SELECT me.cdid, me.artist, me.title, me.year, me.genreid, me.single_track
69       FROM artist artist__row
70       JOIN cd me
71         ON ( me.artist = artist__row.artistid AND ( me.year < ? AND me.year > ? ) )
72       WHERE ( artist__row.artistid = ? )
73   )',
74   [
75     [
76       { sqlt_datatype => 'varchar', sqlt_size => 100, dbic_colname => 'me.year' }
77         => 2000
78     ],
79     [
80       { sqlt_datatype => 'varchar', sqlt_size => 100, dbic_colname => 'me.year' }
81         => 1989
82     ],
83     [ { sqlt_datatype => 'integer', dbic_colname => 'artist__row.artistid' }
84         => 22
85     ],
86   ]
87 );
88
89 # re-test with ::-containing moniker name
90 # (we don't have any currently, so fudge it with lots of local() )
91 {
92   local $schema->source('Artist')->{source_name} = 'Ar::Tist';
93   local $artist2->{related_resultsets};
94
95   is_same_sql_bind(
96     $artist2->cds_90s->as_query,
97     '(
98       SELECT me.cdid, me.artist, me.title, me.year, me.genreid, me.single_track
99         FROM artist ar_tist__row
100         JOIN cd me
101           ON ( me.artist = ar_tist__row.artistid AND ( me.year < ? AND me.year > ? ) )
102         WHERE ( ar_tist__row.artistid = ? )
103     )',
104     [
105       [
106         { sqlt_datatype => 'varchar', sqlt_size => 100, dbic_colname => 'me.year' }
107           => 2000
108       ],
109       [
110       { sqlt_datatype => 'varchar', sqlt_size => 100, dbic_colname => 'me.year' }
111           => 1989
112       ],
113       [ { sqlt_datatype => 'integer', dbic_colname => 'ar_tist__row.artistid' }
114           => 22
115       ],
116     ]
117   );
118 }
119
120
121 my @cds_90s = $cds_90s_rs->all;
122 is(@cds_90s, 6, '6 90s cds found (1990 - 1995) even with non-optimized search');
123 map { ok($_->year < 2000 && $_->year > 1989) } @cds_90s;
124
125 lives_ok {
126   my @cds_90s_95 = $artist2->cds_90s->search({ 'me.year' => 1995 });
127   is(@cds_90s_95, 1, '1 90s (95) cds found even with non-optimized search');
128   map { ok($_->year == 1995) } @cds_90s_95;
129 } 'should preserve chain-head "me" alias (API-consistency)';
130
131 # search for all artists prefetching published cds in the 80s...
132 my @all_artists_with_80_cds = $schema->resultset("Artist")->search
133   ({ 'cds_80s.cdid' => { '!=' => undef } }, { join => 'cds_80s', distinct => 1 });
134
135 is_deeply(
136   [ sort ( map { $_->year } map { $_->cds_80s->all } @all_artists_with_80_cds ) ],
137   [ sort (1980..1989, 1980..1985) ],
138   '16 correct cds found'
139 );
140
141 lives_ok {
142
143 my @all_artists_with_80_cds_pref = $schema->resultset("Artist")->search
144   ({ 'cds_80s.cdid' => { '!=' => undef } }, { prefetch => 'cds_80s' });
145
146 is_deeply(
147   [ sort ( map { $_->year } map { $_->cds_80s->all } @all_artists_with_80_cds_pref ) ],
148   [ sort (1980..1989, 1980..1985) ],
149   '16 correct cds found'
150 );
151
152 } 'prefetchy-fetchy-fetch';
153
154 # create_related a plain cd via the equoivalent coderef cond, with no extra conditions
155 lives_ok {
156   $artist->create_related('cds_cref_cond', { title => 'related creation via coderef cond', year => '2010' } );
157 } 'created_related with simple condition works';
158
159 # try to create_related a 80s cd
160 throws_ok {
161   $artist->create_related('cds_80s', { title => 'related creation 1' });
162 } qr/\QCustom relationship 'cds_80s' not definitive - returns conditions instead of values for column(s): 'year'/,
163 'Create failed - complex cond';
164
165 # now supply an explicit arg overwriting the ambiguous cond
166 my $cd_2020 = $artist->create_related('cds_80s', { title => 'related creation 2', year => '2020' });
167 my $id_2020 = $cd_2020->id;
168 is(
169   $schema->resultset('CD')->find($id_2020)->title,
170   'related creation 2',
171   '2020 CD created correctly'
172 );
173
174 # try a default year from a specific rel
175 my $id_1984 = $artist->create_related('cds_84', { title => 'related creation 3' })->id;
176 is(
177   $schema->resultset('CD')->find($id_1984)->title,
178   'related creation 3',
179   '1984 CD created correctly'
180 );
181
182 # try a specific everything via a non-simplified rel
183 throws_ok {
184   $artist->create_related('cds_90s', { title => 'related_creation 4', year => '2038' });
185 } qr/\QRelationship 'cds_90s' does not resolve to a join-free condition fragment/,
186 'Create failed - non-simplified rel';
187
188 # Do a self-join last-entry search
189 my @last_tracks;
190 for my $cd ($schema->resultset('CD')->search ({}, { order_by => 'cdid'})->all) {
191   push @last_tracks, $cd->tracks
192                          ->search ({}, { order_by => { -desc => 'position'} })
193                           ->next || ();
194 }
195
196 my $last_tracks_rs = $schema->resultset('Track')->search (
197   {'next_tracks.trackid' => undef},
198   { join => 'next_tracks', order_by => 'me.cd' },
199 );
200
201 is_deeply (
202   [$last_tracks_rs->get_column ('trackid')->all],
203   [ map { $_->trackid } @last_tracks ],
204   'last group-entry via self-join works',
205 );
206
207 is_deeply (
208   [map { $_->last_track->id } grep { $_->last_track } $schema->resultset('CD')->search ({}, { order_by => 'cdid', prefetch => 'last_track'})->all],
209   [ map { $_->trackid } @last_tracks ],
210   'last_track via insane subquery condition works',
211 );
212
213 is_deeply (
214   [map { $_->last_track->id } grep { $_->last_track } $schema->resultset('CD')->search ({}, { order_by => 'cdid'})->all],
215   [ map { $_->trackid } @last_tracks ],
216   'last_track via insane subquery condition works, even without prefetch',
217 );
218
219 my $artwork = $schema->resultset('Artwork')->search({},{ order_by => 'cd_id' })->first;
220 my @artists = $artwork->artists->all;
221 is(scalar @artists, 2, 'the two artists are associated');
222
223 my @artwork_artists = $artwork->artwork_to_artist->all;
224 foreach (@artwork_artists) {
225   lives_ok {
226     my $artista = $_->artist;
227     my $artistb = $_->artist_test_m2m;
228     ok($artista->rank < 10 ? $artistb : 1, 'belongs_to with custom rel works.');
229     my $artistc = $_->artist_test_m2m_noopt;
230     ok($artista->rank < 10 ? $artistc : 1, 'belongs_to with custom rel works even in non-simplified.');
231   } 'belongs_to works with custom rels';
232 }
233
234 @artists = ();
235 lives_ok {
236   @artists = $artwork->artists_test_m2m2->all;
237 } 'manytomany with extended rels in the has many works';
238 is(scalar @artists, 2, 'two artists');
239
240 @artists = ();
241 lives_ok {
242   @artists = $artwork->artists_test_m2m->all;
243 } 'can fetch many to many with optimized version';
244 is(scalar @artists, 1, 'only one artist is associated');
245
246 @artists = ();
247 lives_ok {
248   @artists = $artwork->artists_test_m2m_noopt->all;
249 } 'can fetch many to many with non-optimized version';
250 is(scalar @artists, 1, 'only one artist is associated');
251
252
253 # Make a single for each last_track
254 my @singles = map {
255   $_->create_related('cd_single', {
256     title => $_->title . ' (the single)',
257     artist => $artist,
258     year => 1999,
259   }) } @last_tracks
260 ;
261
262 # See if chaining works
263 is_deeply (
264   [ map { $_->title } $last_tracks_rs->search_related('cd_single')->all ],
265   [ map { $_->title } @singles ],
266   'Retrieved singles in proper order'
267 );
268
269 # See if prefetch works
270 is_deeply (
271   [ map { $_->cd_single->title } $last_tracks_rs->search({}, { prefetch => 'cd_single' })->all ],
272   [ map { $_->title } @singles ],
273   'Prefetched singles in proper order'
274 );
275
276 # test set_from_related with a belongs_to custom condition
277 my $cd = $schema->resultset("CD")->find(4);
278 $artist = $cd->search_related('artist');
279 my $track = $schema->resultset("Track")->create( {
280   trackid => 1,
281   cd => 3,
282   position => 99,
283   title => 'Some Track'
284 } );
285 $track->set_from_related( cd_cref_cond => $cd );
286 is ($track->get_column('cd'), 4, 'set from related via coderef cond');
287 is_deeply (
288   { $track->cd->get_columns },
289   { $cd->get_columns },
290   'set from related via coderef cond inflates properly',
291 );
292
293 done_testing;