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