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