added source_info method for source-level metadata (kinda like column_info)
[dbsrgits/DBIx-Class.git] / t / 60core.t
1 use strict;
2 use warnings;  
3
4 use Test::More;
5 use lib qw(t/lib);
6 use DBICTest;
7
8 my $schema = DBICTest->init_schema();
9
10 plan tests => 64;
11
12 # figure out if we've got a version of sqlite that is older than 3.2.6, in
13 # which case COUNT(DISTINCT()) doesn't work
14 my $is_broken_sqlite = 0;
15 my ($sqlite_major_ver,$sqlite_minor_ver,$sqlite_patch_ver) =
16     split /\./, $schema->storage->dbh->get_info(18);
17 if( $schema->storage->dbh->get_info(17) eq 'SQLite' &&
18     ( ($sqlite_major_ver < 3) ||
19       ($sqlite_major_ver == 3 && $sqlite_minor_ver < 2) ||
20       ($sqlite_major_ver == 3 && $sqlite_minor_ver == 2 && $sqlite_patch_ver < 6) ) ) {
21     $is_broken_sqlite = 1;
22 }
23
24
25 my @art = $schema->resultset("Artist")->search({ }, { order_by => 'name DESC'});
26
27 cmp_ok(@art, '==', 3, "Three artists returned");
28
29 my $art = $art[0];
30
31 is($art->name, 'We Are Goth', "Correct order too");
32
33 $art->name('We Are In Rehab');
34
35 is($art->name, 'We Are In Rehab', "Accessor update ok");
36
37 is($art->get_column("name"), 'We Are In Rehab', 'And via get_column');
38
39 ok($art->update, 'Update run');
40
41 my $record_jp = $schema->resultset("Artist")->search(undef, { join => 'cds' })->search(undef, { prefetch => 'cds' })->next;
42
43 ok($record_jp, "prefetch on same rel okay");
44
45 my $record_fn = $schema->resultset("Artist")->search(undef, { join => 'cds' })->search({'cds.cdid' => '1'}, {join => 'artist_undirected_maps'})->next;
46
47 ok($record_fn, "funny join is okay");
48
49 @art = $schema->resultset("Artist")->search({ name => 'We Are In Rehab' });
50
51 cmp_ok(@art, '==', 1, "Changed artist returned by search");
52
53 cmp_ok($art[0]->artistid, '==', 3,'Correct artist too');
54
55 $art->delete;
56
57 @art = $schema->resultset("Artist")->search({ });
58
59 cmp_ok(@art, '==', 2, 'And then there were two');
60
61 ok(!$art->in_storage, "It knows it's dead");
62
63 eval { $art->delete; };
64
65 ok($@, "Can't delete twice: $@");
66
67 is($art->name, 'We Are In Rehab', 'But the object is still live');
68
69 $art->insert;
70
71 ok($art->in_storage, "Re-created");
72
73 @art = $schema->resultset("Artist")->search({ });
74
75 cmp_ok(@art, '==', 3, 'And now there are three again');
76
77 my $new = $schema->resultset("Artist")->create({ artistid => 4 });
78
79 cmp_ok($new->artistid, '==', 4, 'Create produced record ok');
80
81 @art = $schema->resultset("Artist")->search({ });
82
83 cmp_ok(@art, '==', 4, "Oh my god! There's four of them!");
84
85 $new->set_column('name' => 'Man With A Fork');
86
87 is($new->name, 'Man With A Fork', 'set_column ok');
88
89 $new->discard_changes;
90
91 ok(!defined $new->name, 'Discard ok');
92
93 $new->name('Man With A Spoon');
94
95 $new->update;
96
97 my $new_again = $schema->resultset("Artist")->find(4);
98
99 is($new_again->name, 'Man With A Spoon', 'Retrieved correctly');
100
101 is($new_again->ID, 'DBICTest::Artist|artist|artistid=4', 'unique object id generated correctly');
102
103 # Test backwards compatibility
104 {
105   my $warnings = '';
106   local $SIG{__WARN__} = sub { $warnings .= $_[0] };
107
108   my $artist_by_hash = $schema->resultset('Artist')->find(artistid => 4);
109   is($artist_by_hash->name, 'Man With A Spoon', 'Retrieved correctly');
110   is($artist_by_hash->ID, 'DBICTest::Artist|artist|artistid=4', 'unique object id generated correctly');
111   like($warnings, qr/deprecated/, 'warned about deprecated find usage');
112 }
113
114 is($schema->resultset("Artist")->count, 4, 'count ok');
115
116 # test find_or_new
117 {
118   my $existing_obj = $schema->resultset('Artist')->find_or_new({
119     artistid => 4,
120   });
121
122   is($existing_obj->name, 'Man With A Spoon', 'find_or_new: found existing artist');
123   ok($existing_obj->in_storage, 'existing artist is in storage');
124
125   my $new_obj = $schema->resultset('Artist')->find_or_new({
126     artistid => 5,
127     name     => 'find_or_new',
128   });
129
130   is($new_obj->name, 'find_or_new', 'find_or_new: instantiated a new artist');
131   ok(! $new_obj->in_storage, 'new artist is not in storage');
132 }
133
134 my $cd = $schema->resultset("CD")->find(1);
135 my %cols = $cd->get_columns;
136
137 cmp_ok(keys %cols, '==', 4, 'get_columns number of columns ok');
138
139 is($cols{title}, 'Spoonful of bees', 'get_columns values ok');
140
141 %cols = ( title => 'Forkful of bees', year => 2005);
142 $cd->set_columns(\%cols);
143
144 is($cd->title, 'Forkful of bees', 'set_columns ok');
145
146 is($cd->year, 2005, 'set_columns ok');
147
148 $cd->discard_changes;
149
150 # check whether ResultSource->columns returns columns in order originally supplied
151 my @cd = $schema->source("CD")->columns;
152
153 is_deeply( \@cd, [qw/cdid artist title year/], 'column order');
154
155 $cd = $schema->resultset("CD")->search({ title => 'Spoonful of bees' }, { columns => ['title'] })->next;
156 is($cd->title, 'Spoonful of bees', 'subset of columns returned correctly');
157
158 $cd = $schema->resultset("CD")->search(undef, { include_columns => [ 'artist.name' ], join => [ 'artist' ] })->find(1);
159
160 is($cd->title, 'Spoonful of bees', 'Correct CD returned with include');
161 is($cd->get_column('name'), 'Caterwauler McCrae', 'Additional column returned');
162
163 # update_or_insert
164 $new = $schema->resultset("Track")->new( {
165   trackid => 100,
166   cd => 1,
167   position => 4,
168   title => 'Insert or Update',
169 } );
170 $new->update_or_insert;
171 ok($new->in_storage, 'update_or_insert insert ok');
172
173 # test in update mode
174 $new->pos(5);
175 $new->update_or_insert;
176 is( $schema->resultset("Track")->find(100)->pos, 5, 'update_or_insert update ok');
177
178 eval { $schema->class("Track")->load_components('DoesNotExist'); };
179
180 ok $@, $@;
181
182 is($schema->class("Artist")->field_name_for->{name}, 'artist name', 'mk_classdata usage ok');
183
184 my $search = [ { 'tags.tag' => 'Cheesy' }, { 'tags.tag' => 'Blue' } ];
185
186 my( $or_rs ) = $schema->resultset("CD")->search_rs($search, { join => 'tags',
187                                                   order_by => 'cdid' });
188
189 cmp_ok($or_rs->count, '==', 5, 'Search with OR ok');
190
191 my $distinct_rs = $schema->resultset("CD")->search($search, { join => 'tags', distinct => 1 });
192 cmp_ok($distinct_rs->all, '==', 4, 'DISTINCT search with OR ok');
193
194 SKIP: {
195   skip "SQLite < 3.2.6 doesn't understand COUNT(DISTINCT())", 1
196     if $is_broken_sqlite;
197
198   my $tcount = $schema->resultset("Track")->search(
199     {},
200     {       
201        select => {count => {distinct => ['position', 'title']}},
202            as => ['count']
203     }
204   );
205   cmp_ok($tcount->next->get_column('count'), '==', 13, 'multiple column COUNT DISTINCT ok');
206
207 }
208 my $tag_rs = $schema->resultset('Tag')->search(
209                [ { 'me.tag' => 'Cheesy' }, { 'me.tag' => 'Blue' } ]);
210
211 my $rel_rs = $tag_rs->search_related('cd');
212
213 cmp_ok($rel_rs->count, '==', 5, 'Related search ok');
214
215 cmp_ok($or_rs->next->cdid, '==', $rel_rs->next->cdid, 'Related object ok');
216 $or_rs->reset;
217 $rel_rs->reset;
218
219 my $tag = $schema->resultset('Tag')->search(
220                [ { 'me.tag' => 'Blue' } ], { cols=>[qw/tagid/] } )->next;
221
222 cmp_ok($tag->has_column_loaded('tagid'), '==', 1, 'Has tagid loaded');
223 cmp_ok($tag->has_column_loaded('tag'), '==', 0, 'Has not tag  loaded');
224
225 ok($schema->storage(), 'Storage available');
226
227 {
228   my $rs = $schema->resultset("Artist")->search({
229     -and => [
230       artistid => { '>=', 1 },
231       artistid => { '<', 3 }
232     ]
233   });
234
235   $rs->update({ name => 'Test _cond_for_update_delete' });
236
237   my $art;
238
239   $art = $schema->resultset("Artist")->find(1);
240   is($art->name, 'Test _cond_for_update_delete', 'updated first artist name');
241
242   $art = $schema->resultset("Artist")->find(2);
243   is($art->name, 'Test _cond_for_update_delete', 'updated second artist name');
244 }
245
246 # test source_name
247 {
248   # source_name should be set for normal modules
249   is($schema->source('CD')->source_name, 'CD', 'source_name is set to moniker');
250
251   # test the result source that sets source_name explictly
252   ok($schema->source('SourceNameArtists'), 'SourceNameArtists result source exists');
253
254   my @artsn = $schema->resultset('SourceNameArtists')->search({}, { order_by => 'name DESC' });
255   cmp_ok(@artsn, '==', 4, "Four artists returned");
256   
257   # make sure subclasses that don't set source_name are ok
258   ok($schema->source('ArtistSubclass', 'ArtistSubclass exists'));
259 }
260
261 my $newbook = $schema->resultset( 'Bookmark' )->find(1);
262
263 $@ = '';
264 eval {
265 my $newlink = $newbook->link;
266 };
267 ok(!$@, "stringify to false value doesn't cause error");
268
269 # test cascade_delete through many_to_many relations
270 {
271   my $art_del = $schema->resultset("Artist")->find({ artistid => 1 });
272   $art_del->delete;
273   cmp_ok( $schema->resultset("CD")->search({artist => 1}), '==', 0, 'Cascading through has_many top level.');
274   cmp_ok( $schema->resultset("CD_to_Producer")->search({cd => 1}), '==', 0, 'Cascading through has_many children.');
275 }
276
277 # test column_info
278 {
279   $schema->source("Artist")->{_columns}{'artistid'} = {};
280   $schema->source("Artist")->load_column_info_from_storage;
281
282   my $typeinfo = $schema->source("Artist")->column_info('artistid');
283   is($typeinfo->{data_type}, 'INTEGER', 'column_info ok');
284   $schema->source("Artist")->column_info('artistid');
285   ok($schema->source("Artist")->{_columns_info_loaded} == 1, 'Columns info flag set');
286 }
287
288 # test source_info
289 {
290   my $expected = {
291     "source_info_key_A" => "source_info_value_A",
292     "source_info_key_B" => "source_info_value_B",
293     "source_info_key_C" => "source_info_value_C",
294   };
295
296   my $sinfo = $schema->source("Artist")->source_info;
297
298   is_deeply($sinfo, $expected, 'source_info data works');
299 }
300
301 # test remove_columns
302 {
303   is_deeply([$schema->source('CD')->columns], [qw/cdid artist title year/]);
304   $schema->source('CD')->remove_columns('year');
305   is_deeply([$schema->source('CD')->columns], [qw/cdid artist title/]);
306   ok(! exists $schema->source('CD')->_columns->{'year'}, 'year still exists in _columns');
307 }
308