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