Commit | Line | Data |
a78e3fed |
1 | package dbixcl_common_tests; |
2 | |
3 | use strict; |
4 | |
5 | use Test::More; |
6 | use DBIx::Class::Loader; |
7 | use DBI; |
8 | |
9 | sub new { |
10 | my $class = shift; |
11 | |
12 | my $self; |
13 | |
14 | if( ref($_[0]) eq 'HASH') { |
15 | my $args = shift; |
16 | $self = { (%$args) }; |
17 | } |
18 | else { |
19 | $self = { @_ }; |
20 | } |
21 | |
22 | # Only MySQL uses this |
23 | $self->{innodb} ||= ''; |
24 | |
25 | return bless $self => $class; |
26 | } |
27 | |
28 | sub skip_tests { |
29 | my ($self, $why) = @_; |
30 | |
31 | plan skip_all => $why; |
32 | } |
33 | |
34 | sub run_tests { |
35 | my $self = shift; |
36 | |
37 | plan tests => 26; |
38 | |
39 | $self->create(); |
40 | |
41 | my $namespace = 'DBIXCL_Test_' . $self->{vendor}; |
42 | |
43 | my $loader = DBIx::Class::Loader->new( |
44 | dsn => $self->{dsn}, |
45 | user => $self->{user}, |
46 | password => $self->{password}, |
47 | namespace => $namespace, |
48 | constraint => '^loader_test.*', |
49 | relationships => 1, |
50 | ); |
51 | |
52 | my $class1 = $loader->find_class("loader_test1"); |
53 | my $class2 = $loader->find_class("loader_test2"); |
54 | |
55 | is( $class1, "${namespace}::LoaderTest1" ); |
56 | is( $class2, "${namespace}::LoaderTest2" ); |
57 | |
58 | my $obj = $class1->find(1); |
59 | is( $obj->id, 1 ); |
60 | is( $obj->dat, "foo" ); |
61 | is( $class2->count, 4 ); |
62 | |
63 | my ($obj2) = $class2->find( dat => 'bbb' ); |
64 | is( $obj2->id, 2 ); |
65 | |
66 | SKIP: { |
67 | skip $self->{skip_rels}, 20 if $self->{skip_rels}; |
68 | |
69 | my $class3 = $loader->find_class("loader_test3"); |
70 | my $class4 = $loader->find_class("loader_test4"); |
71 | my $class5 = $loader->find_class("loader_test5"); |
72 | my $class6 = $loader->find_class("loader_test6"); |
73 | my $class7 = $loader->find_class("loader_test7"); |
74 | my $class8 = $loader->find_class("loader_test8"); |
75 | my $class9 = $loader->find_class("loader_test9"); |
76 | |
77 | is( $class3, "${namespace}::LoaderTest3" ); |
78 | is( $class4, "${namespace}::LoaderTest4" ); |
79 | is( $class5, "${namespace}::LoaderTest5" ); |
80 | is( $class6, "${namespace}::LoaderTest6" ); |
81 | is( $class7, "${namespace}::LoaderTest7" ); |
82 | is( $class8, "${namespace}::LoaderTest8" ); |
83 | is( $class9, "${namespace}::LoaderTest9" ); |
84 | |
85 | # basic rel test |
86 | my $obj4 = $class4->find(123); |
87 | is( ref($obj4->fkid), $class3); |
88 | |
89 | # fk def in comments should not be parsed |
90 | my $obj5 = $class5->find( id1 => 1, id2 => 1 ); |
91 | is( ref( $obj5->id2 ), '' ); |
92 | |
93 | # mulit-col fk def (works halfway for some, not others...) |
94 | my $obj6 = $class6->find(1); |
95 | isa_ok( $obj6->loader_test2, $class2 ); |
96 | SKIP: { |
97 | skip "Multi-column FKs are only half-working for this vendor", 1 |
98 | unless $self->{multi_fk_broken}; |
99 | is( ref( $obj6->id2 ), '' ); |
100 | } |
101 | |
102 | # fk that references a non-pk key (UNIQUE) |
103 | my $obj8 = $class8->find(1); |
104 | isa_ok( $obj8->loader_test7, $class7 ); |
105 | |
106 | # from Chisel's tests... |
107 | SKIP: { |
108 | if($self->{vendor} =~ /sqlite/i) { |
109 | skip 'SQLite cannot do the advanced tests', 8; |
110 | } |
111 | |
112 | my $class10 = $loader->find_class('loader_test10'); |
113 | my $class11 = $loader->find_class('loader_test11'); |
114 | |
115 | is( $class10, "${namespace}::LoaderTest10" ); |
116 | is( $class11, "${namespace}::LoaderTest11" ); |
117 | |
118 | my $obj10 = $class10->create({ subject => 'xyzzy' }); |
119 | |
120 | $obj10->update(); |
121 | ok( defined $obj10, '$obj10 is defined' ); |
122 | |
123 | my $obj11 = $class11->create({ loader_test10 => $obj10->id() }); |
124 | $obj11->update(); |
125 | ok( defined $obj11, '$obj11 is defined' ); |
126 | |
127 | eval { |
128 | my $obj10_2 = $obj11->loader_test10; |
129 | $obj10_2->loader_test11( $obj11->id11() ); |
130 | $obj10_2->update(); |
131 | }; |
132 | is($@, '', 'No errors after eval{}'); |
133 | |
134 | SKIP: { |
135 | skip 'Previous eval block failed', 3 |
136 | unless ($@ eq ''); |
137 | |
138 | my $results = $class10->search({ subject => 'xyzzy' }); |
139 | is( $results->count(), 1, |
140 | 'One $class10 returned from search' ); |
141 | |
142 | my $obj10_3 = $results->first(); |
143 | isa_ok( $obj10_3, $class10 ); |
144 | is( $obj10_3->loader_test11()->id(), $obj11->id(), |
145 | 'found same $class11 object we expected' ); |
146 | } |
147 | |
148 | for ( $class10, $class11 ) { |
149 | $_->storage->dbh->disconnect; |
150 | } |
151 | } |
152 | |
153 | for ( $class3, $class4, $class5, $class6, $class7, |
154 | $class8, $class9 ) { |
155 | $_->storage->dbh->disconnect; |
156 | } |
157 | } |
158 | |
159 | for ( $class1, $class2 ) { |
160 | $_->storage->dbh->disconnect; |
161 | } |
162 | } |
163 | |
164 | sub dbconnect { |
165 | my ($self, $complain) = @_; |
166 | |
167 | DBI->connect( |
168 | $self->{dsn}, $self->{user}, |
169 | $self->{password}, |
170 | { |
171 | RaiseError => $complain, |
172 | PrintError => $complain, |
173 | AutoCommit => 1, |
174 | } |
175 | ); |
176 | } |
177 | |
178 | sub create { |
179 | my $self = shift; |
180 | |
181 | my @statements = ( |
182 | qq{ |
183 | CREATE TABLE loader_test1 ( |
184 | id $self->{auto_inc_pk}, |
185 | dat VARCHAR(32) |
186 | ) $self->{innodb}; |
187 | }, |
188 | |
189 | q{ INSERT INTO loader_test1 (dat) VALUES('foo'); }, |
190 | q{ INSERT INTO loader_test1 (dat) VALUES('bar'); }, |
191 | q{ INSERT INTO loader_test1 (dat) VALUES('baz'); }, |
192 | |
193 | qq{ |
194 | CREATE TABLE loader_test2 ( |
195 | id $self->{auto_inc_pk}, |
196 | dat VARCHAR(32) |
197 | ) $self->{innodb}; |
198 | }, |
199 | |
200 | q{ INSERT INTO loader_test2 (dat) VALUES('aaa'); }, |
201 | q{ INSERT INTO loader_test2 (dat) VALUES('bbb'); }, |
202 | q{ INSERT INTO loader_test2 (dat) VALUES('ccc'); }, |
203 | q{ INSERT INTO loader_test2 (dat) VALUES('ddd'); }, |
204 | ); |
205 | |
206 | my @statements_reltests = ( |
207 | qq{ |
208 | CREATE TABLE loader_test3 ( |
209 | id INTEGER NOT NULL PRIMARY KEY, |
210 | dat VARCHAR(32) |
211 | ) $self->{innodb}; |
212 | }, |
213 | |
214 | q{ INSERT INTO loader_test3 (id,dat) VALUES(1,'aaa'); }, |
215 | q{ INSERT INTO loader_test3 (id,dat) VALUES(2,'bbb'); }, |
216 | q{ INSERT INTO loader_test3 (id,dat) VALUES(3,'ccc'); }, |
217 | q{ INSERT INTO loader_test3 (id,dat) VALUES(4,'ddd'); }, |
218 | |
219 | qq{ |
220 | CREATE TABLE loader_test4 ( |
221 | id INTEGER NOT NULL PRIMARY KEY, |
222 | fkid INTEGER NOT NULL, |
223 | dat VARCHAR(32), |
224 | FOREIGN KEY (fkid) REFERENCES loader_test3 (id) |
225 | ) $self->{innodb}; |
226 | }, |
227 | |
228 | q{ INSERT INTO loader_test4 (id,fkid,dat) VALUES(123,1,'aaa'); }, |
229 | q{ INSERT INTO loader_test4 (id,fkid,dat) VALUES(124,2,'bbb'); }, |
230 | q{ INSERT INTO loader_test4 (id,fkid,dat) VALUES(125,3,'ccc'); }, |
231 | q{ INSERT INTO loader_test4 (id,fkid,dat) VALUES(126,4,'ddd'); }, |
232 | |
233 | qq{ |
234 | CREATE TABLE loader_test5 ( |
235 | id1 INTEGER, |
236 | id2 INTEGER, -- , id2 INTEGER REFERENCES loader_test1, |
237 | dat TEXT, |
238 | PRIMARY KEY (id1,id2) |
239 | ) $self->{innodb}; |
240 | }, |
241 | |
242 | q{ INSERT INTO loader_test5 (id1,id2,dat) VALUES (1,1,'aaa'); }, |
243 | |
244 | qq{ |
245 | CREATE TABLE loader_test6 ( |
246 | id $self->{auto_inc_pk}, |
247 | id2 INTEGER, |
248 | loader_test2 INTEGER, |
249 | dat TEXT, |
250 | FOREIGN KEY (loader_test2) REFERENCES loader_test2 (id), |
251 | FOREIGN KEY (id, id2 ) REFERENCES loader_test5 (id1,id2) |
252 | ) $self->{innodb}; |
253 | }, |
254 | |
255 | (q{ INSERT INTO loader_test6 (id2,loader_test2,dat) } . |
256 | q{ VALUES (1,1,'aaa'); }), |
257 | |
258 | qq{ |
259 | CREATE TABLE loader_test7 ( |
260 | id INTEGER NOT NULL PRIMARY KEY, |
261 | id2 VARCHAR(8) NOT NULL UNIQUE, |
262 | dat TEXT |
263 | ) $self->{innodb}; |
264 | }, |
265 | |
266 | q{ INSERT INTO loader_test7 (id,id2,dat) VALUES (1,'aaa','bbb'); }, |
267 | |
268 | qq{ |
269 | CREATE TABLE loader_test8 ( |
270 | id INTEGER NOT NULL PRIMARY KEY, |
271 | loader_test7 VARCHAR(8) NOT NULL, |
272 | dat TEXT, |
273 | FOREIGN KEY (loader_test7) REFERENCES loader_test7 (id2) |
274 | ) $self->{innodb}; |
275 | }, |
276 | |
277 | (q{ INSERT INTO loader_test8 (id,loader_test7,dat) } . |
278 | q{ VALUES (1,'aaa','bbb'); }), |
279 | |
280 | qq{ |
281 | CREATE TABLE loader_test9 ( |
282 | loader_test9 TEXT NOT NULL |
283 | ) $self->{innodb}; |
284 | }, |
285 | ); |
286 | |
287 | my @statements_advanced = ( |
288 | qq{ |
289 | CREATE TABLE loader_test10 ( |
290 | id10 $self->{auto_inc_pk}, |
291 | subject VARCHAR(8), |
292 | loader_test11 INTEGER |
293 | ) $self->{innodb}; |
294 | }, |
295 | |
296 | qq{ |
297 | CREATE TABLE loader_test11 ( |
298 | id11 $self->{auto_inc_pk}, |
299 | message VARCHAR(8) DEFAULT 'foo', |
300 | loader_test10 INTEGER NOT NULL, |
301 | FOREIGN KEY (loader_test10) REFERENCES loader_test10 (id10) |
302 | ) $self->{innodb}; |
303 | }, |
304 | |
305 | (q{ ALTER TABLE loader_test10 ADD CONSTRAINT } . |
306 | q{ loader_test11_fk FOREIGN KEY (loader_test11) } . |
307 | q{ REFERENCES loader_test11 (id11); }), |
308 | ); |
309 | |
310 | $self->{created} = 1; |
311 | |
312 | my $dbh = $self->dbconnect(1); |
313 | $dbh->do($_) for (@statements); |
314 | unless($self->{skip_rels}) { |
315 | $dbh->do($_) for (@statements_reltests); |
316 | unless($self->{vendor} =~ /sqlite/i) { |
317 | $dbh->do($_) for (@statements_advanced); |
318 | } |
319 | } |
320 | $dbh->disconnect(); |
321 | } |
322 | |
323 | sub DESTROY { |
324 | my $self = shift; |
325 | |
326 | return unless $self->{created}; |
327 | |
328 | my @tables = qw/ |
329 | loader_test1 |
330 | loader_test2 |
331 | /; |
332 | |
333 | my @tables_reltests = qw/ |
334 | loader_test4 |
335 | loader_test3 |
336 | loader_test6 |
337 | loader_test5 |
338 | loader_test8 |
339 | loader_test7 |
340 | loader_test9 |
341 | /; |
342 | |
343 | my @tables_advanced = qw/ |
344 | loader_test11 |
345 | loader_test10 |
346 | /; |
347 | |
348 | my $drop_fk_mysql = |
349 | q{ALTER TABLE loader_test10 DROP FOREIGN KEY loader_test11_fk;}; |
350 | |
351 | my $drop_fk = |
352 | q{ALTER TABLE loader_test10 DROP CONSTRAINT loader_test11_fk;}; |
353 | |
354 | my $dbh = $self->dbconnect(0); |
355 | |
356 | unless($self->{skip_rels}) { |
357 | $dbh->do("DROP TABLE $_") for (@tables_reltests); |
358 | unless($self->{vendor} =~ /sqlite/i) { |
359 | if($self->{vendor} =~ /mysql/i) { |
360 | $dbh->do($drop_fk_mysql); |
361 | } |
362 | else { |
363 | $dbh->do($drop_fk); |
364 | } |
365 | $dbh->do("DROP TABLE $_") for (@tables_advanced); |
366 | } |
367 | } |
368 | $dbh->do("DROP TABLE $_") for (@tables); |
369 | $dbh->disconnect(); |
370 | } |
371 | |
372 | 1; |