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