Make POD a little more obvious about component order
[dbsrgits/DBIx-Class.git] / t / 66relationship.t
CommitLineData
70350518 1use strict;
2use warnings;
3
4b8dcc58 4use Test::More;
5use lib qw(t/lib);
6use DBICTest;
7
a47e1233 8my $schema = DBICTest->init_schema();
5efe4c79 9
21c45f7b 10plan tests => 56;
0567538f 11
12# has_a test
f9db5527 13my $cd = $schema->resultset("CD")->find(4);
07037f89 14my ($artist) = ($INC{'DBICTest/HelperRels'}
15 ? $cd->artist
16 : $cd->search_related('artist'));
0567538f 17is($artist->name, 'Random Boy Band', 'has_a search_related ok');
18
19# has_many test with an order_by clause defined
f9db5527 20$artist = $schema->resultset("Artist")->find(1);
07037f89 21my @cds = ($INC{'DBICTest/HelperRels'}
22 ? $artist->cds
23 : $artist->search_related('cds'));
24is( $cds[1]->title, 'Spoonful of bees', 'has_many search_related with order_by ok' );
0567538f 25
26# search_related with additional abstract query
07037f89 27@cds = ($INC{'DBICTest/HelperRels'}
28 ? $artist->cds({ title => { like => '%of%' } })
29 : $artist->search_related('cds', { title => { like => '%of%' } } )
30 );
0567538f 31is( $cds[1]->title, 'Forkful of bees', 'search_related with abstract query ok' );
32
33# creating a related object
07037f89 34if ($INC{'DBICTest/HelperRels.pm'}) {
35 $artist->add_to_cds({ title => 'Big Flop', year => 2005 });
36} else {
37 $artist->create_related( 'cds', {
38 title => 'Big Flop',
39 year => 2005,
40 } );
41}
42
0567538f 43is( ($artist->search_related('cds'))[3]->title, 'Big Flop', 'create_related ok' );
44
5b89a768 45my( $rs_from_list ) = $artist->search_related_rs('cds');
46is( ref($rs_from_list), 'DBIx::Class::ResultSet', 'search_related_rs in list context returns rs' );
47
48( $rs_from_list ) = $artist->cds_rs();
49is( ref($rs_from_list), 'DBIx::Class::ResultSet', 'relation_rs in list context returns rs' );
50
0567538f 51# count_related
52is( $artist->count_related('cds'), 4, 'count_related ok' );
53
54# set_from_related
f9db5527 55my $track = $schema->resultset("Track")->create( {
0567538f 56 trackid => 1,
57 cd => 3,
58 position => 98,
59 title => 'Hidden Track'
60} );
61$track->set_from_related( cd => $cd );
62
70350518 63is($track->disc->cdid, 4, 'set_from_related ok, including alternative accessor' );
0567538f 64
2c037e6b 65$track->set_from_related( cd => undef );
66
67ok( !defined($track->cd), 'set_from_related with undef ok');
68
69
0567538f 70# update_from_related, the same as set_from_related, but it calls update afterwards
f9db5527 71$track = $schema->resultset("Track")->create( {
0567538f 72 trackid => 2,
73 cd => 3,
74 position => 99,
365d06b7 75 title => 'Hidden Track 2'
0567538f 76} );
77$track->update_from_related( cd => $cd );
78
f9db5527 79my $t_cd = ($schema->resultset("Track")->search( cd => 4, position => 99 ))[0]->cd;
0567538f 80
70350518 81is( $t_cd->cdid, 4, 'update_from_related ok' );
0567538f 82
83# find_or_create_related with an existing record
84$cd = $artist->find_or_create_related( 'cds', { title => 'Big Flop' } );
85is( $cd->year, 2005, 'find_or_create_related on existing record ok' );
86
87# find_or_create_related creating a new record
88$cd = $artist->find_or_create_related( 'cds', {
89 title => 'Greatest Hits',
90 year => 2006,
91} );
92is( $cd->title, 'Greatest Hits', 'find_or_create_related new record ok' );
93@cds = $artist->search_related('cds');
94is( ($artist->search_related('cds'))[4]->title, 'Greatest Hits', 'find_or_create_related new record search ok' );
95
87772e46 96$artist->delete_related( cds => { title => 'Greatest Hits' });
f9db5527 97cmp_ok( $schema->resultset("CD")->search( title => 'Greatest Hits' ), '==', 0, 'delete_related ok' );
0567538f 98
b3e1f1f5 99# find_or_new_related with an existing record
100$cd = $artist->find_or_new_related( 'cds', { title => 'Big Flop' } );
101is( $cd->year, 2005, 'find_or_new_related on existing record ok' );
102ok( $cd->in_storage, 'find_or_new_related on existing record: is in_storage' );
103
104# find_or_new_related instantiating a new record
105$cd = $artist->find_or_new_related( 'cds', {
106 title => 'Greatest Hits 2: Louder Than Ever',
107 year => 2007,
108} );
109is( $cd->title, 'Greatest Hits 2: Louder Than Ever', 'find_or_new_related new record ok' );
110ok( ! $cd->in_storage, 'find_or_new_related on a new record: not in_storage' );
111
87772e46 112SKIP: {
113 skip "relationship checking needs fixing", 1;
114 # try to add a bogus relationship using the wrong cols
115 eval {
116 DBICTest::Schema::Artist->add_relationship(
117 tracks => 'DBICTest::Schema::Track',
118 { 'foreign.cd' => 'self.cdid' }
119 );
120 };
121 like($@, qr/Unknown column/, 'failed when creating a rel with invalid key, ok');
122}
123
0567538f 124# another bogus relationship using no join condition
125eval {
3712e4f4 126 DBICTest::Schema::Artist->add_relationship( tracks => 'DBICTest::Track' );
0567538f 127};
128like($@, qr/join condition/, 'failed when creating a rel without join condition, ok');
129
b3f358b5 130# many_to_many helper tests
f9db5527 131$cd = $schema->resultset("CD")->find(1);
7411204b 132my @producers = $cd->producers();
133is( $producers[0]->name, 'Matt S Trout', 'many_to_many ok' );
b3f358b5 134is( $cd->producers_sorted->next->name, 'Bob The Builder',
135 'sorted many_to_many ok' );
136is( $cd->producers_sorted(producerid => 3)->next->name, 'Fred The Phenotype',
137 'sorted many_to_many with search condition ok' );
7411204b 138
3a868fb2 139$cd = $schema->resultset('CD')->find(2);
b3f358b5 140my $prod_rs = $cd->producers();
141my $prod_before_count = $schema->resultset('Producer')->count;
142is( $prod_rs->count, 0, "CD doesn't yet have any producers" );
3a868fb2 143my $prod = $schema->resultset('Producer')->find(1);
144$cd->add_to_producers($prod);
3a868fb2 145is( $prod_rs->count(), 1, 'many_to_many add_to_$rel($obj) count ok' );
b3f358b5 146is( $prod_rs->first->name, 'Matt S Trout',
147 'many_to_many add_to_$rel($obj) ok' );
3a868fb2 148$cd->remove_from_producers($prod);
b3f358b5 149is( $schema->resultset('Producer')->find(1)->name, 'Matt S Trout',
150 "producer object exists after remove of link" );
151is( $prod_rs->count, 0, 'many_to_many remove_from_$rel($obj) ok' );
303cf522 152$cd->add_to_producers({ name => 'Testy McProducer' });
b3f358b5 153is( $schema->resultset('Producer')->count, $prod_before_count+1,
154 'add_to_$rel($hash) inserted a new producer' );
303cf522 155is( $prod_rs->count(), 1, 'many_to_many add_to_$rel($hash) count ok' );
b3f358b5 156is( $prod_rs->first->name, 'Testy McProducer',
157 'many_to_many add_to_$rel($hash) ok' );
158$cd->add_to_producers({ name => 'Jack Black' });
159is( $prod_rs->count(), 2, 'many_to_many add_to_$rel($hash) count ok' );
160$cd->set_producers($schema->resultset('Producer')->all);
161is( $cd->producers->count(), $prod_before_count+2,
162 'many_to_many set_$rel(@objs) count ok' );
163$cd->set_producers($schema->resultset('Producer')->find(1));
164is( $cd->producers->count(), 1, 'many_to_many set_$rel($obj) count ok' );
21c45f7b 165$cd->set_producers([$schema->resultset('Producer')->all]);
166is( $cd->producers->count(), $prod_before_count+2,
167 'many_to_many set_$rel(\@objs) count ok' );
168$cd->set_producers([$schema->resultset('Producer')->find(1)]);
169is( $cd->producers->count(), 1, 'many_to_many set_$rel([$obj]) count ok' );
303cf522 170
171eval { $cd->remove_from_producers({ fake => 'hash' }); };
172like( $@, qr/needs an object/, 'remove_from_$rel($hash) dies correctly' );
173
174eval { $cd->add_to_producers(); };
b3f358b5 175like( $@, qr/needs an object or hashref/,
176 'add_to_$rel(undef) dies correctly' );
303cf522 177
3bd6e3e0 178# many_to_many stresstest
179my $twokey = $schema->resultset('TwoKeys')->find(1,1);
180my $fourkey = $schema->resultset('FourKeys')->find(1,2,3,4);
181
182is( $twokey->fourkeys->count, 0, 'twokey has no fourkeys' );
183$twokey->add_to_fourkeys($fourkey, { autopilot => 'engaged' });
184my $got_fourkey = $twokey->fourkeys({ sensors => 'online' })->first;
185is( $twokey->fourkeys->count, 1, 'twokey has one fourkey' );
186is( $got_fourkey->$_, $fourkey->$_,
187 'fourkeys row has the correct value for column '.$_ )
188 for (qw(foo bar hello goodbye sensors));
189$twokey->remove_from_fourkeys($fourkey);
190is( $twokey->fourkeys->count, 0, 'twokey has no fourkeys' );
b3f358b5 191is( $twokey->fourkeys_to_twokeys->count, 0,
192 'twokey has no links to fourkey' );
193
3a868fb2 194
5efe4c79 195# test undirected many-to-many relationship (e.g. "related artists")
f9db5527 196my $undir_maps = $schema->resultset("Artist")->find(1)->artist_undirected_maps;
5efe4c79 197is($undir_maps->count, 1, 'found 1 undirected map for artist 1');
198
f9db5527 199$undir_maps = $schema->resultset("Artist")->find(2)->artist_undirected_maps;
5efe4c79 200is($undir_maps->count, 1, 'found 1 undirected map for artist 2');
201
ad3d2d7c 202my $mapped_rs = $undir_maps->search_related('mapped_artists');
203
204my @art = $mapped_rs->all;
5efe4c79 205
206cmp_ok(@art, '==', 2, "Both artist returned from map");
207
ad3d2d7c 208my $searched = $mapped_rs->search({'mapped_artists.artistid' => {'!=', undef}});
209
210cmp_ok($searched->count, '==', 2, "Both artist returned from map after adding another condition");
211
b8d4bd90 212# check join through cascaded has_many relationships
213$artist = $schema->resultset("Artist")->find(1);
214my $trackset = $artist->cds->search_related('tracks');
215# LEFT join means we also see the trackless additional album...
216cmp_ok($trackset->count, '==', 11, "Correct number of tracks for artist");
ad3d2d7c 217
0f57d214 218# now see about updating eveything that belongs to artist 2 to artist 3
219$artist = $schema->resultset("Artist")->find(2);
220my $nartist = $schema->resultset("Artist")->find(3);
221cmp_ok($artist->cds->count, '==', 1, "Correct orig #cds for artist");
222cmp_ok($nartist->cds->count, '==', 1, "Correct orig #cds for artist");
223$artist->cds->update({artist => $nartist->id});
224cmp_ok($artist->cds->count, '==', 0, "Correct new #cds for artist");
225cmp_ok($nartist->cds->count, '==', 2, "Correct new #cds for artist");
226