add $source->resultset_attributes, include_columns rs attr
[dbsrgits/DBIx-Class.git] / t / run / 01core.tl
1 sub run_tests {
2 my $schema = shift;
3
4 plan tests => 38; 
5
6 my @art = $schema->resultset("Artist")->search({ }, { order_by => 'name DESC'});
7
8 cmp_ok(@art, '==', 3, "Three artists returned");
9
10 my $art = $art[0];
11
12 is($art->name, 'We Are Goth', "Correct order too");
13
14 $art->name('We Are In Rehab');
15
16 is($art->name, 'We Are In Rehab', "Accessor update ok");
17
18 is($art->get_column("name"), 'We Are In Rehab', 'And via get_column');
19
20 ok($art->update, 'Update run');
21
22 @art = $schema->resultset("Artist")->search({ name => 'We Are In Rehab' });
23
24 cmp_ok(@art, '==', 1, "Changed artist returned by search");
25
26 cmp_ok($art[0]->artistid, '==', 3,'Correct artist too');
27
28 $art->delete;
29
30 @art = $schema->resultset("Artist")->search({ });
31
32 cmp_ok(@art, '==', 2, 'And then there were two');
33
34 ok(!$art->in_storage, "It knows it's dead");
35
36 eval { $art->delete; };
37
38 ok($@, "Can't delete twice: $@");
39
40 is($art->name, 'We Are In Rehab', 'But the object is still live');
41
42 $art->insert;
43
44 ok($art->in_storage, "Re-created");
45
46 @art = $schema->resultset("Artist")->search({ });
47
48 cmp_ok(@art, '==', 3, 'And now there are three again');
49
50 my $new = $schema->resultset("Artist")->create({ artistid => 4 });
51
52 cmp_ok($new->artistid, '==', 4, 'Create produced record ok');
53
54 @art = $schema->resultset("Artist")->search({ });
55
56 cmp_ok(@art, '==', 4, "Oh my god! There's four of them!");
57
58 $new->set_column('name' => 'Man With A Fork');
59
60 is($new->name, 'Man With A Fork', 'set_column ok');
61
62 $new->discard_changes;
63
64 ok(!defined $new->name, 'Discard ok');
65
66 $new->name('Man With A Spoon');
67
68 $new->update;
69
70 $new_again = $schema->resultset("Artist")->find(4);
71
72 is($new_again->name, 'Man With A Spoon', 'Retrieved correctly');
73
74 is($new_again->ID, 'DBICTest::Artist|artist|artistid=4', 'unique object id generated correctly');
75
76 is($schema->resultset("Artist")->count, 4, 'count ok');
77
78 my $cd = $schema->resultset("CD")->find(1);
79 my %cols = $cd->get_columns;
80
81 cmp_ok(keys %cols, '==', 4, 'get_columns number of columns ok');
82
83 is($cols{title}, 'Spoonful of bees', 'get_columns values ok');
84
85 %cols = ( title => 'Forkful of bees', year => 2005);
86 $cd->set_columns(\%cols);
87
88 is($cd->title, 'Forkful of bees', 'set_columns ok');
89
90 is($cd->year, 2005, 'set_columns ok');
91
92 $cd->discard_changes;
93
94 # check whether ResultSource->columns returns columns in order originally supplied
95 my @cd = $schema->source("CD")->columns;
96
97 is_deeply( \@cd, [qw/cdid artist title year/], 'column order');
98
99 $cd = $schema->resultset("CD")->search({ title => 'Spoonful of bees' }, { cols => ['title'] })->next;
100 is($cd->title, 'Spoonful of bees', 'subset of columns returned correctly');
101
102 $cd = $schema->resultset("CD")->search(undef, { include_columns => [ 'artist.name' ], join => [ 'artist' ] })->find(1);
103
104 is($cd->title, 'Spoonful of bees', 'Correct CD returned with include');
105 is($cd->get_column('name'), 'Caterwauler McCrae', 'Additional column returned');
106
107 # insert_or_update
108 $new = $schema->resultset("Track")->new( {
109   trackid => 100,
110   cd => 1,
111   position => 1,
112   title => 'Insert or Update',
113 } );
114 $new->insert_or_update;
115 ok($new->in_storage, 'insert_or_update insert ok');
116
117 # test in update mode
118 $new->pos(5);
119 $new->insert_or_update;
120 is( $schema->resultset("Track")->find(100)->pos, 5, 'insert_or_update update ok');
121
122 eval { $schema->class("Track")->load_components('DoesNotExist'); };
123
124 ok $@, $@;
125
126 is($schema->class("Artist")->field_name_for->{name}, 'artist name', 'mk_classdata usage ok');
127
128 my $search = [ { 'tags.tag' => 'Cheesy' }, { 'tags.tag' => 'Blue' } ];
129
130 my $or_rs = $schema->resultset("CD")->search($search, { join => 'tags',
131                                                   order_by => 'cdid' });
132
133 cmp_ok($or_rs->count, '==', 5, 'Search with OR ok');
134
135 my $distinct_rs = $schema->resultset("CD")->search($search, { join => 'tags', distinct => 1 });
136
137 cmp_ok($distinct_rs->all, '==', 4, 'DISTINCT search with OR ok');
138
139 my $tag_rs = $schema->resultset('Tag')->search(
140                [ { 'me.tag' => 'Cheesy' }, { 'me.tag' => 'Blue' } ]);
141
142 my $rel_rs = $tag_rs->search_related('cd');
143
144 cmp_ok($rel_rs->count, '==', 5, 'Related search ok');
145
146 cmp_ok($or_rs->next->cdid, '==', $rel_rs->next->cdid, 'Related object ok');
147
148
149 ok($schema->storage(), 'Storage available');
150
151 $schema->source("Artist")->{_columns}{'artistid'} = {};
152
153 my $typeinfo = $schema->source("Artist")->column_info('artistid');
154 is($typeinfo->{data_type}, 'INTEGER', 'column_info ok');
155 }
156
157 1;