t/lib/dbixcsl_common_tests.pm

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