support for fully-qualified components in result_component_map and tests
[dbsrgits/DBIx-Class-Schema-Loader.git] / t / lib / dbixcsl_common_tests.pm
1 package dbixcsl_common_tests;
2
3 use strict;
4 use warnings;
5
6 use Test::More;
7 use Test::Exception;
8 use DBIx::Class::Schema::Loader;
9 use Class::Unload;
10 use File::Path;
11 use DBI;
12 use Digest::MD5;
13 use File::Find 'find';
14 use Class::Unload ();
15 use DBIx::Class::Schema::Loader::Utils 'dumper_squashed';
16 use List::MoreUtils 'apply';
17 use DBIx::Class::Schema::Loader::Optional::Dependencies ();
18 use Try::Tiny;
19 use namespace::clean;
20
21 use dbixcsl_test_dir qw/$tdir/;
22
23 my $DUMP_DIR = "$tdir/common_dump";
24 rmtree $DUMP_DIR;
25
26 use constant RESCAN_WARNINGS => qr/(?i:loader_test|LoaderTest)\d+s? has no primary key|^Dumping manual schema|^Schema dump completed|collides with an inherited method|invalidates \d+ active statement|^Bad table or view/;
27
28 sub new {
29     my $class = shift;
30
31     my $self;
32
33     if( ref($_[0]) eq 'HASH') {
34        my $args = shift;
35        $self = { (%$args) };
36     }
37     else {
38        $self = { @_ };
39     }
40
41     # Only MySQL uses this
42     $self->{innodb} ||= '';
43
44     # DB2 and Firebird don't support 'field type NULL'
45     $self->{null} = 'NULL' unless defined $self->{null};
46     
47     $self->{verbose} = $ENV{TEST_VERBOSE} || 0;
48
49     # Optional extra tables and tests
50     $self->{extra} ||= {};
51
52     $self->{basic_date_datatype} ||= 'DATE';
53
54     # Not all DBS do SQL-standard CURRENT_TIMESTAMP
55     $self->{default_function} ||= "current_timestamp";
56     $self->{default_function_def} ||= "timestamp default $self->{default_function}";
57
58     $self = bless $self, $class;
59
60     $self->setup_data_type_tests;
61
62     return $self;
63 }
64
65 sub skip_tests {
66     my ($self, $why) = @_;
67
68     plan skip_all => $why;
69 }
70
71 sub _monikerize {
72     my $name = shift;
73     return 'LoaderTest2X' if $name =~ /^loader_test2$/i;
74     return undef;
75 }
76
77 sub run_tests {
78     my $self = shift;
79
80     my @connect_info;
81
82     if ($self->{dsn}) {
83         push @connect_info, [ @{$self}{qw/dsn user password connect_info_opts/ } ];
84     }
85     else {
86         foreach my $info (@{ $self->{connect_info} || [] }) {
87             push @connect_info, [ @{$info}{qw/dsn user password connect_info_opts/ } ];
88         }
89     }
90     
91     if ($ENV{SCHEMA_LOADER_TESTS_EXTRA_ONLY}) {
92         $self->run_only_extra_tests(\@connect_info);
93         return;
94     }
95
96     my $extra_count = $self->{extra}{count} || 0;
97
98     my $col_accessor_map_tests = 5;
99     my $num_rescans = 5;
100     $num_rescans-- if $self->{vendor} =~ /^(?:sybase|mysql)\z/i;
101     $num_rescans++ if $self->{vendor} eq 'mssql';
102     $num_rescans++ if $self->{vendor} eq 'Firebird';
103
104     plan tests => @connect_info *
105         (192 + $num_rescans * $col_accessor_map_tests + $extra_count + ($self->{data_type_tests}{test_count} || 0));
106
107     foreach my $info_idx (0..$#connect_info) {
108         my $info = $connect_info[$info_idx];
109
110         @{$self}{qw/dsn user password connect_info_opts/} = @$info;
111
112         $self->create();
113
114         my $schema_class = $self->setup_schema($info);
115         $self->test_schema($schema_class);
116
117         rmtree $DUMP_DIR
118             unless $ENV{SCHEMA_LOADER_TESTS_NOCLEANUP} && $info_idx == $#connect_info;
119     }
120 }
121
122 sub run_only_extra_tests {
123     my ($self, $connect_info) = @_;
124
125     plan tests => @$connect_info * (4 + ($self->{extra}{count} || 0) + ($self->{data_type_tests}{test_count} || 0));
126
127     rmtree $DUMP_DIR;
128
129     foreach my $info_idx (0..$#$connect_info) {
130         my $info = $connect_info->[$info_idx];
131
132         @{$self}{qw/dsn user password connect_info_opts/} = @$info;
133
134         $self->drop_extra_tables_only;
135
136         my $dbh = $self->dbconnect(1);
137         $dbh->do($_) for @{ $self->{pre_create} || [] };
138         $dbh->do($_) for @{ $self->{extra}{create} || [] };
139
140         if (not ($self->{vendor} eq 'mssql' && $dbh->{Driver}{Name} eq 'Sybase')) {
141             $dbh->do($_) for @{ $self->{data_type_tests}{ddl} || []};
142         }
143
144         $self->{_created} = 1;
145
146         my $file_count = grep /CREATE (?:TABLE|VIEW)/i, @{ $self->{extra}{create} || [] };
147         $file_count++; # schema
148         
149         if (not ($self->{vendor} eq 'mssql' && $dbh->{Driver}{Name} eq 'Sybase')) {
150             $file_count++ for @{ $self->{data_type_tests}{table_names} || [] };
151         }
152
153         my $schema_class = $self->setup_schema($info, $file_count);
154         my ($monikers, $classes) = $self->monikers_and_classes($schema_class);
155         my $conn = $schema_class->clone;
156
157         $self->test_data_types($conn);
158         $self->{extra}{run}->($conn, $monikers, $classes, $self) if $self->{extra}{run};
159
160         if (not ($ENV{SCHEMA_LOADER_TESTS_NOCLEANUP} && $info_idx == $#$connect_info)) {
161             $self->drop_extra_tables_only;
162             rmtree $DUMP_DIR;
163         }
164     }
165 }
166
167 sub drop_extra_tables_only {
168     my $self = shift;
169
170     my $dbh = $self->dbconnect(0);
171
172     local $^W = 0; # for ADO
173
174     $dbh->do($_) for @{ $self->{extra}{pre_drop_ddl} || [] };
175     $dbh->do("DROP TABLE $_") for @{ $self->{extra}{drop} || [] };
176
177     if (not ($self->{vendor} eq 'mssql' && $dbh->{Driver}{Name} eq 'Sybase')) {
178         foreach my $data_type_table (@{ $self->{data_type_tests}{table_names} || [] }) {
179             $dbh->do("DROP TABLE $data_type_table");
180         }
181     }
182 }
183
184 # defined in sub create
185 my (@statements, @statements_reltests, @statements_advanced,
186     @statements_advanced_sqlite, @statements_inline_rels,
187     @statements_implicit_rels);
188
189 sub setup_schema {
190     my ($self, $connect_info, $expected_count) = @_;
191
192     my $schema_class = 'DBIXCSL_Test::Schema';
193
194     my $debug = ($self->{verbose} > 1) ? 1 : 0;
195
196     if (
197       $ENV{SCHEMA_LOADER_TESTS_USE_MOOSE}
198         &&
199       ! DBIx::Class::Schema::Loader::Optional::Dependencies->req_ok_for('use_moose')
200     ) {
201       die sprintf ("Missing dependencies for SCHEMA_LOADER_TESTS_USE_MOOSE: %s\n",
202         DBIx::Class::Schema::Loader::Optional::Dependencies->req_missing_for('use_moose')
203       );
204     }
205
206     my %loader_opts = (
207         constraint              =>
208           qr/^(?:\S+\.)?(?:(?:$self->{vendor}|extra)_?)?loader_?test[0-9]+(?!.*_)/i,
209         relationships           => 1,
210         additional_classes      => 'TestAdditional',
211         additional_base_classes => 'TestAdditionalBase',
212         left_base_classes       => [ qw/TestLeftBase/ ],
213         components              => [ qw/TestComponent +TestComponentFQN/ ],
214         inflect_plural          => { loader_test4_fkid => 'loader_test4zes' },
215         inflect_singular        => { fkid => 'fkid_singular' },
216         moniker_map             => \&_monikerize,
217         custom_column_info      => \&_custom_column_info,
218         debug                   => $debug,
219         use_namespaces          => 0,
220         dump_directory          => $DUMP_DIR,
221         datetime_timezone       => 'Europe/Berlin',
222         datetime_locale         => 'de_DE',
223         use_moose               => $ENV{SCHEMA_LOADER_TESTS_USE_MOOSE},
224         col_collision_map       => { '^(can)\z' => 'caught_collision_%s' },
225         rel_collision_map       => { '^(set_primary_key)\z' => 'caught_rel_collision_%s' },
226         col_accessor_map        => \&test_col_accessor_map,
227         result_component_map    => { LoaderTest2X => 'TestComponentForMap', LoaderTest1 => '+TestComponentForMapFQN' },
228         %{ $self->{loader_options} || {} },
229     );
230
231     $loader_opts{db_schema} = $self->{db_schema} if $self->{db_schema};
232
233     Class::Unload->unload($schema_class);
234
235     my $file_count;
236     {
237         my @loader_warnings;
238         local $SIG{__WARN__} = sub { push(@loader_warnings, @_); };
239          eval qq{
240              package $schema_class;
241              use base qw/DBIx::Class::Schema::Loader/;
242      
243              __PACKAGE__->loader_options(\%loader_opts);
244              __PACKAGE__->connection(\@\$connect_info);
245          };
246  
247         ok(!$@, "Loader initialization") or diag $@;
248
249         find sub { return if -d; $file_count++ }, $DUMP_DIR;
250
251         my $standard_sources = not defined $expected_count;
252
253         if ($standard_sources) {
254             $expected_count = 36;
255
256             if (not ($self->{vendor} eq 'mssql' && $connect_info->[0] =~ /Sybase/)) {
257                 $expected_count++ for @{ $self->{data_type_tests}{table_names} || [] };
258             }
259
260             $expected_count += grep /CREATE (?:TABLE|VIEW)/i,
261                 @{ $self->{extra}{create} || [] };
262      
263             $expected_count -= grep /CREATE TABLE/, @statements_inline_rels
264                 if $self->{skip_rels} || $self->{no_inline_rels};
265      
266             $expected_count -= grep /CREATE TABLE/, @statements_implicit_rels
267                 if $self->{skip_rels} || $self->{no_implicit_rels};
268      
269             $expected_count -= grep /CREATE TABLE/, ($self->{vendor} =~ /sqlite/ ? @statements_advanced_sqlite : @statements_advanced), @statements_reltests
270                 if $self->{skip_rels};
271         }
272  
273         is $file_count, $expected_count, 'correct number of files generated';
274  
275         my $warn_count = 2;
276  
277         $warn_count++ for grep /^Bad table or view/, @loader_warnings;
278  
279         $warn_count++ for grep /renaming \S+ relation/, @loader_warnings;
280  
281         $warn_count++ for grep /\b(?!loader_test9)\w+ has no primary key/i, @loader_warnings;
282
283         $warn_count++ for grep /^Column '\w+' in table '\w+' collides with an inherited method\./, @loader_warnings;
284
285         $warn_count++ for grep /^Relationship '\w+' in source '\w+' for columns '[^']+' collides with an inherited method\./, @loader_warnings;
286
287         $warn_count++ for grep { my $w = $_; grep $w =~ $_, @{ $self->{warnings} || [] } } @loader_warnings;
288
289         $warn_count-- for grep { my $w = $_; grep $w =~ $_, @{ $self->{failtrigger_warnings} || [] } } @loader_warnings;
290
291         if ($standard_sources) {
292             if($self->{skip_rels}) {
293                 SKIP: {
294                     is(scalar(@loader_warnings), $warn_count, "No loader warnings")
295                         or diag @loader_warnings;
296                     skip "No missing PK warnings without rels", 1;
297                 }
298             }
299             else {
300                 $warn_count++;
301                 is(scalar(@loader_warnings), $warn_count, "Expected loader warning")
302                     or diag @loader_warnings;
303                 is(grep(/loader_test9 has no primary key/i, @loader_warnings), 1,
304                      "Missing PK warning");
305             }
306         }
307         else {
308             SKIP: {
309                 is scalar(@loader_warnings), $warn_count, 'Correct number of warnings'
310                     or diag @loader_warnings;
311                 skip "not testing standard sources", 1;
312             }
313         }
314     }
315
316     exit if ($file_count||0) != $expected_count;
317    
318     return $schema_class;
319 }
320
321 sub test_schema {
322     my $self = shift;
323     my $schema_class = shift;
324
325     my $conn = $schema_class->clone;
326
327     ($self->{before_tests_run} || sub {})->($conn);
328
329     my ($monikers, $classes) = $self->monikers_and_classes($schema_class);
330
331     my $moniker1 = $monikers->{loader_test1s};
332     my $class1   = $classes->{loader_test1s};
333     my $rsobj1   = $conn->resultset($moniker1);
334     check_no_duplicate_unique_constraints($class1);
335
336     my $moniker2 = $monikers->{loader_test2};
337     my $class2   = $classes->{loader_test2};
338     my $rsobj2   = $conn->resultset($moniker2);
339     check_no_duplicate_unique_constraints($class2);
340
341     my $moniker23 = $monikers->{LOADER_test23} || $monikers->{loader_test23};
342     my $class23   = $classes->{LOADER_test23}  || $classes->{loader_test23};
343     my $rsobj23   = $conn->resultset($moniker1);
344
345     my $moniker24 = $monikers->{LoAdEr_test24} || $monikers->{loader_test24};
346     my $class24   = $classes->{LoAdEr_test24}  || $classes->{loader_test24};
347     my $rsobj24   = $conn->resultset($moniker2);
348
349     my $moniker35 = $monikers->{loader_test35};
350     my $class35   = $classes->{loader_test35};
351     my $rsobj35   = $conn->resultset($moniker35);
352
353     isa_ok( $rsobj1, "DBIx::Class::ResultSet" );
354     isa_ok( $rsobj2, "DBIx::Class::ResultSet" );
355     isa_ok( $rsobj23, "DBIx::Class::ResultSet" );
356     isa_ok( $rsobj24, "DBIx::Class::ResultSet" );
357     isa_ok( $rsobj35, "DBIx::Class::ResultSet" );
358
359     my @columns_lt2 = $class2->columns;
360     is_deeply( \@columns_lt2, [ qw/id dat dat2 set_primary_key can dbix_class_testcomponent testcomponent_fqn meta/ ], "Column Ordering" );
361
362     is $class2->column_info('can')->{accessor}, 'caught_collision_can',
363         'accessor for column name that conflicts with a UNIVERSAL method renamed based on col_collision_map';
364
365     is $class2->column_info('set_primary_key')->{accessor}, undef,
366         'accessor for column name that conflicts with a result base class method removed';
367
368     is $class2->column_info('dbix_class_testcomponent')->{accessor}, undef,
369         'accessor for column name that conflicts with a component class method removed';
370
371     is $class2->column_info('testcomponent_fqn')->{accessor}, undef,
372         'accessor for column name that conflicts with a fully qualified component class method removed';
373
374     is $class2->column_info('meta')->{accessor}, undef,
375         'accessor for column name that conflicts with Moose removed';
376
377     my %uniq1 = $class1->unique_constraints;
378     my $uniq1_test = 0;
379     foreach my $ucname (keys %uniq1) {
380         my $cols_arrayref = $uniq1{$ucname};
381         if(@$cols_arrayref == 1 && $cols_arrayref->[0] eq 'dat') {
382            $uniq1_test = 1;
383            last;
384         }
385     }
386     ok($uniq1_test, "Unique constraint");
387
388     is($moniker1, 'LoaderTest1', 'moniker singularisation');
389
390     my %uniq2 = $class2->unique_constraints;
391     my $uniq2_test = 0;
392     foreach my $ucname (keys %uniq2) {
393         my $cols_arrayref = $uniq2{$ucname};
394         if(@$cols_arrayref == 2
395            && $cols_arrayref->[0] eq 'dat2'
396            && $cols_arrayref->[1] eq 'dat') {
397             $uniq2_test = 2;
398             last;
399         }
400     }
401     ok($uniq2_test, "Multi-col unique constraint");
402
403     is($moniker2, 'LoaderTest2X', "moniker_map testing");
404
405     SKIP: {
406         can_ok( $class1, 'test_additional_base' )
407             or skip "Pre-requisite test failed", 1;
408         is( $class1->test_additional_base, "test_additional_base",
409             "Additional Base method" );
410     }
411
412     SKIP: {
413         can_ok( $class1, 'test_additional_base_override' )
414             or skip "Pre-requisite test failed", 1;
415         is( $class1->test_additional_base_override,
416             "test_left_base_override",
417             "Left Base overrides Additional Base method" );
418     }
419
420     SKIP: {
421         can_ok( $class1, 'test_additional_base_additional' )
422             or skip "Pre-requisite test failed", 1;
423         is( $class1->test_additional_base_additional, "test_additional",
424             "Additional Base can use Additional package method" );
425     }
426
427     SKIP: {
428         can_ok( $class1, 'dbix_class_testcomponent' )
429             or skip "Pre-requisite test failed", 1;
430         is( $class1->dbix_class_testcomponent,
431             'dbix_class_testcomponent works',
432             'Additional Component' );
433     }
434
435     is try { $class2->dbix_class_testcomponentformap }, 'dbix_class_testcomponentformap works',
436         'component from result_component_map';
437
438     isnt try { $class1->dbix_class_testcomponentformap }, 'dbix_class_testcomponentformap works',
439         'component from result_component_map not added to not mapped Result';
440
441     is try { $class1->testcomponent_fqn }, 'TestComponentFQN works',
442         'fully qualified component class';
443
444     is try { $class1->testcomponentformap_fqn }, 'TestComponentForMapFQN works',
445         'fully qualified component class from result_component_map';
446
447     isnt try { $class2->testcomponentformap_fqn }, 'TestComponentForMapFQN works',
448         'fully qualified component class from result_component_map not added to not mapped Result';
449
450     SKIP: {
451         can_ok( $class1, 'loader_test1_classmeth' )
452             or skip "Pre-requisite test failed", 1;
453         is( $class1->loader_test1_classmeth, 'all is well', 'Class method' );
454     }
455
456     ok( $class1->column_info('id')->{is_auto_increment}, 'is_auto_increment detection' );
457
458     my $obj = try { $rsobj1->find(1) };
459
460     is( try { $obj->id },  1, "Find got the right row" );
461     is( try { $obj->dat }, "foo", "Column value" );
462     is( $rsobj2->count, 4, "Count" );
463     my $saved_id;
464     eval {
465         my $new_obj1 = $rsobj1->create({ dat => 'newthing' });
466         $saved_id = $new_obj1->id;
467     };
468     ok(!$@, "Inserting new record using a PK::Auto key didn't die") or diag $@;
469     ok($saved_id, "Got PK::Auto-generated id");
470
471     my $new_obj1 = $rsobj1->search({ dat => 'newthing' })->first;
472     ok($new_obj1, "Found newly inserted PK::Auto record");
473     is($new_obj1->id, $saved_id, "Correct PK::Auto-generated id");
474
475     my ($obj2) = $rsobj2->search({ dat => 'bbb' })->first;
476     is( $obj2->id, 2 );
477
478     SKIP: {
479         skip 'no DEFAULT on Access', 7 if $self->{vendor} eq 'Access';
480
481         is(
482             $class35->column_info('a_varchar')->{default_value}, 'foo',
483             'constant character default',
484         );
485
486         is(
487             $class35->column_info('an_int')->{default_value}, 42,
488             'constant integer default',
489         );
490
491         is(
492             $class35->column_info('a_negative_int')->{default_value}, -42,
493             'constant negative integer default',
494         );
495
496         cmp_ok(
497             $class35->column_info('a_double')->{default_value}, '==', 10.555,
498             'constant numeric default',
499         );
500
501         cmp_ok(
502             $class35->column_info('a_negative_double')->{default_value}, '==', -10.555,
503             'constant negative numeric default',
504         );
505
506         my $function_default = $class35->column_info('a_function')->{default_value};
507
508         isa_ok( $function_default, 'SCALAR', 'default_value for function default' );
509         is_deeply(
510             $function_default, \$self->{default_function},
511             'default_value for function default is correct'
512         );
513     }
514
515     SKIP: {
516         skip $self->{skip_rels}, 120 if $self->{skip_rels};
517
518         my $moniker3 = $monikers->{loader_test3};
519         my $class3   = $classes->{loader_test3};
520         my $rsobj3   = $conn->resultset($moniker3);
521
522         my $moniker4 = $monikers->{loader_test4};
523         my $class4   = $classes->{loader_test4};
524         my $rsobj4   = $conn->resultset($moniker4);
525
526         my $moniker5 = $monikers->{loader_test5};
527         my $class5   = $classes->{loader_test5};
528         my $rsobj5   = $conn->resultset($moniker5);
529
530         my $moniker6 = $monikers->{loader_test6};
531         my $class6   = $classes->{loader_test6};
532         my $rsobj6   = $conn->resultset($moniker6);
533
534         my $moniker7 = $monikers->{loader_test7};
535         my $class7   = $classes->{loader_test7};
536         my $rsobj7   = $conn->resultset($moniker7);
537
538         my $moniker8 = $monikers->{loader_test8};
539         my $class8   = $classes->{loader_test8};
540         my $rsobj8   = $conn->resultset($moniker8);
541
542         my $moniker9 = $monikers->{loader_test9};
543         my $class9   = $classes->{loader_test9};
544         my $rsobj9   = $conn->resultset($moniker9);
545
546         my $moniker16 = $monikers->{loader_test16};
547         my $class16   = $classes->{loader_test16};
548         my $rsobj16   = $conn->resultset($moniker16);
549
550         my $moniker17 = $monikers->{loader_test17};
551         my $class17   = $classes->{loader_test17};
552         my $rsobj17   = $conn->resultset($moniker17);
553
554         my $moniker18 = $monikers->{loader_test18};
555         my $class18   = $classes->{loader_test18};
556         my $rsobj18   = $conn->resultset($moniker18);
557
558         my $moniker19 = $monikers->{loader_test19};
559         my $class19   = $classes->{loader_test19};
560         my $rsobj19   = $conn->resultset($moniker19);
561
562         my $moniker20 = $monikers->{loader_test20};
563         my $class20   = $classes->{loader_test20};
564         my $rsobj20   = $conn->resultset($moniker20);
565
566         my $moniker21 = $monikers->{loader_test21};
567         my $class21   = $classes->{loader_test21};
568         my $rsobj21   = $conn->resultset($moniker21);
569
570         my $moniker22 = $monikers->{loader_test22};
571         my $class22   = $classes->{loader_test22};
572         my $rsobj22   = $conn->resultset($moniker22);
573
574         my $moniker25 = $monikers->{loader_test25};
575         my $class25   = $classes->{loader_test25};
576         my $rsobj25   = $conn->resultset($moniker25);
577
578         my $moniker26 = $monikers->{loader_test26};
579         my $class26   = $classes->{loader_test26};
580         my $rsobj26   = $conn->resultset($moniker26);
581
582         my $moniker27 = $monikers->{loader_test27};
583         my $class27   = $classes->{loader_test27};
584         my $rsobj27   = $conn->resultset($moniker27);
585
586         my $moniker28 = $monikers->{loader_test28};
587         my $class28   = $classes->{loader_test28};
588         my $rsobj28   = $conn->resultset($moniker28);
589
590         my $moniker29 = $monikers->{loader_test29};
591         my $class29   = $classes->{loader_test29};
592         my $rsobj29   = $conn->resultset($moniker29);
593
594         my $moniker31 = $monikers->{loader_test31};
595         my $class31   = $classes->{loader_test31};
596         my $rsobj31   = $conn->resultset($moniker31);
597
598         my $moniker32 = $monikers->{loader_test32};
599         my $class32   = $classes->{loader_test32};
600         my $rsobj32   = $conn->resultset($moniker32);
601
602         my $moniker33 = $monikers->{loader_test33};
603         my $class33   = $classes->{loader_test33};
604         my $rsobj33   = $conn->resultset($moniker33);
605
606         my $moniker34 = $monikers->{loader_test34};
607         my $class34   = $classes->{loader_test34};
608         my $rsobj34   = $conn->resultset($moniker34);
609
610         my $moniker36 = $monikers->{loader_test36};
611         my $class36   = $classes->{loader_test36};
612         my $rsobj36   = $conn->resultset($moniker36);
613         
614         isa_ok( $rsobj3, "DBIx::Class::ResultSet" );
615         isa_ok( $rsobj4, "DBIx::Class::ResultSet" );
616         isa_ok( $rsobj5, "DBIx::Class::ResultSet" );
617         isa_ok( $rsobj6, "DBIx::Class::ResultSet" );
618         isa_ok( $rsobj7, "DBIx::Class::ResultSet" );
619         isa_ok( $rsobj8, "DBIx::Class::ResultSet" );
620         isa_ok( $rsobj9, "DBIx::Class::ResultSet" );
621         isa_ok( $rsobj16, "DBIx::Class::ResultSet" );
622         isa_ok( $rsobj17, "DBIx::Class::ResultSet" );
623         isa_ok( $rsobj18, "DBIx::Class::ResultSet" );
624         isa_ok( $rsobj19, "DBIx::Class::ResultSet" );
625         isa_ok( $rsobj20, "DBIx::Class::ResultSet" );
626         isa_ok( $rsobj21, "DBIx::Class::ResultSet" );
627         isa_ok( $rsobj22, "DBIx::Class::ResultSet" );
628         isa_ok( $rsobj25, "DBIx::Class::ResultSet" );
629         isa_ok( $rsobj26, "DBIx::Class::ResultSet" );
630         isa_ok( $rsobj27, "DBIx::Class::ResultSet" );
631         isa_ok( $rsobj28, "DBIx::Class::ResultSet" );
632         isa_ok( $rsobj29, "DBIx::Class::ResultSet" );
633         isa_ok( $rsobj31, "DBIx::Class::ResultSet" );
634         isa_ok( $rsobj32, "DBIx::Class::ResultSet" );
635         isa_ok( $rsobj33, "DBIx::Class::ResultSet" );
636         isa_ok( $rsobj34, "DBIx::Class::ResultSet" );
637         isa_ok( $rsobj36, "DBIx::Class::ResultSet" );
638
639         # basic rel test
640         my $obj4 = try { $rsobj4->find(123) } || $rsobj4->search({ id => 123 })->first;
641         isa_ok( try { $obj4->fkid_singular }, $class3);
642
643         # test renaming rel that conflicts with a class method
644         ok ($obj4->has_relationship('belongs_to_rel'), 'relationship name that conflicts with a method renamed');
645
646         isa_ok( try { $obj4->belongs_to_rel }, $class3);
647
648         ok ($obj4->has_relationship('caught_rel_collision_set_primary_key'),
649             'relationship name that conflicts with a method renamed based on rel_collision_map');
650         isa_ok( try { $obj4->caught_rel_collision_set_primary_key }, $class3);
651
652         ok($class4->column_info('fkid')->{is_foreign_key}, 'Foreign key detected');
653
654         my $obj3 = try { $rsobj3->find(1) } || $rsobj3->search({ id => 1 })->first;
655         my $rs_rel4 = try { $obj3->search_related('loader_test4zes') };
656         isa_ok( try { $rs_rel4->first }, $class4);
657
658         is( $class4->column_info('crumb_crisp_coating')->{accessor},  'trivet',
659             'col_accessor_map is being run' );
660
661         # check rel naming with prepositions
662         ok ($rsobj4->result_source->has_relationship('loader_test5s_to'),
663             "rel with preposition 'to' pluralized correctly");
664
665         ok ($rsobj4->result_source->has_relationship('loader_test5s_from'),
666             "rel with preposition 'from' pluralized correctly");
667
668         # check default relationship attributes
669         is try { $rsobj3->result_source->relationship_info('loader_test4zes')->{attrs}{cascade_delete} }, 0,
670             'cascade_delete => 0 on has_many by default';
671
672         is try { $rsobj3->result_source->relationship_info('loader_test4zes')->{attrs}{cascade_copy} }, 0,
673             'cascade_copy => 0 on has_many by default';
674
675         ok ((not try { exists $rsobj3->result_source->relationship_info('loader_test4zes')->{attrs}{on_delete} }),
676             'has_many does not have on_delete');
677
678         ok ((not try { exists $rsobj3->result_source->relationship_info('loader_test4zes')->{attrs}{on_update} }),
679             'has_many does not have on_update');
680
681         ok ((not try { exists $rsobj3->result_source->relationship_info('loader_test4zes')->{attrs}{is_deferrable} }),
682             'has_many does not have is_deferrable');
683
684         is try { $rsobj4->result_source->relationship_info('fkid_singular')->{attrs}{on_delete} }, 'CASCADE',
685             "on_delete => 'CASCADE' on belongs_to by default";
686
687         is try { $rsobj4->result_source->relationship_info('fkid_singular')->{attrs}{on_update} }, 'CASCADE',
688             "on_update => 'CASCADE' on belongs_to by default";
689
690         is try { $rsobj4->result_source->relationship_info('fkid_singular')->{attrs}{is_deferrable} }, 1,
691             "is_deferrable => 1 on belongs_to by default";
692
693         ok ((not try { exists $rsobj4->result_source->relationship_info('fkid_singular')->{attrs}{cascade_delete} }),
694             'belongs_to does not have cascade_delete');
695
696         ok ((not try { exists $rsobj4->result_source->relationship_info('fkid_singular')->{attrs}{cascade_copy} }),
697             'belongs_to does not have cascade_copy');
698
699         is try { $rsobj27->result_source->relationship_info('loader_test28')->{attrs}{cascade_delete} }, 0,
700             'cascade_delete => 0 on might_have by default';
701
702         is try { $rsobj27->result_source->relationship_info('loader_test28')->{attrs}{cascade_copy} }, 0,
703             'cascade_copy => 0 on might_have by default';
704
705         ok ((not try { exists $rsobj27->result_source->relationship_info('loader_test28')->{attrs}{on_delete} }),
706             'might_have does not have on_delete');
707
708         ok ((not try { exists $rsobj27->result_source->relationship_info('loader_test28')->{attrs}{on_update} }),
709             'might_have does not have on_update');
710
711         ok ((not try { exists $rsobj27->result_source->relationship_info('loader_test28')->{attrs}{is_deferrable} }),
712             'might_have does not have is_deferrable');
713
714         # find on multi-col pk
715         if ($conn->_loader->preserve_case) {
716             my $obj5 = $rsobj5->find({id1 => 1, iD2 => 1});
717             is $obj5->i_d2, 1, 'Find on multi-col PK';
718         }
719         else {
720             my $obj5 = $rsobj5->find({id1 => 1, id2 => 1});
721             is $obj5->id2, 1, 'Find on multi-col PK';
722         }
723
724         # mulit-col fk def
725         my $obj6 = try { $rsobj6->find(1) } || $rsobj6->search({ id => 1 })->first;
726         isa_ok( try { $obj6->loader_test2 }, $class2);
727         isa_ok( try { $obj6->loader_test5 }, $class5);
728
729         ok($class6->column_info('loader_test2_id')->{is_foreign_key}, 'Foreign key detected');
730         ok($class6->column_info('id')->{is_foreign_key}, 'Foreign key detected');
731
732         my $id2_info = eval { $class6->column_info('id2') } ||
733                         $class6->column_info('Id2');
734         ok($id2_info->{is_foreign_key}, 'Foreign key detected');
735
736         # fk that references a non-pk key (UNIQUE)
737         my $obj8 = try { $rsobj8->find(1) } || $rsobj8->search({ id => 1 })->first;
738         isa_ok( try { $obj8->loader_test7 }, $class7);
739
740         ok($class8->column_info('loader_test7')->{is_foreign_key}, 'Foreign key detected');
741
742         # test double-fk 17 ->-> 16
743         my $obj17 = try { $rsobj17->find(33) } || $rsobj17->search({ id => 33 })->first;
744
745         my $rs_rel16_one = try { $obj17->loader16_one };
746         isa_ok($rs_rel16_one, $class16);
747         is(try { $rs_rel16_one->dat }, 'y16', "Multiple FKs to same table");
748
749         ok($class17->column_info('loader16_one')->{is_foreign_key}, 'Foreign key detected');
750
751         my $rs_rel16_two = try { $obj17->loader16_two };
752         isa_ok($rs_rel16_two, $class16);
753         is(try { $rs_rel16_two->dat }, 'z16', "Multiple FKs to same table");
754
755         ok($class17->column_info('loader16_two')->{is_foreign_key}, 'Foreign key detected');
756
757         my $obj16 = try { $rsobj16->find(2) } || $rsobj16->search({ id => 2 })->first;
758         my $rs_rel17 = try { $obj16->search_related('loader_test17_loader16_ones') };
759         isa_ok(try { $rs_rel17->first }, $class17);
760         is(try { $rs_rel17->first->id }, 3, "search_related with multiple FKs from same table");
761         
762         # XXX test m:m 18 <- 20 -> 19
763         ok($class20->column_info('parent')->{is_foreign_key}, 'Foreign key detected');
764         ok($class20->column_info('child')->{is_foreign_key}, 'Foreign key detected');
765         
766         # XXX test double-fk m:m 21 <- 22 -> 21
767         ok($class22->column_info('parent')->{is_foreign_key}, 'Foreign key detected');
768         ok($class22->column_info('child')->{is_foreign_key}, 'Foreign key detected');
769
770         # test double multi-col fk 26 -> 25
771         my $obj26 = try { $rsobj26->find(33) } || $rsobj26->search({ id => 33 })->first;
772
773         my $rs_rel25_one = try { $obj26->loader_test25_id_rel1 };
774         isa_ok($rs_rel25_one, $class25);
775         is(try { $rs_rel25_one->dat }, 'x25', "Multiple multi-col FKs to same table");
776
777         ok($class26->column_info('id')->{is_foreign_key}, 'Foreign key detected');
778         ok($class26->column_info('rel1')->{is_foreign_key}, 'Foreign key detected');
779         ok($class26->column_info('rel2')->{is_foreign_key}, 'Foreign key detected');
780
781         my $rs_rel25_two = try { $obj26->loader_test25_id_rel2 };
782         isa_ok($rs_rel25_two, $class25);
783         is(try { $rs_rel25_two->dat }, 'y25', "Multiple multi-col FKs to same table");
784
785         my $obj25 = try { $rsobj25->find(3,42) } || $rsobj25->search({ id1 => 3, id2 => 42 })->first;
786         my $rs_rel26 = try { $obj25->search_related('loader_test26_id_rel1s') };
787         isa_ok(try { $rs_rel26->first }, $class26);
788         is(try { $rs_rel26->first->id }, 3, "search_related with multiple multi-col FKs from same table");
789
790         # test one-to-one rels
791         my $obj27 = try { $rsobj27->find(1) } || $rsobj27->search({ id => 1 })->first;
792         my $obj28 = try { $obj27->loader_test28 };
793         isa_ok($obj28, $class28);
794         is(try { $obj28->get_column('id') }, 1, "One-to-one relationship with PRIMARY FK");
795
796         ok($class28->column_info('id')->{is_foreign_key}, 'Foreign key detected');
797
798         my $obj29 = try { $obj27->loader_test29 };
799         isa_ok($obj29, $class29);
800         is(try { $obj29->id }, 1, "One-to-one relationship with UNIQUE FK");
801
802         ok($class29->column_info('fk')->{is_foreign_key}, 'Foreign key detected');
803
804         $obj27 = try { $rsobj27->find(2) } || $rsobj27->search({ id => 2 })->first;
805         is(try { $obj27->loader_test28 }, undef, "Undef for missing one-to-one row");
806         is(try { $obj27->loader_test29 }, undef, "Undef for missing one-to-one row");
807
808         # test outer join for nullable referring columns:
809         is $class32->column_info('rel2')->{is_nullable}, 1,
810           'is_nullable detection';
811
812         ok($class32->column_info('rel1')->{is_foreign_key}, 'Foreign key detected');
813         ok($class32->column_info('rel2')->{is_foreign_key}, 'Foreign key detected');
814         
815         my $obj32 = try { $rsobj32->find(1, { prefetch => [qw/rel1 rel2/] }) }
816             || try { $rsobj32->search({ id => 1 }, { prefetch => [qw/rel1 rel2/] })->first }
817             || $rsobj32->search({ id => 1 })->first;
818
819         my $obj34 = eval { $rsobj34->find(1, { prefetch => [qw/loader_test33_id_rel1 loader_test33_id_rel2/] }) }
820             || eval { $rsobj34->search({ id => 1 }, { prefetch => [qw/loader_test33_id_rel1 loader_test33_id_rel2/] })->first }
821             || $rsobj34->search({ id => 1 })->first;
822         diag $@ if $@;
823
824         isa_ok($obj32,$class32);
825         isa_ok($obj34,$class34);
826
827         ok($class34->column_info('id')->{is_foreign_key}, 'Foreign key detected');
828         ok($class34->column_info('rel1')->{is_foreign_key}, 'Foreign key detected');
829         ok($class34->column_info('rel2')->{is_foreign_key}, 'Foreign key detected');
830
831         my $rs_rel31_one = try { $obj32->rel1 };
832         my $rs_rel31_two = try { $obj32->rel2 };
833         isa_ok($rs_rel31_one, $class31);
834         is($rs_rel31_two, undef);
835
836         my $rs_rel33_one = try { $obj34->loader_test33_id_rel1 };
837         my $rs_rel33_two = try { $obj34->loader_test33_id_rel2 };
838
839         isa_ok($rs_rel33_one, $class33);
840         isa_ok($rs_rel33_two, $class33);
841
842         # from Chisel's tests...
843         my $moniker10 = $monikers->{loader_test10};
844         my $class10   = $classes->{loader_test10};
845         my $rsobj10   = $conn->resultset($moniker10);
846
847         my $moniker11 = $monikers->{loader_test11};
848         my $class11   = $classes->{loader_test11};
849         my $rsobj11   = $conn->resultset($moniker11);
850
851         isa_ok( $rsobj10, "DBIx::Class::ResultSet" );
852         isa_ok( $rsobj11, "DBIx::Class::ResultSet" );
853
854         ok($class10->column_info('loader_test11')->{is_foreign_key}, 'Foreign key detected');
855         ok($class11->column_info('loader_test10')->{is_foreign_key}, 'Foreign key detected');
856
857         my $obj10 = $rsobj10->create({ subject => 'xyzzy' });
858
859         $obj10->update();
860         ok( defined $obj10, 'Create row' );
861
862         my $obj11 = $rsobj11->create({ loader_test10 => (try { $obj10->id() } || $obj10->id10) });
863         $obj11->update();
864         ok( defined $obj11, 'Create related row' );
865
866         eval {
867             my $obj10_2 = $obj11->loader_test10;
868             $obj10_2->update({ loader_test11 => $obj11->id11 });
869         };
870         diag $@ if $@;
871         ok(!$@, "Setting up circular relationship");
872
873         SKIP: {
874             skip 'Previous eval block failed', 3 if $@;
875     
876             my $results = $rsobj10->search({ subject => 'xyzzy' });
877             is( $results->count(), 1, 'No duplicate row created' );
878
879             my $obj10_3 = $results->first();
880             isa_ok( $obj10_3, $class10 );
881             is( $obj10_3->loader_test11()->id(), $obj11->id(),
882                 'Circular rel leads back to same row' );
883         }
884
885         SKIP: {
886             skip 'This vendor cannot do inline relationship definitions', 11
887                 if $self->{no_inline_rels};
888
889             my $moniker12 = $monikers->{loader_test12};
890             my $class12   = $classes->{loader_test12};
891             my $rsobj12   = $conn->resultset($moniker12);
892
893             my $moniker13 = $monikers->{loader_test13};
894             my $class13   = $classes->{loader_test13};
895             my $rsobj13   = $conn->resultset($moniker13);
896
897             isa_ok( $rsobj12, "DBIx::Class::ResultSet" ); 
898             isa_ok( $rsobj13, "DBIx::Class::ResultSet" );
899
900             ok($class13->column_info('id')->{is_foreign_key}, 'Foreign key detected');
901             ok($class13->column_info('loader_test12')->{is_foreign_key}, 'Foreign key detected');
902             ok($class13->column_info('dat')->{is_foreign_key}, 'Foreign key detected');
903
904             my $obj13 = try { $rsobj13->find(1) } || $rsobj13->search({ id => 1 })->first;
905             isa_ok( $obj13->id, $class12 );
906             isa_ok( $obj13->loader_test12, $class12);
907             isa_ok( $obj13->dat, $class12);
908
909             my $obj12 = try { $rsobj12->find(1) } || $rsobj12->search({ id => 1 })->first;
910             isa_ok( try { $obj12->loader_test13 }, $class13 );
911
912             # relname is preserved when another fk is added
913
914             skip 'Sybase cannot add FKs via ALTER TABLE', 2
915                 if $self->{vendor} eq 'sybase';
916
917             {
918                 local $SIG{__WARN__} = sub { warn @_ unless $_[0] =~ /invalidates \d+ active statement/ };
919                 $conn->storage->disconnect; # for mssql and access
920             }
921
922             isa_ok try { $rsobj3->find(1)->loader_test4zes }, 'DBIx::Class::ResultSet';
923
924             $conn->storage->disconnect; # for access
925
926             $conn->storage->dbh->do('ALTER TABLE loader_test4 ADD fkid2 INTEGER REFERENCES loader_test3 (id)');
927
928             $conn->storage->disconnect; # for firebird
929
930             $self->rescan_without_warnings($conn);
931
932             isa_ok try { $rsobj3->find(1)->loader_test4zes }, 'DBIx::Class::ResultSet',
933                 'relationship name preserved when another foreign key is added in remote table';
934         }
935
936         SKIP: {
937             skip 'This vendor cannot do out-of-line implicit rel defs', 4
938                 if $self->{no_implicit_rels};
939             my $moniker14 = $monikers->{loader_test14};
940             my $class14   = $classes->{loader_test14};
941             my $rsobj14   = $conn->resultset($moniker14);
942
943             my $moniker15 = $monikers->{loader_test15};
944             my $class15   = $classes->{loader_test15};
945             my $rsobj15   = $conn->resultset($moniker15);
946
947             isa_ok( $rsobj14, "DBIx::Class::ResultSet" ); 
948             isa_ok( $rsobj15, "DBIx::Class::ResultSet" );
949
950             ok($class15->column_info('loader_test14')->{is_foreign_key}, 'Foreign key detected');
951
952             my $obj15 = try { $rsobj15->find(1) } || $rsobj15->search({ id => 1 })->first;
953             isa_ok( $obj15->loader_test14, $class14 );
954         }
955     }
956
957     # test custom_column_info and datetime_timezone/datetime_locale
958     {
959         my $class35 = $classes->{loader_test35};
960         my $class36 = $classes->{loader_test36};
961
962         ok($class35->column_info('an_int')->{is_numeric}, 'custom_column_info');
963
964         is($class36->column_info('a_date')->{locale},'de_DE','datetime_locale');
965         is($class36->column_info('a_date')->{timezone},'Europe/Berlin','datetime_timezone');
966
967         ok($class36->column_info('b_char_as_data')->{inflate_datetime},'custom_column_info');
968         is($class36->column_info('b_char_as_data')->{locale},'de_DE','datetime_locale');
969         is($class36->column_info('b_char_as_data')->{timezone},'Europe/Berlin','datetime_timezone');
970
971         ok($class36->column_info('c_char_as_data')->{inflate_date},'custom_column_info');
972         is($class36->column_info('c_char_as_data')->{locale},'de_DE','datetime_locale');
973         is($class36->column_info('c_char_as_data')->{timezone},'Europe/Berlin','datetime_timezone');
974     }
975
976     # rescan and norewrite test
977     {
978         my @statements_rescan = (
979             qq{
980                 CREATE TABLE loader_test30 (
981                     id INTEGER NOT NULL PRIMARY KEY,
982                     loader_test2 INTEGER NOT NULL,
983                     FOREIGN KEY (loader_test2) REFERENCES loader_test2 (id)
984                 ) $self->{innodb}
985             },
986             q{ INSERT INTO loader_test30 (id,loader_test2) VALUES(123, 1) },
987             q{ INSERT INTO loader_test30 (id,loader_test2) VALUES(321, 2) },
988         );
989
990         # get md5
991         my $digest  = Digest::MD5->new;
992
993         my $find_cb = sub {
994             return if -d;
995             return if /^(?:LoaderTest30|LoaderTest1|LoaderTest2X)\.pm\z/;
996
997             open my $fh, '<', $_ or die "Could not open $_ for reading: $!";
998             binmode $fh;
999             $digest->addfile($fh);
1000         };
1001
1002         find $find_cb, $DUMP_DIR;
1003
1004 #        system "rm -f /tmp/before_rescan/* /tmp/after_rescan/*";
1005 #        system "cp $tdir/common_dump/DBIXCSL_Test/Schema/*.pm /tmp/before_rescan";
1006
1007         my $before_digest = $digest->b64digest;
1008
1009         $conn->storage->disconnect; # needed for Firebird and Informix
1010         my $dbh = $self->dbconnect(1);
1011         $dbh->do($_) for @statements_rescan;
1012         $dbh->disconnect;
1013
1014         sleep 1;
1015
1016         my @new = $self->rescan_without_warnings($conn);
1017
1018         is_deeply(\@new, [ qw/LoaderTest30/ ], "Rescan");
1019
1020 #        system "cp t/_common_dump/DBIXCSL_Test/Schema/*.pm /tmp/after_rescan";
1021
1022         $digest = Digest::MD5->new;
1023         find $find_cb, $DUMP_DIR;
1024         my $after_digest = $digest->b64digest;
1025
1026         is $before_digest, $after_digest,
1027             'dumped files are not rewritten when there is no modification';
1028
1029         my $rsobj30   = $conn->resultset('LoaderTest30');
1030         isa_ok($rsobj30, 'DBIx::Class::ResultSet');
1031
1032         SKIP: {
1033             skip 'no rels', 2 if $self->{skip_rels};
1034
1035             my $obj30 = try { $rsobj30->find(123) } || $rsobj30->search({ id => 123 })->first;
1036             isa_ok( $obj30->loader_test2, $class2);
1037
1038             ok($rsobj30->result_source->column_info('loader_test2')->{is_foreign_key},
1039                'Foreign key detected');
1040         }
1041
1042         $conn->storage->disconnect; # for Firebird
1043         $conn->storage->dbh->do("DROP TABLE loader_test30");
1044
1045         @new = $self->rescan_without_warnings($conn);
1046
1047         is_deeply(\@new, [], 'no new tables on rescan');
1048
1049         throws_ok { $conn->resultset('LoaderTest30') }
1050             qr/Can't find source/,
1051             'source unregistered for dropped table after rescan';
1052     }
1053
1054     $self->test_data_types($conn);
1055
1056     # run extra tests
1057     $self->{extra}{run}->($conn, $monikers, $classes, $self) if $self->{extra}{run};
1058
1059     $self->test_preserve_case($conn);
1060
1061     $self->drop_tables unless $ENV{SCHEMA_LOADER_TESTS_NOCLEANUP};
1062
1063     $conn->storage->disconnect;
1064 }
1065
1066 sub test_data_types {
1067     my ($self, $conn) = @_;
1068
1069     SKIP: {
1070         if (my $test_count = $self->{data_type_tests}{test_count}) {
1071             if ($self->{vendor} eq 'mssql' && $conn->storage->dbh->{Driver}{Name} eq 'Sybase') {
1072                 skip 'DBD::Sybase does not work with the data_type tests on latest SQL Server', $test_count;
1073             }
1074
1075             my $data_type_tests = $self->{data_type_tests};
1076
1077             foreach my $moniker (@{ $data_type_tests->{table_monikers} }) {
1078                 my $columns = $data_type_tests->{columns}{$moniker};
1079
1080                 my $rsrc = $conn->resultset($moniker)->result_source;
1081
1082                 while (my ($col_name, $expected_info) = each %$columns) {
1083                     my %info = %{ $rsrc->column_info($col_name) };
1084                     delete @info{qw/is_nullable timezone locale sequence/};
1085
1086                     my $text_col_def = dumper_squashed \%info;
1087
1088                     my $text_expected_info = dumper_squashed $expected_info;
1089
1090                     is_deeply \%info, $expected_info,
1091                         "test column $col_name has definition: $text_col_def expecting: $text_expected_info";
1092                 }
1093             }
1094         }
1095     }
1096 }
1097
1098 sub test_preserve_case {
1099     my ($self, $conn) = @_;
1100
1101     my ($oqt, $cqt) = $self->get_oqt_cqt(always => 1); # open quote, close quote
1102
1103     my $dbh = $self->dbconnect;
1104
1105     $dbh->do($_) for (
1106 qq|
1107     CREATE TABLE ${oqt}LoaderTest40${cqt} (
1108         ${oqt}Id${cqt} INTEGER NOT NULL PRIMARY KEY,
1109         ${oqt}Foo3Bar${cqt} VARCHAR(100) NOT NULL
1110     ) $self->{innodb}
1111 |,
1112 qq|
1113     CREATE TABLE ${oqt}LoaderTest41${cqt} (
1114         ${oqt}Id${cqt} INTEGER NOT NULL PRIMARY KEY,
1115         ${oqt}LoaderTest40Id${cqt} INTEGER,
1116         FOREIGN KEY (${oqt}LoaderTest40Id${cqt}) REFERENCES ${oqt}LoaderTest40${cqt} (${oqt}Id${cqt})
1117     ) $self->{innodb}
1118 |,
1119 qq| INSERT INTO ${oqt}LoaderTest40${cqt} VALUES (1, 'foo') |,
1120 qq| INSERT INTO ${oqt}LoaderTest41${cqt} VALUES (1, 1) |,
1121     );
1122     $conn->storage->disconnect;
1123
1124     local $conn->_loader->{preserve_case} = 1;
1125     $conn->_loader->_setup;
1126
1127     $self->rescan_without_warnings($conn);
1128
1129     if (not $self->{skip_rels}) {
1130         is try { $conn->resultset('LoaderTest41')->find(1)->loader_test40->foo3_bar }, 'foo',
1131             'rel and accessor for mixed-case column name in mixed case table';
1132     }
1133     else {
1134         is try { $conn->resultset('LoaderTest40')->find(1)->foo3_bar }, 'foo',
1135             'accessor for mixed-case column name in mixed case table';
1136     }
1137 }
1138
1139 sub monikers_and_classes {
1140     my ($self, $schema_class) = @_;
1141     my ($monikers, $classes);
1142
1143     foreach my $source_name ($schema_class->sources) {
1144         my $table_name = $schema_class->source($source_name)->from;
1145
1146         $table_name = $$table_name if ref $table_name;
1147
1148         $monikers->{$table_name} = $source_name;
1149         $classes->{$table_name} = $schema_class . q{::} . $source_name;
1150
1151         # some DBs (Firebird) uppercase everything
1152         $monikers->{lc $table_name} = $source_name;
1153         $classes->{lc $table_name} = $schema_class . q{::} . $source_name;
1154     }
1155
1156     return ($monikers, $classes);
1157 }
1158
1159 sub check_no_duplicate_unique_constraints {
1160     my ($class) = @_;
1161
1162     # unique_constraints() automatically includes the PK, if any
1163     my %uc_cols;
1164     ++$uc_cols{ join ", ", @$_ }
1165         for values %{ { $class->unique_constraints } };
1166     my $dup_uc = grep { $_ > 1 } values %uc_cols;
1167
1168     is($dup_uc, 0, "duplicate unique constraints ($class)")
1169         or diag "uc_cols: @{[ %uc_cols ]}";
1170 }
1171
1172 sub dbconnect {
1173     my ($self, $complain) = @_;
1174
1175     require DBIx::Class::Storage::DBI;
1176     my $storage = DBIx::Class::Storage::DBI->new;
1177
1178     $complain = defined $complain ? $complain : 1;
1179
1180     $storage->connect_info([
1181         @{ $self }{qw/dsn user password/},
1182         {
1183             unsafe => 1,
1184             RaiseError => $complain,
1185             ShowErrorStatement => $complain,
1186             PrintError => 0,
1187             %{ $self->{connect_info_opts} || {} },
1188         },
1189     ]);
1190
1191     my $dbh = $storage->dbh;
1192     die "Failed to connect to database: $@" if !$dbh;
1193
1194     $self->{storage} = $storage; # storage DESTROY disconnects
1195
1196     return $dbh;
1197 }
1198
1199 sub get_oqt_cqt {
1200     my $self = shift;
1201     my %opts = @_;
1202
1203     if ((not $opts{always}) && $self->{preserve_case_mode_is_exclusive}) {
1204         return ('', '');
1205     }
1206
1207     # XXX should get quote_char from the storage of an initialized loader.
1208     my ($oqt, $cqt); # open quote, close quote
1209     if (ref $self->{quote_char}) {
1210         ($oqt, $cqt) = @{ $self->{quote_char} };
1211     }
1212     else {
1213         $oqt = $cqt = $self->{quote_char} || '';
1214     }
1215
1216     return ($oqt, $cqt);
1217 }
1218
1219 sub create {
1220     my $self = shift;
1221
1222     $self->{_created} = 1;
1223
1224     $self->drop_tables;
1225
1226     my $make_auto_inc = $self->{auto_inc_cb} || sub {};
1227     @statements = (
1228         qq{
1229             CREATE TABLE loader_test1s (
1230                 id $self->{auto_inc_pk},
1231                 dat VARCHAR(32) NOT NULL UNIQUE
1232             ) $self->{innodb}
1233         },
1234         $make_auto_inc->(qw/loader_test1s id/),
1235
1236         q{ INSERT INTO loader_test1s (dat) VALUES('foo') },
1237         q{ INSERT INTO loader_test1s (dat) VALUES('bar') }, 
1238         q{ INSERT INTO loader_test1s (dat) VALUES('baz') }, 
1239
1240         # also test method collision
1241         qq{ 
1242             CREATE TABLE loader_test2 (
1243                 id $self->{auto_inc_pk},
1244                 dat VARCHAR(32) NOT NULL,
1245                 dat2 VARCHAR(32) NOT NULL,
1246                 set_primary_key INTEGER $self->{null},
1247                 can INTEGER $self->{null},
1248                 dbix_class_testcomponent INTEGER $self->{null},
1249                 testcomponent_fqn INTEGER $self->{null},
1250                 meta INTEGER $self->{null},
1251                 UNIQUE (dat2, dat)
1252             ) $self->{innodb}
1253         },
1254         $make_auto_inc->(qw/loader_test2 id/),
1255
1256         q{ INSERT INTO loader_test2 (dat, dat2) VALUES('aaa', 'zzz') }, 
1257         q{ INSERT INTO loader_test2 (dat, dat2) VALUES('bbb', 'yyy') }, 
1258         q{ INSERT INTO loader_test2 (dat, dat2) VALUES('ccc', 'xxx') }, 
1259         q{ INSERT INTO loader_test2 (dat, dat2) VALUES('ddd', 'www') }, 
1260
1261         qq{
1262             CREATE TABLE LOADER_test23 (
1263                 ID INTEGER NOT NULL PRIMARY KEY,
1264                 DAT VARCHAR(32) NOT NULL UNIQUE
1265             ) $self->{innodb}
1266         },
1267
1268         qq{
1269             CREATE TABLE LoAdEr_test24 (
1270                 iD INTEGER NOT NULL PRIMARY KEY,
1271                 DaT VARCHAR(32) NOT NULL UNIQUE
1272             ) $self->{innodb}
1273         },
1274
1275 # Access does not support DEFAULT
1276         $self->{vendor} ne 'Access' ? qq{
1277             CREATE TABLE loader_test35 (
1278                 id INTEGER NOT NULL PRIMARY KEY,
1279                 a_varchar VARCHAR(100) DEFAULT 'foo',
1280                 an_int INTEGER DEFAULT 42,
1281                 a_negative_int INTEGER DEFAULT -42,
1282                 a_double DOUBLE PRECISION DEFAULT 10.555,
1283                 a_negative_double DOUBLE PRECISION DEFAULT -10.555,
1284                 a_function $self->{default_function_def}
1285             ) $self->{innodb}
1286         } : qq{
1287             CREATE TABLE loader_test35 (
1288                 id INTEGER NOT NULL PRIMARY KEY,
1289                 a_varchar VARCHAR(100),
1290                 an_int INTEGER,
1291                 a_negative_int INTEGER,
1292                 a_double DOUBLE,
1293                 a_negative_double DOUBLE,
1294                 a_function DATETIME
1295             )
1296         },
1297
1298         qq{
1299             CREATE TABLE loader_test36 (
1300                 id INTEGER NOT NULL PRIMARY KEY,
1301                 a_date $self->{basic_date_datatype},
1302                 b_char_as_data VARCHAR(100),
1303                 c_char_as_data VARCHAR(100)
1304             ) $self->{innodb}
1305         },
1306     );
1307
1308     # some DBs require mixed case identifiers to be quoted
1309     my ($oqt, $cqt) = $self->get_oqt_cqt;
1310
1311     @statements_reltests = (
1312         qq{
1313             CREATE TABLE loader_test3 (
1314                 id INTEGER NOT NULL PRIMARY KEY,
1315                 dat VARCHAR(32)
1316             ) $self->{innodb}
1317         },
1318
1319         q{ INSERT INTO loader_test3 (id,dat) VALUES(1,'aaa') }, 
1320         q{ INSERT INTO loader_test3 (id,dat) VALUES(2,'bbb') }, 
1321         q{ INSERT INTO loader_test3 (id,dat) VALUES(3,'ccc') }, 
1322         q{ INSERT INTO loader_test3 (id,dat) VALUES(4,'ddd') }, 
1323
1324         qq{
1325             CREATE TABLE loader_test4 (
1326                 id INTEGER NOT NULL PRIMARY KEY,
1327                 fkid INTEGER NOT NULL,
1328                 dat VARCHAR(32),
1329                 crumb_crisp_coating VARCHAR(32) $self->{null},
1330                 belongs_to INTEGER $self->{null},
1331                 set_primary_key INTEGER $self->{null},
1332                 FOREIGN KEY( fkid ) REFERENCES loader_test3 (id),
1333                 FOREIGN KEY( belongs_to ) REFERENCES loader_test3 (id),
1334                 FOREIGN KEY( set_primary_key ) REFERENCES loader_test3 (id)
1335             ) $self->{innodb}
1336         },
1337
1338         q{ INSERT INTO loader_test4 (id,fkid,dat,belongs_to,set_primary_key) VALUES(123,1,'aaa',1,1) },
1339         q{ INSERT INTO loader_test4 (id,fkid,dat,belongs_to,set_primary_key) VALUES(124,2,'bbb',2,2) }, 
1340         q{ INSERT INTO loader_test4 (id,fkid,dat,belongs_to,set_primary_key) VALUES(125,3,'ccc',3,3) },
1341         q{ INSERT INTO loader_test4 (id,fkid,dat,belongs_to,set_primary_key) VALUES(126,4,'ddd',4,4) },
1342
1343         qq|
1344             CREATE TABLE loader_test5 (
1345                 id1 INTEGER NOT NULL,
1346                 ${oqt}iD2${cqt} INTEGER NOT NULL,
1347                 dat VARCHAR(8),
1348                 from_id INTEGER $self->{null},
1349                 to_id INTEGER $self->{null},
1350                 PRIMARY KEY (id1,${oqt}iD2${cqt}),
1351                 FOREIGN KEY (from_id) REFERENCES loader_test4 (id),
1352                 FOREIGN KEY (to_id) REFERENCES loader_test4 (id)
1353             ) $self->{innodb}
1354         |,
1355
1356         qq| INSERT INTO loader_test5 (id1,${oqt}iD2${cqt},dat) VALUES (1,1,'aaa') |,
1357
1358         qq|
1359             CREATE TABLE loader_test6 (
1360                 id INTEGER NOT NULL PRIMARY KEY,
1361                 ${oqt}Id2${cqt} INTEGER,
1362                 loader_test2_id INTEGER,
1363                 dat VARCHAR(8),
1364                 FOREIGN KEY (loader_test2_id)  REFERENCES loader_test2 (id),
1365                 FOREIGN KEY(id,${oqt}Id2${cqt}) REFERENCES loader_test5 (id1,${oqt}iD2${cqt})
1366             ) $self->{innodb}
1367         |,
1368
1369         (qq| INSERT INTO loader_test6 (id, ${oqt}Id2${cqt},loader_test2_id,dat) | .
1370          q{ VALUES (1, 1,1,'aaa') }),
1371
1372         qq{
1373             CREATE TABLE loader_test7 (
1374                 id INTEGER NOT NULL PRIMARY KEY,
1375                 id2 VARCHAR(8) NOT NULL UNIQUE,
1376                 dat VARCHAR(8)
1377             ) $self->{innodb}
1378         },
1379
1380         q{ INSERT INTO loader_test7 (id,id2,dat) VALUES (1,'aaa','bbb') },
1381
1382         qq{
1383             CREATE TABLE loader_test8 (
1384                 id INTEGER NOT NULL PRIMARY KEY,
1385                 loader_test7 VARCHAR(8) NOT NULL,
1386                 dat VARCHAR(8),
1387                 FOREIGN KEY (loader_test7) REFERENCES loader_test7 (id2)
1388             ) $self->{innodb}
1389         },
1390
1391         (q{ INSERT INTO loader_test8 (id,loader_test7,dat) } .
1392          q{ VALUES (1,'aaa','bbb') }),
1393
1394         qq{
1395             CREATE TABLE loader_test9 (
1396                 loader_test9 VARCHAR(8) NOT NULL
1397             ) $self->{innodb}
1398         },
1399
1400         qq{
1401             CREATE TABLE loader_test16 (
1402                 id INTEGER NOT NULL PRIMARY KEY,
1403                 dat  VARCHAR(8)
1404             ) $self->{innodb}
1405         },
1406
1407         qq{ INSERT INTO loader_test16 (id,dat) VALUES (2,'x16') },
1408         qq{ INSERT INTO loader_test16 (id,dat) VALUES (4,'y16') },
1409         qq{ INSERT INTO loader_test16 (id,dat) VALUES (6,'z16') },
1410
1411         qq{
1412             CREATE TABLE loader_test17 (
1413                 id INTEGER NOT NULL PRIMARY KEY,
1414                 loader16_one INTEGER,
1415                 loader16_two INTEGER,
1416                 FOREIGN KEY (loader16_one) REFERENCES loader_test16 (id),
1417                 FOREIGN KEY (loader16_two) REFERENCES loader_test16 (id)
1418             ) $self->{innodb}
1419         },
1420
1421         qq{ INSERT INTO loader_test17 (id, loader16_one, loader16_two) VALUES (3, 2, 4) },
1422         qq{ INSERT INTO loader_test17 (id, loader16_one, loader16_two) VALUES (33, 4, 6) },
1423
1424         qq{
1425             CREATE TABLE loader_test18 (
1426                 id INTEGER NOT NULL PRIMARY KEY,
1427                 dat  VARCHAR(8)
1428             ) $self->{innodb}
1429         },
1430
1431         qq{ INSERT INTO loader_test18 (id,dat) VALUES (1,'x18') },
1432         qq{ INSERT INTO loader_test18 (id,dat) VALUES (2,'y18') },
1433         qq{ INSERT INTO loader_test18 (id,dat) VALUES (3,'z18') },
1434
1435         qq{
1436             CREATE TABLE loader_test19 (
1437                 id INTEGER NOT NULL PRIMARY KEY,
1438                 dat  VARCHAR(8)
1439             ) $self->{innodb}
1440         },
1441
1442         qq{ INSERT INTO loader_test19 (id,dat) VALUES (4,'x19') },
1443         qq{ INSERT INTO loader_test19 (id,dat) VALUES (5,'y19') },
1444         qq{ INSERT INTO loader_test19 (id,dat) VALUES (6,'z19') },
1445
1446         qq{
1447             CREATE TABLE loader_test20 (
1448                 parent INTEGER NOT NULL,
1449                 child INTEGER NOT NULL,
1450                 PRIMARY KEY (parent, child),
1451                 FOREIGN KEY (parent) REFERENCES loader_test18 (id),
1452                 FOREIGN KEY (child) REFERENCES loader_test19 (id)
1453             ) $self->{innodb}
1454         },
1455
1456         q{ INSERT INTO loader_test20 (parent, child) VALUES (1,4) },
1457         q{ INSERT INTO loader_test20 (parent, child) VALUES (2,5) },
1458         q{ INSERT INTO loader_test20 (parent, child) VALUES (3,6) },
1459
1460         qq{
1461             CREATE TABLE loader_test21 (
1462                 id INTEGER NOT NULL PRIMARY KEY,
1463                 dat  VARCHAR(8)
1464             ) $self->{innodb}
1465         },
1466
1467         q{ INSERT INTO loader_test21 (id,dat) VALUES (7,'a21')},
1468         q{ INSERT INTO loader_test21 (id,dat) VALUES (11,'b21')},
1469         q{ INSERT INTO loader_test21 (id,dat) VALUES (13,'c21')},
1470         q{ INSERT INTO loader_test21 (id,dat) VALUES (17,'d21')},
1471
1472         qq{
1473             CREATE TABLE loader_test22 (
1474                 parent INTEGER NOT NULL,
1475                 child INTEGER NOT NULL,
1476                 PRIMARY KEY (parent, child),
1477                 FOREIGN KEY (parent) REFERENCES loader_test21 (id),
1478                 FOREIGN KEY (child) REFERENCES loader_test21 (id)
1479             ) $self->{innodb}
1480         },
1481
1482         q{ INSERT INTO loader_test22 (parent, child) VALUES (7,11)},
1483         q{ INSERT INTO loader_test22 (parent, child) VALUES (11,13)},
1484         q{ INSERT INTO loader_test22 (parent, child) VALUES (13,17)},
1485
1486         qq{
1487             CREATE TABLE loader_test25 (
1488                 id1 INTEGER NOT NULL,
1489                 id2 INTEGER NOT NULL,
1490                 dat VARCHAR(8),
1491                 PRIMARY KEY (id1,id2)
1492             ) $self->{innodb}
1493         },
1494
1495         q{ INSERT INTO loader_test25 (id1,id2,dat) VALUES (33,5,'x25') },
1496         q{ INSERT INTO loader_test25 (id1,id2,dat) VALUES (33,7,'y25') },
1497         q{ INSERT INTO loader_test25 (id1,id2,dat) VALUES (3,42,'z25') },
1498
1499         qq{
1500             CREATE TABLE loader_test26 (
1501                id INTEGER NOT NULL PRIMARY KEY,
1502                rel1 INTEGER NOT NULL,
1503                rel2 INTEGER NOT NULL,
1504                FOREIGN KEY (id, rel1) REFERENCES loader_test25 (id1, id2),
1505                FOREIGN KEY (id, rel2) REFERENCES loader_test25 (id1, id2)
1506             ) $self->{innodb}
1507         },
1508
1509         q{ INSERT INTO loader_test26 (id,rel1,rel2) VALUES (33,5,7) },
1510         q{ INSERT INTO loader_test26 (id,rel1,rel2) VALUES (3,42,42) },
1511
1512         qq{
1513             CREATE TABLE loader_test27 (
1514                 id INTEGER NOT NULL PRIMARY KEY
1515             ) $self->{innodb}
1516         },
1517
1518         q{ INSERT INTO loader_test27 (id) VALUES (1) },
1519         q{ INSERT INTO loader_test27 (id) VALUES (2) },
1520
1521         qq{
1522             CREATE TABLE loader_test28 (
1523                 id INTEGER NOT NULL PRIMARY KEY,
1524                 FOREIGN KEY (id) REFERENCES loader_test27 (id)
1525             ) $self->{innodb}
1526         },
1527
1528         q{ INSERT INTO loader_test28 (id) VALUES (1) },
1529
1530         qq{
1531             CREATE TABLE loader_test29 (
1532                 id INTEGER NOT NULL PRIMARY KEY,
1533                 fk INTEGER NOT NULL UNIQUE,
1534                 FOREIGN KEY (fk) REFERENCES loader_test27 (id)
1535             ) $self->{innodb}
1536         },
1537
1538         q{ INSERT INTO loader_test29 (id,fk) VALUES (1,1) },
1539
1540         qq{
1541           CREATE TABLE loader_test31 (
1542             id INTEGER NOT NULL PRIMARY KEY
1543           ) $self->{innodb}
1544         },
1545         q{ INSERT INTO loader_test31 (id) VALUES (1) },
1546
1547         qq{
1548           CREATE TABLE loader_test32 (
1549             id INTEGER NOT NULL PRIMARY KEY,
1550             rel1 INTEGER NOT NULL,
1551             rel2 INTEGER $self->{null},
1552             FOREIGN KEY (rel1) REFERENCES loader_test31(id),
1553             FOREIGN KEY (rel2) REFERENCES loader_test31(id)
1554           ) $self->{innodb}
1555         },
1556         q{ INSERT INTO loader_test32 (id,rel1) VALUES (1,1) },
1557
1558         qq{
1559           CREATE TABLE loader_test33 (
1560             id1 INTEGER NOT NULL,
1561             id2 INTEGER NOT NULL,
1562             PRIMARY KEY (id1,id2)
1563           ) $self->{innodb}
1564         },
1565         q{ INSERT INTO loader_test33 (id1,id2) VALUES (1,2) },
1566
1567         qq{
1568           CREATE TABLE loader_test34 (
1569             id INTEGER NOT NULL PRIMARY KEY,
1570             rel1 INTEGER NOT NULL,
1571             rel2 INTEGER $self->{null},
1572             FOREIGN KEY (id,rel1) REFERENCES loader_test33(id1,id2),
1573             FOREIGN KEY (id,rel2) REFERENCES loader_test33(id1,id2)
1574           ) $self->{innodb}
1575         },
1576         q{ INSERT INTO loader_test34 (id,rel1,rel2) VALUES (1,2,2) },
1577     );
1578
1579     @statements_advanced = (
1580         qq{
1581             CREATE TABLE loader_test10 (
1582                 id10 $self->{auto_inc_pk},
1583                 subject VARCHAR(8),
1584                 loader_test11 INTEGER $self->{null}
1585             ) $self->{innodb}
1586         },
1587         $make_auto_inc->(qw/loader_test10 id10/),
1588
1589 # Access does not support DEFAULT.
1590         qq{
1591             CREATE TABLE loader_test11 (
1592                 id11 $self->{auto_inc_pk},
1593                 a_message VARCHAR(8) @{[ $self->{vendor} ne 'Access' ? "DEFAULT 'foo'" : '' ]},
1594                 loader_test10 INTEGER $self->{null},
1595                 FOREIGN KEY (loader_test10) REFERENCES loader_test10 (id10)
1596             ) $self->{innodb}
1597         },
1598         $make_auto_inc->(qw/loader_test11 id11/),
1599
1600         (lc($self->{vendor}) ne 'informix' ?
1601             (q{ ALTER TABLE loader_test10 ADD CONSTRAINT loader_test11_fk } .
1602              q{ FOREIGN KEY (loader_test11) } .
1603              q{ REFERENCES loader_test11 (id11) })
1604         :
1605             (q{ ALTER TABLE loader_test10 ADD CONSTRAINT } .
1606              q{ FOREIGN KEY (loader_test11) } .
1607              q{ REFERENCES loader_test11 (id11) } .
1608              q{ CONSTRAINT loader_test11_fk })
1609         ),
1610     );
1611
1612     @statements_advanced_sqlite = (
1613         qq{
1614             CREATE TABLE loader_test10 (
1615                 id10 $self->{auto_inc_pk},
1616                 subject VARCHAR(8)
1617             ) $self->{innodb}
1618         },
1619         $make_auto_inc->(qw/loader_test10 id10/),
1620
1621         qq{
1622             CREATE TABLE loader_test11 (
1623                 id11 $self->{auto_inc_pk},
1624                 a_message VARCHAR(8) DEFAULT 'foo',
1625                 loader_test10 INTEGER $self->{null},
1626                 FOREIGN KEY (loader_test10) REFERENCES loader_test10 (id10)
1627             ) $self->{innodb}
1628         },
1629         $make_auto_inc->(qw/loader_test11 id11/),
1630
1631         (q{ ALTER TABLE loader_test10 ADD COLUMN } .
1632          q{ loader_test11 INTEGER REFERENCES loader_test11 (id11) }),
1633     );
1634
1635     @statements_inline_rels = (
1636         qq{
1637             CREATE TABLE loader_test12 (
1638                 id INTEGER NOT NULL PRIMARY KEY,
1639                 id2 VARCHAR(8) NOT NULL UNIQUE,
1640                 dat VARCHAR(8) NOT NULL UNIQUE
1641             ) $self->{innodb}
1642         },
1643
1644         q{ INSERT INTO loader_test12 (id,id2,dat) VALUES (1,'aaa','bbb') },
1645
1646         qq{
1647             CREATE TABLE loader_test13 (
1648                 id INTEGER NOT NULL PRIMARY KEY REFERENCES loader_test12,
1649                 loader_test12 VARCHAR(8) NOT NULL REFERENCES loader_test12 (id2),
1650                 dat VARCHAR(8) REFERENCES loader_test12 (dat)
1651             ) $self->{innodb}
1652         },
1653
1654         (q{ INSERT INTO loader_test13 (id,loader_test12,dat) } .
1655          q{ VALUES (1,'aaa','bbb') }),
1656     );
1657
1658
1659     @statements_implicit_rels = (
1660         qq{
1661             CREATE TABLE loader_test14 (
1662                 id INTEGER NOT NULL PRIMARY KEY,
1663                 dat VARCHAR(8)
1664             ) $self->{innodb}
1665         },
1666  
1667         q{ INSERT INTO loader_test14 (id,dat) VALUES (123,'aaa') },
1668
1669         qq{
1670             CREATE TABLE loader_test15 (
1671                 id INTEGER NOT NULL PRIMARY KEY,
1672                 loader_test14 INTEGER NOT NULL,
1673                 FOREIGN KEY (loader_test14) REFERENCES loader_test14
1674             ) $self->{innodb}
1675         },
1676
1677         q{ INSERT INTO loader_test15 (id,loader_test14) VALUES (1,123) },
1678     );
1679
1680     $self->drop_tables;
1681
1682     my $dbh = $self->dbconnect(1);
1683
1684     $dbh->do($_) for @{ $self->{pre_create} || [] };
1685
1686     $dbh->do($_) foreach (@statements);
1687
1688     if (not ($self->{vendor} eq 'mssql' && $dbh->{Driver}{Name} eq 'Sybase')) {
1689         $dbh->do($_) foreach (@{ $self->{data_type_tests}{ddl} || [] });
1690     }
1691
1692     unless($self->{skip_rels}) {
1693         # hack for now, since DB2 doesn't like inline comments, and we need
1694         # to test one for mysql, which works on everyone else...
1695         # this all needs to be refactored anyways.
1696
1697         for my $stmt (@statements_reltests) {
1698             try {
1699                 $dbh->do($stmt);
1700             }
1701             catch {
1702                 die "Error executing '$stmt': $_\n";
1703             };
1704         }
1705         if($self->{vendor} =~ /sqlite/i) {
1706             $dbh->do($_) for (@statements_advanced_sqlite);
1707         }
1708         else {
1709             $dbh->do($_) for (@statements_advanced);
1710         }
1711         unless($self->{no_inline_rels}) {
1712             $dbh->do($_) for (@statements_inline_rels);
1713         }
1714         unless($self->{no_implicit_rels}) {
1715             $dbh->do($_) for (@statements_implicit_rels);
1716         }
1717     }
1718
1719     $dbh->do($_) for @{ $self->{extra}->{create} || [] };
1720     $dbh->disconnect();
1721 }
1722
1723 sub drop_tables {
1724     my $self = shift;
1725
1726     my @tables = qw/
1727         loader_test1
1728         loader_test1s
1729         loader_test2
1730         LOADER_test23
1731         LoAdEr_test24
1732         loader_test35
1733         loader_test36
1734     /;
1735     
1736     my @tables_auto_inc = (
1737         [ qw/loader_test1s id/ ],
1738         [ qw/loader_test2 id/ ],
1739     );
1740
1741     my @tables_reltests = qw/
1742         loader_test4
1743         loader_test3
1744         loader_test6
1745         loader_test5
1746         loader_test8
1747         loader_test7
1748         loader_test9
1749         loader_test17
1750         loader_test16
1751         loader_test20
1752         loader_test19
1753         loader_test18
1754         loader_test22
1755         loader_test21
1756         loader_test26
1757         loader_test25
1758         loader_test28
1759         loader_test29
1760         loader_test27
1761         loader_test32
1762         loader_test31
1763         loader_test34
1764         loader_test33
1765     /;
1766
1767     my @tables_advanced = qw/
1768         loader_test11
1769         loader_test10
1770     /;
1771     
1772     my @tables_advanced_auto_inc = (
1773         [ qw/loader_test10 id10/ ],
1774         [ qw/loader_test11 id11/ ],
1775     );
1776
1777     my @tables_inline_rels = qw/
1778         loader_test13
1779         loader_test12
1780     /;
1781
1782     my @tables_implicit_rels = qw/
1783         loader_test15
1784         loader_test14
1785     /;
1786
1787     my @tables_rescan = qw/ loader_test30 /;
1788
1789     my @tables_preserve_case_tests = qw/ LoaderTest41 LoaderTest40 /;
1790
1791     my $drop_fk_mysql =
1792         q{ALTER TABLE loader_test10 DROP FOREIGN KEY loader_test11_fk};
1793
1794     my $drop_fk =
1795         q{ALTER TABLE loader_test10 DROP CONSTRAINT loader_test11_fk};
1796
1797     # For some reason some tests do this twice (I guess dependency issues?)
1798     # do it twice for all drops
1799     for (1,2) {
1800         local $^W = 0; # for ADO
1801
1802         my $dbh = $self->dbconnect(0);
1803
1804         $dbh->do($_) for @{ $self->{extra}{pre_drop_ddl} || [] };
1805
1806         $dbh->do("DROP TABLE $_") for @{ $self->{extra}{drop} || [] };
1807
1808         my $drop_auto_inc = $self->{auto_inc_drop_cb} || sub {};
1809
1810         unless($self->{skip_rels}) {
1811             $dbh->do("DROP TABLE $_") for (@tables_reltests);
1812             $dbh->do("DROP TABLE $_") for (@tables_reltests);
1813             if($self->{vendor} =~ /mysql/i) {
1814                 $dbh->do($drop_fk_mysql);
1815             }
1816             else {
1817                 $dbh->do($drop_fk);
1818             }
1819             $dbh->do($_) for map { $drop_auto_inc->(@$_) } @tables_advanced_auto_inc;
1820             $dbh->do("DROP TABLE $_") for (@tables_advanced);
1821
1822             unless($self->{no_inline_rels}) {
1823                 $dbh->do("DROP TABLE $_") for (@tables_inline_rels);
1824             }
1825             unless($self->{no_implicit_rels}) {
1826                 $dbh->do("DROP TABLE $_") for (@tables_implicit_rels);
1827             }
1828         }
1829         $dbh->do($_) for map { $drop_auto_inc->(@$_) } @tables_auto_inc;
1830         $dbh->do("DROP TABLE $_") for (@tables, @tables_rescan);
1831
1832         if (not ($self->{vendor} eq 'mssql' && $dbh->{Driver}{Name} eq 'Sybase')) {
1833             foreach my $data_type_table (@{ $self->{data_type_tests}{table_names} || [] }) {
1834                 $dbh->do("DROP TABLE $data_type_table");
1835             }
1836         }
1837
1838         my ($oqt, $cqt) = $self->get_oqt_cqt(always => 1);
1839
1840         $dbh->do("DROP TABLE ${oqt}${_}${cqt}") for @tables_preserve_case_tests;
1841
1842         $dbh->disconnect;
1843     }
1844 }
1845
1846 sub _custom_column_info {
1847     my ( $table_name, $column_name, $column_info ) = @_;
1848
1849     $table_name = lc ( $table_name );
1850     $column_name = lc ( $column_name );
1851
1852     if ( $table_name eq 'loader_test35' 
1853         and $column_name eq 'an_int' 
1854     ){
1855         return { is_numeric => 1 }
1856     }
1857     # Set inflate_datetime or  inflate_date to check 
1858     #   datetime_timezone and datetime_locale
1859     if ( $table_name eq 'loader_test36' ){
1860         return { inflate_datetime => 1 } if 
1861             ( $column_name eq 'b_char_as_data' );
1862         return { inflate_date => 1 } if 
1863             ( $column_name eq 'c_char_as_data' );
1864     }
1865
1866     return;
1867 }
1868
1869 my %DATA_TYPE_MULTI_TABLE_OVERRIDES = (
1870     oracle => qr/\blong\b/i,
1871     mssql  => qr/\b(?:timestamp|rowversion)\b/i,
1872     informix => qr/\b(?:bigserial|serial8)\b/i,
1873 );
1874
1875 sub setup_data_type_tests {
1876     my $self = shift;
1877
1878     return unless my $types = $self->{data_types};
1879
1880     my $tests = $self->{data_type_tests} = {};
1881
1882     # split types into tables based on overrides
1883     my (@types, @split_off_types, @first_table_types);
1884     {
1885         my $split_off_re = $DATA_TYPE_MULTI_TABLE_OVERRIDES{lc($self->{vendor})} || qr/(?!)/;
1886
1887         @types = keys %$types;
1888         @split_off_types   = grep  /$split_off_re/, @types;
1889         @first_table_types = grep !/$split_off_re/, @types;
1890     }
1891
1892     @types = +{ map +($_, $types->{$_}), @first_table_types },
1893         map +{ $_, $types->{$_} }, @split_off_types;
1894
1895     my $test_count = 0;
1896     my $table_num  = 10000;
1897
1898     foreach my $types (@types) {
1899         my $table_name    = "loader_test$table_num";
1900         push @{ $tests->{table_names} }, $table_name;
1901
1902         my $table_moniker = "LoaderTest$table_num";
1903         push @{ $tests->{table_monikers} }, $table_moniker;
1904
1905         $table_num++;
1906
1907         my $cols = $tests->{columns}{$table_moniker} = {};
1908
1909         my $ddl = "CREATE TABLE $table_name (\n    id INTEGER NOT NULL PRIMARY KEY,\n";
1910
1911         my %seen_col_names;
1912
1913         while (my ($col_def, $expected_info) = each %$types) {
1914             (my $type_alias = $col_def) =~ s/\( ([^)]+) \)//xg;
1915
1916             my $size = $1;
1917             $size = '' unless defined $size;
1918             $size =~ s/\s+//g;
1919             my @size = split /,/, $size;
1920
1921             # some DBs don't like very long column names
1922             if ($self->{vendor} =~ /^(?:firebird|sqlanywhere|oracle|db2)\z/i) {
1923                 my ($col_def, $default) = $type_alias =~ /^(.*)(default.*)?\z/i;
1924
1925                 $type_alias = substr $col_def, 0, 15;
1926
1927                 $type_alias .= '_with_dflt' if $default;
1928             }
1929
1930             $type_alias =~ s/\s/_/g;
1931             $type_alias =~ s/\W//g;
1932
1933             my $col_name = 'col_' . $type_alias;
1934             
1935             if (@size) {
1936                 my $size_name = join '_', apply { s/\W//g } @size;
1937
1938                 $col_name .= "_sz_$size_name";
1939             }
1940
1941             # XXX would be better to check _loader->preserve_case
1942             $col_name = lc $col_name;
1943
1944             $col_name .= '_' . $seen_col_names{$col_name} if $seen_col_names{$col_name}++;
1945
1946             $ddl .= "    $col_name $col_def,\n";
1947
1948             $cols->{$col_name} = $expected_info;
1949
1950             $test_count++;
1951         }
1952
1953         $ddl =~ s/,\n\z/\n)/;
1954
1955         push @{ $tests->{ddl} }, $ddl;
1956     }
1957
1958     $tests->{test_count} = $test_count;
1959
1960     return $test_count;
1961 }
1962
1963 sub rescan_without_warnings {
1964     my ($self, $conn) = @_;
1965
1966     local $SIG{__WARN__} = sub { warn @_ unless $_[0] =~ RESCAN_WARNINGS };
1967     return $conn->rescan;
1968 }
1969
1970 sub test_col_accessor_map {
1971     my ( $column_name, $default_name, $context ) = @_;
1972     if( lc($column_name) eq 'crumb_crisp_coating' ) {
1973
1974         is( $default_name, 'crumb_crisp_coating', 'col_accessor_map was passed the default name' );
1975         ok( $context->{$_}, "col_accessor_map func was passed the $_" )
1976             for qw( table_name table_class table_moniker schema_class );
1977
1978         return 'trivet';
1979     } else {
1980         return $default_name;
1981     }
1982 }
1983
1984 sub DESTROY {
1985     my $self = shift;
1986     unless ($ENV{SCHEMA_LOADER_TESTS_NOCLEANUP}) {
1987       $self->drop_tables if $self->{_created};
1988       rmtree $DUMP_DIR
1989     }
1990 }
1991
1992 1;
1993 # vim:et sts=4 sw=4 tw=0: