Commit | Line | Data |
fbd83464 |
1 | package dbixcsl_common_tests; |
a78e3fed |
2 | |
3 | use strict; |
4 | |
5 | use Test::More; |
18fca96a |
6 | use DBIx::Class::Schema::Loader; |
a78e3fed |
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} ||= ''; |
af6c2665 |
24 | |
25 | $self->{verbose} = $ENV{TEST_VERBOSE} || 0; |
a78e3fed |
26 | |
27 | return bless $self => $class; |
28 | } |
29 | |
30 | sub skip_tests { |
31 | my ($self, $why) = @_; |
32 | |
33 | plan skip_all => $why; |
34 | } |
35 | |
36 | sub run_tests { |
37 | my $self = shift; |
38 | |
a4a19f3c |
39 | plan tests => 27; |
a78e3fed |
40 | |
41 | $self->create(); |
42 | |
fbd83464 |
43 | my $namespace = 'DBIXCSL_Test_' . $self->{vendor}; |
a78e3fed |
44 | |
af6c2665 |
45 | my $debug = ($self->{verbose} > 1) ? 1 : 0; |
46 | |
a4a19f3c |
47 | my $schema_pkg = "$namespace\::Schema"; |
a78e3fed |
48 | |
af96f52e |
49 | my %loader_opts = ( |
50 | dsn => $self->{dsn}, |
51 | user => $self->{user}, |
52 | password => $self->{password}, |
53 | namespace => $namespace, |
54 | constraint => '^(?:\S+\.)?(?i:loader_test)[0-9]+$', |
55 | relationships => 1, |
56 | debug => $debug, |
57 | ); |
58 | |
59 | $loader_opts{db_schema} = $self->{db_schema} if $self->{db_schema}; |
60 | $loader_opts{drop_db_schema} = $self->{drop_db_schema} if $self->{drop_db_schema}; |
61 | |
a4a19f3c |
62 | eval qq{ |
63 | package $schema_pkg; |
af96f52e |
64 | use base qw/DBIx::Class::Schema::Loader/; |
65 | |
66 | __PACKAGE__->load_from_connection(\%loader_opts); |
a4a19f3c |
67 | }; |
68 | ok(!$@, "Loader initialization failed: $@"); |
69 | |
af96f52e |
70 | my $conn = $schema_pkg->connect($self->{dsn},$self->{user},$self->{password}); |
a4a19f3c |
71 | |
72 | my $moniker1 = $conn->moniker('loader_test1'); |
af6c2665 |
73 | my $rsobj1 = $conn->resultset($moniker1); |
a4a19f3c |
74 | my $moniker2 = $conn->moniker('loader_test2'); |
af6c2665 |
75 | my $rsobj2 = $conn->resultset($moniker2); |
a78e3fed |
76 | |
af6c2665 |
77 | isa_ok( $rsobj1, "DBIx::Class::ResultSet" ); |
78 | isa_ok( $rsobj2, "DBIx::Class::ResultSet" ); |
a78e3fed |
79 | |
af6c2665 |
80 | my $obj = $rsobj1->find(1); |
a78e3fed |
81 | is( $obj->id, 1 ); |
82 | is( $obj->dat, "foo" ); |
af6c2665 |
83 | is( $rsobj2->count, 4 ); |
a78e3fed |
84 | |
af6c2665 |
85 | my ($obj2) = $rsobj2->find( dat => 'bbb' ); |
a78e3fed |
86 | is( $obj2->id, 2 ); |
87 | |
88 | SKIP: { |
89 | skip $self->{skip_rels}, 20 if $self->{skip_rels}; |
90 | |
a4a19f3c |
91 | my $moniker3 = $conn->moniker('loader_test3'); |
af6c2665 |
92 | my $rsobj3 = $conn->resultset($moniker3); |
a4a19f3c |
93 | my $moniker4 = $conn->moniker('loader_test4'); |
af6c2665 |
94 | my $rsobj4 = $conn->resultset($moniker4); |
a4a19f3c |
95 | my $moniker5 = $conn->moniker('loader_test5'); |
af6c2665 |
96 | my $rsobj5 = $conn->resultset($moniker5); |
a4a19f3c |
97 | my $moniker6 = $conn->moniker('loader_test6'); |
af6c2665 |
98 | my $rsobj6 = $conn->resultset($moniker6); |
a4a19f3c |
99 | my $moniker7 = $conn->moniker('loader_test7'); |
af6c2665 |
100 | my $rsobj7 = $conn->resultset($moniker7); |
a4a19f3c |
101 | my $moniker8 = $conn->moniker('loader_test8'); |
af6c2665 |
102 | my $rsobj8 = $conn->resultset($moniker8); |
a4a19f3c |
103 | my $moniker9 = $conn->moniker('loader_test9'); |
af6c2665 |
104 | my $rsobj9 = $conn->resultset($moniker9); |
105 | |
106 | isa_ok( $rsobj3, "DBIx::Class::ResultSet" ); |
107 | isa_ok( $rsobj4, "DBIx::Class::ResultSet" ); |
108 | isa_ok( $rsobj5, "DBIx::Class::ResultSet" ); |
109 | isa_ok( $rsobj6, "DBIx::Class::ResultSet" ); |
110 | isa_ok( $rsobj7, "DBIx::Class::ResultSet" ); |
111 | isa_ok( $rsobj8, "DBIx::Class::ResultSet" ); |
112 | isa_ok( $rsobj9, "DBIx::Class::ResultSet" ); |
a78e3fed |
113 | |
114 | # basic rel test |
af6c2665 |
115 | my $obj4 = $rsobj4->find(123); |
116 | isa_ok( $obj4->fkid, "$namespace\::$moniker3"); |
a78e3fed |
117 | |
118 | # fk def in comments should not be parsed |
af6c2665 |
119 | my $obj5 = $rsobj5->find( id1 => 1, id2 => 1 ); |
a78e3fed |
120 | is( ref( $obj5->id2 ), '' ); |
121 | |
af6c2665 |
122 | # mulit-col fk def (works for some, not others...) |
123 | my $obj6 = $rsobj6->find(1); |
124 | isa_ok( $obj6->loader_test2, "$namespace\::$moniker2" ); |
a78e3fed |
125 | SKIP: { |
126 | skip "Multi-column FKs are only half-working for this vendor", 1 |
127 | unless $self->{multi_fk_broken}; |
128 | is( ref( $obj6->id2 ), '' ); |
129 | } |
130 | |
131 | # fk that references a non-pk key (UNIQUE) |
af6c2665 |
132 | my $obj8 = $rsobj8->find(1); |
133 | isa_ok( $obj8->loader_test7, "$namespace\::$moniker7" ); |
a78e3fed |
134 | |
135 | # from Chisel's tests... |
136 | SKIP: { |
137 | if($self->{vendor} =~ /sqlite/i) { |
138 | skip 'SQLite cannot do the advanced tests', 8; |
139 | } |
140 | |
a4a19f3c |
141 | my $moniker10 = $conn->moniker('loader_test10'); |
af6c2665 |
142 | my $rsobj10 = $conn->resultset($moniker10); |
a4a19f3c |
143 | my $moniker11 = $conn->moniker('loader_test11'); |
af6c2665 |
144 | my $rsobj11 = $conn->resultset($moniker11); |
a78e3fed |
145 | |
af6c2665 |
146 | isa_ok( $rsobj10, "DBIx::Class::ResultSet" ); |
147 | isa_ok( $rsobj11, "DBIx::Class::ResultSet" ); |
a78e3fed |
148 | |
af6c2665 |
149 | my $obj10 = $rsobj10->create({ subject => 'xyzzy' }); |
a78e3fed |
150 | |
151 | $obj10->update(); |
152 | ok( defined $obj10, '$obj10 is defined' ); |
153 | |
af6c2665 |
154 | my $obj11 = $rsobj11->create({ loader_test10 => $obj10->id() }); |
a78e3fed |
155 | $obj11->update(); |
156 | ok( defined $obj11, '$obj11 is defined' ); |
157 | |
158 | eval { |
159 | my $obj10_2 = $obj11->loader_test10; |
160 | $obj10_2->loader_test11( $obj11->id11() ); |
161 | $obj10_2->update(); |
162 | }; |
163 | is($@, '', 'No errors after eval{}'); |
164 | |
165 | SKIP: { |
166 | skip 'Previous eval block failed', 3 |
167 | unless ($@ eq ''); |
168 | |
af6c2665 |
169 | my $results = $rsobj10->search({ subject => 'xyzzy' }); |
a78e3fed |
170 | is( $results->count(), 1, |
af6c2665 |
171 | 'One $rsobj10 returned from search' ); |
a78e3fed |
172 | |
173 | my $obj10_3 = $results->first(); |
af6c2665 |
174 | isa_ok( $obj10_3, "$namespace\::$moniker10" ); |
a78e3fed |
175 | is( $obj10_3->loader_test11()->id(), $obj11->id(), |
af6c2665 |
176 | 'found same $rsobj11 object we expected' ); |
a78e3fed |
177 | } |
178 | } |
a78e3fed |
179 | } |
180 | } |
181 | |
182 | sub dbconnect { |
183 | my ($self, $complain) = @_; |
184 | |
af96f52e |
185 | my $dbh = DBI->connect( |
a78e3fed |
186 | $self->{dsn}, $self->{user}, |
187 | $self->{password}, |
188 | { |
189 | RaiseError => $complain, |
190 | PrintError => $complain, |
191 | AutoCommit => 1, |
192 | } |
193 | ); |
af96f52e |
194 | |
195 | die "Failed to connect to database: $DBI::errstr" if !$dbh; |
196 | |
197 | return $dbh; |
a78e3fed |
198 | } |
199 | |
200 | sub create { |
201 | my $self = shift; |
202 | |
203 | my @statements = ( |
204 | qq{ |
205 | CREATE TABLE loader_test1 ( |
206 | id $self->{auto_inc_pk}, |
207 | dat VARCHAR(32) |
208 | ) $self->{innodb}; |
209 | }, |
210 | |
211 | q{ INSERT INTO loader_test1 (dat) VALUES('foo'); }, |
212 | q{ INSERT INTO loader_test1 (dat) VALUES('bar'); }, |
213 | q{ INSERT INTO loader_test1 (dat) VALUES('baz'); }, |
214 | |
215 | qq{ |
216 | CREATE TABLE loader_test2 ( |
217 | id $self->{auto_inc_pk}, |
218 | dat VARCHAR(32) |
219 | ) $self->{innodb}; |
220 | }, |
221 | |
222 | q{ INSERT INTO loader_test2 (dat) VALUES('aaa'); }, |
223 | q{ INSERT INTO loader_test2 (dat) VALUES('bbb'); }, |
224 | q{ INSERT INTO loader_test2 (dat) VALUES('ccc'); }, |
225 | q{ INSERT INTO loader_test2 (dat) VALUES('ddd'); }, |
226 | ); |
227 | |
228 | my @statements_reltests = ( |
229 | qq{ |
230 | CREATE TABLE loader_test3 ( |
231 | id INTEGER NOT NULL PRIMARY KEY, |
232 | dat VARCHAR(32) |
233 | ) $self->{innodb}; |
234 | }, |
235 | |
236 | q{ INSERT INTO loader_test3 (id,dat) VALUES(1,'aaa'); }, |
237 | q{ INSERT INTO loader_test3 (id,dat) VALUES(2,'bbb'); }, |
238 | q{ INSERT INTO loader_test3 (id,dat) VALUES(3,'ccc'); }, |
239 | q{ INSERT INTO loader_test3 (id,dat) VALUES(4,'ddd'); }, |
240 | |
241 | qq{ |
242 | CREATE TABLE loader_test4 ( |
243 | id INTEGER NOT NULL PRIMARY KEY, |
244 | fkid INTEGER NOT NULL, |
245 | dat VARCHAR(32), |
246 | FOREIGN KEY (fkid) REFERENCES loader_test3 (id) |
247 | ) $self->{innodb}; |
248 | }, |
249 | |
250 | q{ INSERT INTO loader_test4 (id,fkid,dat) VALUES(123,1,'aaa'); }, |
251 | q{ INSERT INTO loader_test4 (id,fkid,dat) VALUES(124,2,'bbb'); }, |
252 | q{ INSERT INTO loader_test4 (id,fkid,dat) VALUES(125,3,'ccc'); }, |
253 | q{ INSERT INTO loader_test4 (id,fkid,dat) VALUES(126,4,'ddd'); }, |
254 | |
255 | qq{ |
256 | CREATE TABLE loader_test5 ( |
9b046d97 |
257 | id1 INTEGER NOT NULL, |
258 | id2 INTEGER NOT NULL, -- , id2 INTEGER REFERENCES loader_test1, |
259 | dat VARCHAR(8), |
a78e3fed |
260 | PRIMARY KEY (id1,id2) |
261 | ) $self->{innodb}; |
262 | }, |
263 | |
264 | q{ INSERT INTO loader_test5 (id1,id2,dat) VALUES (1,1,'aaa'); }, |
265 | |
266 | qq{ |
267 | CREATE TABLE loader_test6 ( |
268 | id $self->{auto_inc_pk}, |
269 | id2 INTEGER, |
270 | loader_test2 INTEGER, |
9b046d97 |
271 | dat VARCHAR(8), |
a78e3fed |
272 | FOREIGN KEY (loader_test2) REFERENCES loader_test2 (id), |
273 | FOREIGN KEY (id, id2 ) REFERENCES loader_test5 (id1,id2) |
274 | ) $self->{innodb}; |
275 | }, |
276 | |
277 | (q{ INSERT INTO loader_test6 (id2,loader_test2,dat) } . |
278 | q{ VALUES (1,1,'aaa'); }), |
279 | |
280 | qq{ |
281 | CREATE TABLE loader_test7 ( |
282 | id INTEGER NOT NULL PRIMARY KEY, |
283 | id2 VARCHAR(8) NOT NULL UNIQUE, |
9b046d97 |
284 | dat VARCHAR(8) |
a78e3fed |
285 | ) $self->{innodb}; |
286 | }, |
287 | |
288 | q{ INSERT INTO loader_test7 (id,id2,dat) VALUES (1,'aaa','bbb'); }, |
289 | |
290 | qq{ |
291 | CREATE TABLE loader_test8 ( |
292 | id INTEGER NOT NULL PRIMARY KEY, |
293 | loader_test7 VARCHAR(8) NOT NULL, |
9b046d97 |
294 | dat VARCHAR(8), |
a78e3fed |
295 | FOREIGN KEY (loader_test7) REFERENCES loader_test7 (id2) |
296 | ) $self->{innodb}; |
297 | }, |
298 | |
299 | (q{ INSERT INTO loader_test8 (id,loader_test7,dat) } . |
300 | q{ VALUES (1,'aaa','bbb'); }), |
301 | |
302 | qq{ |
303 | CREATE TABLE loader_test9 ( |
9b046d97 |
304 | loader_test9 VARCHAR(8) NOT NULL |
a78e3fed |
305 | ) $self->{innodb}; |
306 | }, |
307 | ); |
308 | |
309 | my @statements_advanced = ( |
310 | qq{ |
311 | CREATE TABLE loader_test10 ( |
312 | id10 $self->{auto_inc_pk}, |
313 | subject VARCHAR(8), |
314 | loader_test11 INTEGER |
315 | ) $self->{innodb}; |
316 | }, |
317 | |
318 | qq{ |
319 | CREATE TABLE loader_test11 ( |
320 | id11 $self->{auto_inc_pk}, |
321 | message VARCHAR(8) DEFAULT 'foo', |
af96f52e |
322 | loader_test10 INTEGER, |
a78e3fed |
323 | FOREIGN KEY (loader_test10) REFERENCES loader_test10 (id10) |
324 | ) $self->{innodb}; |
325 | }, |
326 | |
327 | (q{ ALTER TABLE loader_test10 ADD CONSTRAINT } . |
328 | q{ loader_test11_fk FOREIGN KEY (loader_test11) } . |
329 | q{ REFERENCES loader_test11 (id11); }), |
330 | ); |
331 | |
af96f52e |
332 | $self->drop_tables; |
333 | |
a78e3fed |
334 | $self->{created} = 1; |
335 | |
336 | my $dbh = $self->dbconnect(1); |
337 | $dbh->do($_) for (@statements); |
338 | unless($self->{skip_rels}) { |
af96f52e |
339 | # hack for now, since DB2 doesn't like inline comments, and we need |
340 | # to test one for mysql, which works on everyone else... |
341 | # this all needs to be refactored anyways. |
342 | if($self->{vendor} =~ /DB2/i) { |
343 | @statements_reltests = map { s/--.*\n//; $_ } @statements_reltests; |
344 | } |
a78e3fed |
345 | $dbh->do($_) for (@statements_reltests); |
346 | unless($self->{vendor} =~ /sqlite/i) { |
347 | $dbh->do($_) for (@statements_advanced); |
348 | } |
349 | } |
350 | $dbh->disconnect(); |
351 | } |
352 | |
af96f52e |
353 | sub drop_tables { |
a78e3fed |
354 | my $self = shift; |
355 | |
356 | return unless $self->{created}; |
357 | |
358 | my @tables = qw/ |
359 | loader_test1 |
360 | loader_test2 |
361 | /; |
362 | |
363 | my @tables_reltests = qw/ |
364 | loader_test4 |
365 | loader_test3 |
366 | loader_test6 |
367 | loader_test5 |
368 | loader_test8 |
369 | loader_test7 |
370 | loader_test9 |
371 | /; |
372 | |
373 | my @tables_advanced = qw/ |
374 | loader_test11 |
375 | loader_test10 |
376 | /; |
377 | |
378 | my $drop_fk_mysql = |
379 | q{ALTER TABLE loader_test10 DROP FOREIGN KEY loader_test11_fk;}; |
380 | |
381 | my $drop_fk = |
382 | q{ALTER TABLE loader_test10 DROP CONSTRAINT loader_test11_fk;}; |
383 | |
384 | my $dbh = $self->dbconnect(0); |
385 | |
386 | unless($self->{skip_rels}) { |
387 | $dbh->do("DROP TABLE $_") for (@tables_reltests); |
388 | unless($self->{vendor} =~ /sqlite/i) { |
389 | if($self->{vendor} =~ /mysql/i) { |
390 | $dbh->do($drop_fk_mysql); |
391 | } |
392 | else { |
393 | $dbh->do($drop_fk); |
394 | } |
395 | $dbh->do("DROP TABLE $_") for (@tables_advanced); |
396 | } |
397 | } |
398 | $dbh->do("DROP TABLE $_") for (@tables); |
af96f52e |
399 | $dbh->disconnect; |
a78e3fed |
400 | } |
401 | |
af96f52e |
402 | sub DESTROY { shift->drop_tables; } |
403 | |
a78e3fed |
404 | 1; |