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