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