Move the Subclassing POD under Examples
[dbsrgits/DBIx-Class.git] / t / 72pg.t
CommitLineData
70350518 1use strict;
4617b792 2use warnings;
70350518 3
4use Test::More;
9ba627e3 5use Test::Exception;
bbdda281 6use Sub::Name;
199fbc45 7use DBIx::Class::Optional::Dependencies ();
70350518 8use lib qw(t/lib);
9use DBICTest;
10
199fbc45 11plan skip_all => 'Test needs ' . DBIx::Class::Optional::Dependencies->req_missing_for ('test_rdbms_pg')
12 unless DBIx::Class::Optional::Dependencies->req_ok_for ('test_rdbms_pg');
89add887 13
0567538f 14my ($dsn, $user, $pass) = @ENV{map { "DBICTEST_PG_${_}" } qw/DSN USER PASS/};
15
67b79ae1 16plan skip_all => <<'EOM' unless $dsn && $user;
17Set $ENV{DBICTEST_PG_DSN}, _USER and _PASS to run this test
bab40dee 18( NOTE: This test drops and creates tables called 'artist', 'cd',
19'timestamp_primary_key_test', 'track', 'casecheck', 'array_test' and
20'sequence_test' as well as following sequences: 'pkid1_seq', 'pkid2_seq' and
67b79ae1 21'nonpkid_seq'. as well as following schemas: 'dbic_t_schema',
bab40dee 22'dbic_t_schema_2', 'dbic_t_schema_3', 'dbic_t_schema_4', and 'dbic_t_schema_5')
609c5f1b 23EOM
0567538f 24
52c53388 25### load any test classes that are defined further down in the file via BEGIN blocks
efd38994 26
70a201a5 27our @test_classes; #< array that will be pushed into by test classes defined in this file
28DBICTest::Schema->load_classes( map {s/.+:://;$_} @test_classes ) if @test_classes;
efd38994 29
52c53388 30### pre-connect tests (keep each test separate as to make sure rebless() runs)
bab40dee 31 {
32 my $s = DBICTest::Schema->connect($dsn, $user, $pass);
eabde904 33
bab40dee 34 ok (!$s->storage->_dbh, 'definitely not connected');
eabde904 35
bab40dee 36 # Check that datetime_parser returns correctly before we explicitly connect.
37 SKIP: {
68de9438 38 skip (
39 "Pg parser detection test needs " . DBIx::Class::Optional::Dependencies->req_missing_for ('test_dt_pg'),
40 2
41 ) unless DBIx::Class::Optional::Dependencies->req_ok_for ('test_dt_pg');
114780ee 42
bab40dee 43 my $store = ref $s->storage;
44 is($store, 'DBIx::Class::Storage::DBI', 'Started with generic storage');
114780ee 45
bab40dee 46 my $parser = $s->storage->datetime_parser;
47 is( $parser, 'DateTime::Format::Pg', 'datetime_parser is as expected');
48 }
114780ee 49
bab40dee 50 ok (!$s->storage->_dbh, 'still not connected');
51 }
bbdda281 52
bab40dee 53 {
54 my $s = DBICTest::Schema->connect($dsn, $user, $pass);
55 # make sure sqlt_type overrides work (::Storage::DBI::Pg does this)
56 ok (!$s->storage->_dbh, 'definitely not connected');
57 is ($s->storage->sqlt_type, 'PostgreSQL', 'sqlt_type correct pre-connection');
58 ok (!$s->storage->_dbh, 'still not connected');
59 }
0567538f 60
97277745 61# test LIMIT support
62{
63 my $schema = DBICTest::Schema->connect($dsn, $user, $pass);
64 drop_test_schema($schema);
65 create_test_schema($schema);
66 for (1..6) {
67 $schema->resultset('Artist')->create({ name => 'Artist ' . $_ });
68 }
69 my $it = $schema->resultset('Artist')->search( {},
70 { rows => 3,
71 offset => 2,
72 order_by => 'artistid' }
73 );
74 is( $it->count, 3, "LIMIT count ok" ); # ask for 3 rows out of 6 artists
75 is( $it->next->name, "Artist 3", "iterator->next ok" );
76 $it->next;
77 $it->next;
78 $it->next;
79 is( $it->next, undef, "next past end of resultset ok" );
e5372da4 80
81 # Limit with select-lock
82 lives_ok {
83 $schema->txn_do (sub {
84 isa_ok (
85 $schema->resultset('Artist')->find({artistid => 1}, {for => 'update', rows => 1}),
86 'DBICTest::Schema::Artist',
87 );
88 });
89 } 'Limited FOR UPDATE select works';
97277745 90}
91
bbdda281 92# check if we indeed do support stuff
93my $test_server_supports_insert_returning = do {
af1f4f84 94
95 my $si = DBICTest::Schema->connect($dsn, $user, $pass)->storage->_server_info;
96 die "Unparseable Pg server version: $si->{dbms_version}\n"
97 unless $si->{normalized_dbms_version};
98
99 $si->{normalized_dbms_version} < 8.002 ? 0 : 1;
bbdda281 100};
101is (
102 DBICTest::Schema->connect($dsn, $user, $pass)->storage->_use_insert_returning,
103 $test_server_supports_insert_returning,
104 'insert returning capability guessed correctly'
105);
106
107my $schema;
108for my $use_insert_returning ($test_server_supports_insert_returning
109 ? (0,1)
110 : (0)
111) {
112
6892eb09 113 no warnings qw/once redefine/;
114 my $old_connection = DBICTest::Schema->can('connection');
bbdda281 115 local *DBICTest::Schema::connection = subname 'DBICTest::Schema::connection' => sub {
6892eb09 116 my $s = shift->$old_connection(@_);
bbdda281 117 $s->storage->_use_insert_returning ($use_insert_returning);
118 $s;
119 };
120
121### test capability override
122 {
123 my $s = DBICTest::Schema->connect($dsn, $user, $pass);
124
125 ok (!$s->storage->_dbh, 'definitely not connected');
126
127 ok (
128 ! ($s->storage->_use_insert_returning xor $use_insert_returning),
129 'insert returning capability set correctly',
130 );
131 ok (!$s->storage->_dbh, 'still not connected (capability override works)');
132 }
133
70a201a5 134### connect, create postgres-specific test schema
c3af542a 135
bab40dee 136 $schema = DBICTest::Schema->connect($dsn, $user, $pass);
1f5aae08 137 $schema->storage->ensure_connected;
138
bab40dee 139 drop_test_schema($schema);
140 create_test_schema($schema);
609c5f1b 141
70a201a5 142### begin main tests
4617b792 143
372da819 144# run a BIG bunch of tests for last-insert-id / Auto-PK / sequence
145# discovery
bab40dee 146 run_apk_tests($schema); #< older set of auto-pk tests
147 run_extended_apk_tests($schema); #< new extended set of auto-pk tests
70a201a5 148
bab40dee 149### type_info tests
52c53388 150
bab40dee 151 my $test_type_info = {
152 'artistid' => {
153 'data_type' => 'integer',
154 'is_nullable' => 0,
155 'size' => 4,
156 },
157 'name' => {
158 'data_type' => 'character varying',
159 'is_nullable' => 1,
160 'size' => 100,
161 'default_value' => undef,
162 },
163 'rank' => {
164 'data_type' => 'integer',
165 'is_nullable' => 0,
166 'size' => 4,
167 'default_value' => 13,
52c53388 168
bab40dee 169 },
170 'charfield' => {
171 'data_type' => 'character',
172 'is_nullable' => 1,
173 'size' => 10,
174 'default_value' => undef,
175 },
176 'arrayfield' => {
177 'data_type' => 'integer[]',
178 'is_nullable' => 1,
179 'size' => undef,
180 'default_value' => undef,
181 },
182 };
52c53388 183
bab40dee 184 my $type_info = $schema->storage->columns_info_for('dbic_t_schema.artist');
185 my $artistid_defval = delete $type_info->{artistid}->{default_value};
186 like($artistid_defval,
187 qr/^nextval\('([^\.]*\.){0,1}artist_artistid_seq'::(?:text|regclass)\)/,
188 'columns_info_for - sequence matches Pg get_autoinc_seq expectations');
189 is_deeply($type_info, $test_type_info,
190 'columns_info_for - column data types');
52c53388 191
70a201a5 192
a953d8d9 193
a953d8d9 194
bab40dee 195####### Array tests
70a201a5 196
bab40dee 197 BEGIN {
198 package DBICTest::Schema::ArrayTest;
199 push @main::test_classes, __PACKAGE__;
70a201a5 200
bab40dee 201 use strict;
202 use warnings;
41519379 203 use base 'DBICTest::BaseResult';
70a201a5 204
bab40dee 205 __PACKAGE__->table('dbic_t_schema.array_test');
206 __PACKAGE__->add_columns(qw/id arrayfield/);
207 __PACKAGE__->column_info_from_storage(1);
208 __PACKAGE__->set_primary_key('id');
70a201a5 209
bab40dee 210 }
211 SKIP: {
0647e0cc 212 skip "Need DBD::Pg 2.9.2 or newer for array tests", 4 if $DBD::Pg::VERSION < 2.009002;
70a201a5 213
41519379 214 my $arr_rs = $schema->resultset('ArrayTest');
215
bab40dee 216 lives_ok {
41519379 217 $arr_rs->create({
bab40dee 218 arrayfield => [1, 2],
219 });
220 } 'inserting arrayref as pg array data';
70a201a5 221
bab40dee 222 lives_ok {
41519379 223 $arr_rs->update({
bab40dee 224 arrayfield => [3, 4],
225 });
226 } 'updating arrayref as pg array data';
ac45262b 227
41519379 228 $arr_rs->create({
0647e0cc 229 arrayfield => [5, 6],
9ba627e3 230 });
c224117b 231
f6faeab8 232 lives_ok {
233 $schema->populate('ArrayTest', [
234 [ qw/arrayfield/ ],
235 [ [0,0] ],
236 ]);
237 } 'inserting arrayref using void ctx populate';
238
41519379 239 # Search using arrays
240 lives_ok {
241 is_deeply (
242 $arr_rs->search({ arrayfield => { -value => [3,4] } })->first->arrayfield,
243 [3,4],
244 'Array value matches'
245 );
246 } 'searching by arrayref';
a780a0ff 247
bab40dee 248 lives_ok {
41519379 249 is_deeply (
250 $arr_rs->search({ arrayfield => { '=' => { -value => [3,4] }} })->first->arrayfield,
49eeb48d 251 [3,4],
41519379 252 'Array value matches explicit equal'
253 );
254 } 'searching by arrayref (explicit equal sign)';
a780a0ff 255
41519379 256 lives_ok {
257 is_deeply (
258 $arr_rs->search({ arrayfield => { '>' => { -value => [3,1] }} })->first->arrayfield,
259 [3,4],
260 'Array value matches greater than'
261 );
262 } 'searching by arrayref (greater than)';
263
264 lives_ok {
265 is (
266 $arr_rs->search({ arrayfield => { '>' => { -value => [3,7] }} })->count,
267 1,
268 'Greater than search found [5,6]',
269 );
270 } 'searching by arrayref (greater than)';
271
272 # Find using arrays
273 lives_ok {
274 is_deeply (
275 $arr_rs->find({ arrayfield => { -value => [3,4] } })->arrayfield,
276 [3,4],
277 'Array value matches implicit equal'
278 );
279 } 'find by arrayref';
280
281 lives_ok {
282 is_deeply (
283 $arr_rs->find({ arrayfield => { '=' => { -value => [3,4] }} })->arrayfield,
284 [3,4],
285 'Array value matches explicit equal'
286 );
287 } 'find by arrayref (equal)';
288
289 # test inferred condition for creation
4ca1fd6f 290 for my $cond (
41519379 291 { -value => [3,4] },
292 \[ '= ?' => [arrayfield => [3, 4]] ],
293 ) {
294 local $TODO = 'No introspection of complex conditions :(';
295 my $arr_rs_cond = $arr_rs->search({ arrayfield => $cond });
296
297 my $row = $arr_rs_cond->create({});
a780a0ff 298 is_deeply ($row->arrayfield, [3,4], 'Array value taken from $rs condition');
299 $row->discard_changes;
300 is_deeply ($row->arrayfield, [3,4], 'Array value made it to storage');
301 }
bab40dee 302 }
9ba627e3 303
70a201a5 304########## Case check
305
bab40dee 306 BEGIN {
307 package DBICTest::Schema::Casecheck;
308 push @main::test_classes, __PACKAGE__;
70a201a5 309
bab40dee 310 use strict;
311 use warnings;
312 use base 'DBIx::Class::Core';
70a201a5 313
bab40dee 314 __PACKAGE__->table('dbic_t_schema.casecheck');
315 __PACKAGE__->add_columns(qw/id name NAME uc_name/);
316 __PACKAGE__->column_info_from_storage(1);
317 __PACKAGE__->set_primary_key('id');
318 }
70a201a5 319
bab40dee 320 my $name_info = $schema->source('Casecheck')->column_info( 'name' );
321 is( $name_info->{size}, 1, "Case sensitive matching info for 'name'" );
ae515736 322
bab40dee 323 my $NAME_info = $schema->source('Casecheck')->column_info( 'NAME' );
324 is( $NAME_info->{size}, 2, "Case sensitive matching info for 'NAME'" );
ae515736 325
bab40dee 326 my $uc_name_info = $schema->source('Casecheck')->column_info( 'uc_name' );
327 is( $uc_name_info->{size}, 3, "Case insensitive matching info for 'uc_name'" );
ae515736 328
70a201a5 329
ecccae8a 330## Test ResultSet->update
331my $artist = $schema->resultset('Artist')->first;
332my $cds = $artist->cds_unordered->search({
333 year => { '!=' => 2010 }
334}, { prefetch => 'liner_notes' });
67b79ae1 335lives_ok { $cds->update({ year => '2010' }) } 'Update on prefetched rs';
70a201a5 336
337## Test SELECT ... FOR UPDATE
338
bab40dee 339 SKIP: {
67b79ae1 340 if(eval { require Sys::SigAction }) {
bab40dee 341 Sys::SigAction->import( 'set_sig_handler' );
342 }
343 else {
344 skip "Sys::SigAction is not available", 6;
345 }
346
347 my ($timed_out, $artist2);
348
349 for my $t (
350 {
351 # Make sure that an error was raised, and that the update failed
352 update_lock => 1,
353 test_sub => sub {
354 ok($timed_out, "update from second schema times out");
355 ok($artist2->is_column_changed('name'), "'name' column is still dirty from second schema");
356 },
52c53388 357 },
bab40dee 358 {
359 # Make sure that an error was NOT raised, and that the update succeeded
360 update_lock => 0,
361 test_sub => sub {
362 ok(! $timed_out, "update from second schema DOES NOT timeout");
363 ok(! $artist2->is_column_changed('name'), "'name' column is NOT dirty from second schema");
364 },
52c53388 365 },
bab40dee 366 ) {
367 # create a new schema
368 my $schema2 = DBICTest::Schema->connect($dsn, $user, $pass);
369 $schema2->source("Artist")->name("dbic_t_schema.artist");
370
371 $schema->txn_do( sub {
d2408067 372 my $rs = $schema->resultset('Artist')->search(
bab40dee 373 {
374 artistid => 1
375 },
376 $t->{update_lock} ? { for => 'update' } : {}
d2408067 377 );
378 ok ($rs->count, 'Count works');
379
380 my $artist = $rs->next;
bab40dee 381 is($artist->artistid, 1, "select returns artistid = 1");
382
383 $timed_out = 0;
384 eval {
385 my $h = set_sig_handler( 'ALRM', sub { die "DBICTestTimeout" } );
386 alarm(2);
387 $artist2 = $schema2->resultset('Artist')->find(1);
388 $artist2->name('fooey');
389 $artist2->update;
390 alarm(0);
391 };
392 $timed_out = $@ =~ /DBICTestTimeout/;
393 });
394
395 $t->{test_sub}->();
396 }
397 }
95ba7ee4 398
70a201a5 399
372da819 400######## other older Auto-pk tests
70a201a5 401
bab40dee 402 $schema->source("SequenceTest")->name("dbic_t_schema.sequence_test");
403 for (1..5) {
404 my $st = $schema->resultset('SequenceTest')->create({ name => 'foo' });
405 is($st->pkid1, $_, "Auto-PK for sequence without default: First primary key");
406 is($st->pkid2, $_ + 9, "Auto-PK for sequence without default: Second primary key");
407 is($st->nonpkid, $_ + 19, "Auto-PK for sequence without default: Non-primary key");
408 }
409 my $st = $schema->resultset('SequenceTest')->create({ name => 'foo', pkid1 => 55 });
410 is($st->pkid1, 55, "Auto-PK for sequence without default: First primary key set manually");
4d4dc518 411
412
38d5ea9f 413######## test non-serial auto-pk
4d4dc518 414
bbdda281 415 if ($schema->storage->_use_insert_returning) {
bab40dee 416 $schema->source('TimestampPrimaryKey')->name('dbic_t_schema.timestamp_primary_key_test');
417 my $row = $schema->resultset('TimestampPrimaryKey')->create({});
418 ok $row->id;
419 }
d093d3eb 420
be860760 421######## test with_deferred_fk_checks
422
bab40dee 423 $schema->source('CD')->name('dbic_t_schema.cd');
424 $schema->source('Track')->name('dbic_t_schema.track');
425 lives_ok {
426 $schema->storage->with_deferred_fk_checks(sub {
427 $schema->resultset('Track')->create({
428 trackid => 999, cd => 999, position => 1, title => 'deferred FK track'
429 });
430 $schema->resultset('CD')->create({
431 artist => 1, cdid => 999, year => '2003', title => 'deferred FK cd'
432 });
be860760 433 });
bab40dee 434 } 'with_deferred_fk_checks code survived';
be860760 435
bab40dee 436 is eval { $schema->resultset('Track')->find(999)->title }, 'deferred FK track',
8273e845 437 'code in with_deferred_fk_checks worked';
be860760 438
bab40dee 439 throws_ok {
440 $schema->resultset('Track')->create({
441 trackid => 1, cd => 9999, position => 1, title => 'Track1'
442 });
443 } qr/constraint/i, 'with_deferred_fk_checks is off';
444}
be860760 445
2fbf50ff 446done_testing;
609c5f1b 447
5935c90a 448END {
f66c9ba8 449 return unless $schema;
5935c90a 450 drop_test_schema($schema);
65d35121 451 eapk_drop_all($schema);
452 undef $schema;
5935c90a 453};
70a201a5 454
455
456######### SUBROUTINES
457
458sub create_test_schema {
2fbf50ff 459 my $schema = shift;
460 $schema->storage->dbh_do(sub {
461 my (undef,$dbh) = @_;
70a201a5 462
2fbf50ff 463 local $dbh->{Warn} = 0;
70a201a5 464
2fbf50ff 465 my $std_artist_table = <<EOS;
70a201a5 466(
467 artistid serial PRIMARY KEY
468 , name VARCHAR(100)
469 , rank INTEGER NOT NULL DEFAULT '13'
470 , charfield CHAR(10)
471 , arrayfield INTEGER[]
472)
473EOS
474
881def48 475 $dbh->do("CREATE SCHEMA dbic_t_schema");
476 $dbh->do("CREATE TABLE dbic_t_schema.artist $std_artist_table");
4d4dc518 477
478 $dbh->do(<<EOS);
479CREATE TABLE dbic_t_schema.timestamp_primary_key_test (
be860760 480 id timestamp default current_timestamp
4d4dc518 481)
482EOS
2fbf50ff 483 $dbh->do(<<EOS);
be860760 484CREATE TABLE dbic_t_schema.cd (
485 cdid int PRIMARY KEY,
486 artist int,
487 title varchar(255),
488 year varchar(4),
489 genreid int,
490 single_track int
491)
492EOS
493 $dbh->do(<<EOS);
494CREATE TABLE dbic_t_schema.track (
495 trackid int,
496 cd int REFERENCES dbic_t_schema.cd(cdid) DEFERRABLE,
497 position int,
498 title varchar(255),
499 last_updated_on date,
3d98c75e 500 last_updated_at date
be860760 501)
502EOS
503
504 $dbh->do(<<EOS);
881def48 505CREATE TABLE dbic_t_schema.sequence_test (
70a201a5 506 pkid1 integer
507 , pkid2 integer
508 , nonpkid integer
509 , name VARCHAR(100)
510 , CONSTRAINT pk PRIMARY KEY(pkid1, pkid2)
511)
512EOS
2fbf50ff 513 $dbh->do("CREATE SEQUENCE pkid1_seq START 1 MAXVALUE 999999 MINVALUE 0");
514 $dbh->do("CREATE SEQUENCE pkid2_seq START 10 MAXVALUE 999999 MINVALUE 0");
515 $dbh->do("CREATE SEQUENCE nonpkid_seq START 20 MAXVALUE 999999 MINVALUE 0");
516 $dbh->do(<<EOS);
881def48 517CREATE TABLE dbic_t_schema.casecheck (
70a201a5 518 id serial PRIMARY KEY
519 , "name" VARCHAR(1)
520 , "NAME" VARCHAR(2)
521 , "UC_NAME" VARCHAR(3)
70a201a5 522)
523EOS
2fbf50ff 524 $dbh->do(<<EOS);
881def48 525CREATE TABLE dbic_t_schema.array_test (
70a201a5 526 id serial PRIMARY KEY
527 , arrayfield INTEGER[]
528)
529EOS
881def48 530 $dbh->do("CREATE SCHEMA dbic_t_schema_2");
531 $dbh->do("CREATE TABLE dbic_t_schema_2.artist $std_artist_table");
532 $dbh->do("CREATE SCHEMA dbic_t_schema_3");
533 $dbh->do("CREATE TABLE dbic_t_schema_3.artist $std_artist_table");
534 $dbh->do('set search_path=dbic_t_schema,public');
535 $dbh->do("CREATE SCHEMA dbic_t_schema_4");
536 $dbh->do("CREATE SCHEMA dbic_t_schema_5");
2fbf50ff 537 $dbh->do(<<EOS);
881def48 538 CREATE TABLE dbic_t_schema_4.artist
70a201a5 539 (
540 artistid integer not null default nextval('artist_artistid_seq'::regclass) PRIMARY KEY
541 , name VARCHAR(100)
542 , rank INTEGER NOT NULL DEFAULT '13'
543 , charfield CHAR(10)
544 , arrayfield INTEGER[]
545 );
546EOS
881def48 547 $dbh->do('set search_path=public,dbic_t_schema,dbic_t_schema_3');
2fbf50ff 548 $dbh->do('create sequence public.artist_artistid_seq'); #< in the public schema
549 $dbh->do(<<EOS);
881def48 550 CREATE TABLE dbic_t_schema_5.artist
70a201a5 551 (
552 artistid integer not null default nextval('public.artist_artistid_seq'::regclass) PRIMARY KEY
553 , name VARCHAR(100)
554 , rank INTEGER NOT NULL DEFAULT '13'
555 , charfield CHAR(10)
556 , arrayfield INTEGER[]
557 );
558EOS
881def48 559 $dbh->do('set search_path=dbic_t_schema,public');
2fbf50ff 560 });
70a201a5 561}
562
563
564
565sub drop_test_schema {
52c53388 566 my ( $schema, $warn_exceptions ) = @_;
2fbf50ff 567
568 $schema->storage->dbh_do(sub {
569 my (undef,$dbh) = @_;
570
571 local $dbh->{Warn} = 0;
572
573 for my $stat (
881def48 574 'DROP SCHEMA dbic_t_schema_5 CASCADE',
e78c769d 575 'DROP SEQUENCE public.artist_artistid_seq CASCADE',
881def48 576 'DROP SCHEMA dbic_t_schema_4 CASCADE',
577 'DROP SCHEMA dbic_t_schema CASCADE',
e78c769d 578 'DROP SEQUENCE pkid1_seq CASCADE',
579 'DROP SEQUENCE pkid2_seq CASCADE',
580 'DROP SEQUENCE nonpkid_seq CASCADE',
881def48 581 'DROP SCHEMA dbic_t_schema_2 CASCADE',
582 'DROP SCHEMA dbic_t_schema_3 CASCADE',
2fbf50ff 583 ) {
584 eval { $dbh->do ($stat) };
52c53388 585 diag $@ if $@ && $warn_exceptions;
2fbf50ff 586 }
587 });
3ff5b740 588}
7603d2a5 589
372da819 590
ee9f372c 591### auto-pk / last_insert_id / sequence discovery
592sub run_apk_tests {
593 my $schema = shift;
594
595 # This is in Core now, but it's here just to test that it doesn't break
596 $schema->class('Artist')->load_components('PK::Auto');
597 cmp_ok( $schema->resultset('Artist')->count, '==', 0, 'this should start with an empty artist table');
598
599 # test that auto-pk also works with the defined search path by
600 # un-schema-qualifying the table name
601 apk_t_set($schema,'artist');
602
603 my $unq_new;
604 lives_ok {
605 $unq_new = $schema->resultset('Artist')->create({ name => 'baz' });
606 } 'insert into unqualified, shadowed table succeeds';
607
608 is($unq_new && $unq_new->artistid, 1, "and got correct artistid");
609
610 my @test_schemas = ( [qw| dbic_t_schema_2 1 |],
611 [qw| dbic_t_schema_3 1 |],
612 [qw| dbic_t_schema_4 2 |],
613 [qw| dbic_t_schema_5 1 |],
614 );
615 foreach my $t ( @test_schemas ) {
616 my ($sch_name, $start_num) = @$t;
617 #test with dbic_t_schema_2
618 apk_t_set($schema,"$sch_name.artist");
619 my $another_new;
620 lives_ok {
621 $another_new = $schema->resultset('Artist')->create({ name => 'Tollbooth Willy'});
622 is( $another_new->artistid,$start_num, "got correct artistid for $sch_name")
623 or diag "USED SEQUENCE: ".($schema->source('Artist')->column_info('artistid')->{sequence} || '<none>');
624 } "$sch_name liid 1 did not die"
625 or diag "USED SEQUENCE: ".($schema->source('Artist')->column_info('artistid')->{sequence} || '<none>');
626 lives_ok {
627 $another_new = $schema->resultset('Artist')->create({ name => 'Adam Sandler'});
628 is( $another_new->artistid,$start_num+1, "got correct artistid for $sch_name")
629 or diag "USED SEQUENCE: ".($schema->source('Artist')->column_info('artistid')->{sequence} || '<none>');
630 } "$sch_name liid 2 did not die"
631 or diag "USED SEQUENCE: ".($schema->source('Artist')->column_info('artistid')->{sequence} || '<none>');
632
633 }
634
635 lives_ok {
636 apk_t_set($schema,'dbic_t_schema.artist');
637 my $new = $schema->resultset('Artist')->create({ name => 'foo' });
638 is($new->artistid, 4, "Auto-PK worked");
639 $new = $schema->resultset('Artist')->create({ name => 'bar' });
640 is($new->artistid, 5, "Auto-PK worked");
641 } 'old auto-pk tests did not die either';
642}
643
ee9f372c 644# sets the artist table name and clears sequence name cache
645sub apk_t_set {
646 my ( $s, $n ) = @_;
647 $s->source("Artist")->name($n);
648 $s->source('Artist')->column_info('artistid')->{sequence} = undef; #< clear sequence name cache
649}
650
372da819 651
5935c90a 652######## EXTENDED AUTO-PK TESTS
653
f295a73c 654my @eapk_id_columns;
5935c90a 655BEGIN {
656 package DBICTest::Schema::ExtAPK;
657 push @main::test_classes, __PACKAGE__;
658
659 use strict;
660 use warnings;
d88ecca6 661 use base 'DBIx::Class::Core';
5935c90a 662
f295a73c 663 __PACKAGE__->table('apk');
5935c90a 664
f295a73c 665 @eapk_id_columns = qw( id1 id2 id3 id4 );
5935c90a 666 __PACKAGE__->add_columns(
667 map { $_ => { data_type => 'integer', is_auto_increment => 1 } }
f295a73c 668 @eapk_id_columns
5935c90a 669 );
670
f295a73c 671 __PACKAGE__->set_primary_key('id2'); #< note the SECOND column is
672 #the primary key
5935c90a 673}
674
f295a73c 675my @eapk_schemas;
676BEGIN{ @eapk_schemas = map "dbic_apk_$_", 0..5 }
ae645c61 677my %seqs; #< hash of schema.table.col => currval of its (DBIC) primary key sequence
5935c90a 678
372da819 679sub run_extended_apk_tests {
11da4e66 680 my $schema = shift;
372da819 681
f295a73c 682 #save the search path and reset it at the end
49aa7e87 683 my $search_path_save = eapk_get_search_path($schema);
f295a73c 684
52c53388 685 eapk_drop_all($schema);
bab40dee 686 %seqs = ();
5935c90a 687
f295a73c 688 # make the test schemas and sequences
11da4e66 689 $schema->storage->dbh_do(sub {
f295a73c 690 my ( undef, $dbh ) = @_;
691
692 $dbh->do("CREATE SCHEMA $_")
693 for @eapk_schemas;
694
695 $dbh->do("CREATE SEQUENCE $eapk_schemas[5].fooseq");
ae645c61 696 $dbh->do("SELECT setval('$eapk_schemas[5].fooseq',400)");
697 $seqs{"$eapk_schemas[1].apk.id2"} = 400;
698
fb7be534 699 $dbh->do("CREATE SEQUENCE $eapk_schemas[4].fooseq");
ae645c61 700 $dbh->do("SELECT setval('$eapk_schemas[4].fooseq',300)");
701 $seqs{"$eapk_schemas[3].apk.id2"} = 300;
702
93a6e8fa 703 $dbh->do("CREATE SEQUENCE $eapk_schemas[3].fooseq");
ae645c61 704 $dbh->do("SELECT setval('$eapk_schemas[3].fooseq',200)");
705 $seqs{"$eapk_schemas[4].apk.id2"} = 200;
f295a73c 706
78c9596c 707 $dbh->do("SET search_path = ".join ',', reverse @eapk_schemas );
11da4e66 708 });
372da819 709
f295a73c 710 # clear our search_path cache
711 $schema->storage->{_pg_search_path} = undef;
712
713 eapk_create( $schema,
714 with_search_path => [0,1],
715 );
716 eapk_create( $schema,
717 with_search_path => [1,0,'public'],
718 nextval => "$eapk_schemas[5].fooseq",
719 );
720 eapk_create( $schema,
721 with_search_path => ['public',0,1],
722 qualify_table => 2,
723 );
fb7be534 724 eapk_create( $schema,
725 with_search_path => [3,1,0,'public'],
726 nextval => "$eapk_schemas[4].fooseq",
727 );
93a6e8fa 728 eapk_create( $schema,
729 with_search_path => [3,1,0,'public'],
730 nextval => "$eapk_schemas[3].fooseq",
731 qualify_table => 4,
732 );
372da819 733
78c9596c 734 eapk_poke( $schema );
f295a73c 735 eapk_poke( $schema, 0 );
eb80e8d5 736 eapk_poke( $schema, 2 );
93a6e8fa 737 eapk_poke( $schema, 4 );
f295a73c 738 eapk_poke( $schema, 1 );
eb80e8d5 739 eapk_poke( $schema, 0 );
f295a73c 740 eapk_poke( $schema, 1 );
78c9596c 741 eapk_poke( $schema );
93a6e8fa 742 eapk_poke( $schema, 4 );
fb7be534 743 eapk_poke( $schema, 3 );
eb80e8d5 744 eapk_poke( $schema, 1 );
745 eapk_poke( $schema, 2 );
746 eapk_poke( $schema, 0 );
f295a73c 747
748 # set our search path back
749 eapk_set_search_path( $schema, @$search_path_save );
750}
751
752# do a DBIC create on the apk table in the given schema number (which is an
753# index of @eapk_schemas)
754
f295a73c 755sub eapk_poke {
756 my ($s, $schema_num) = @_;
372da819 757
f295a73c 758 my $schema_name = defined $schema_num
759 ? $eapk_schemas[$schema_num]
760 : '';
761
78c9596c 762 my $schema_name_actual = $schema_name || eapk_find_visible_schema($s);
f295a73c 763
8ce84173 764 $s->source('ExtAPK')->name($schema_name ? $schema_name.'.apk' : 'apk');
f295a73c 765 #< clear sequence name cache
766 $s->source('ExtAPK')->column_info($_)->{sequence} = undef
767 for @eapk_id_columns;
768
769 no warnings 'uninitialized';
770 lives_ok {
771 my $new;
772 for my $inc (1,2,3) {
ae645c61 773 $new = $schema->resultset('ExtAPK')->create({ id1 => 1});
691f3663 774 my $proper_seqval = ++$seqs{"$schema_name_actual.apk.id2"};
775 is( $new->id2, $proper_seqval, "$schema_name_actual.apk.id2 correct inc $inc" )
776 or eapk_seq_diag($s,$schema_name);
777 $new->discard_changes;
ae645c61 778 is( $new->id1, 1 );
779 for my $id ('id3','id4') {
f295a73c 780 my $proper_seqval = ++$seqs{"$schema_name_actual.apk.$id"};
93a6e8fa 781 is( $new->$id, $proper_seqval, "$schema_name_actual.apk.$id correct inc $inc" )
fb7be534 782 or eapk_seq_diag($s,$schema_name);
f295a73c 783 }
784 }
785 } "create in schema '$schema_name' lives"
fb7be534 786 or eapk_seq_diag($s,$schema_name);
372da819 787}
788
f295a73c 789# print diagnostic info on which sequences were found in the ExtAPK
790# class
791sub eapk_seq_diag {
792 my $s = shift;
78c9596c 793 my $schema = shift || eapk_find_visible_schema($s);
f295a73c 794
795 diag "$schema.apk sequences: ",
796 join(', ',
797 map "$_:".($s->source('ExtAPK')->column_info($_)->{sequence} || '<none>'),
798 @eapk_id_columns
799 );
800}
801
49aa7e87 802# get the postgres search path as an arrayref
803sub eapk_get_search_path {
804 my ( $s ) = @_;
805 # cache the search path as ['schema','schema',...] in the storage
806 # obj
807
808 return $s->storage->dbh_do(sub {
809 my (undef, $dbh) = @_;
810 my @search_path;
811 my ($sp_string) = $dbh->selectrow_array('SHOW search_path');
812 while ( $sp_string =~ s/("[^"]+"|[^,]+),?// ) {
813 unless( defined $1 and length $1 ) {
814 die "search path sanity check failed: '$1'";
815 }
816 push @search_path, $1;
817 }
818 \@search_path
819 });
820}
f295a73c 821sub eapk_set_search_path {
822 my ($s,@sp) = @_;
823 my $sp = join ',',@sp;
824 $s->storage->dbh_do( sub { $_[1]->do("SET search_path = $sp") } );
825}
826
827# create the apk table in the given schema, can set whether the table name is qualified, what the nextval is for the second ID
5935c90a 828sub eapk_create {
829 my ($schema, %a) = @_;
830
372da819 831 $schema->storage->dbh_do(sub {
5935c90a 832 my (undef,$dbh) = @_;
372da819 833
5935c90a 834 my $searchpath_save;
835 if ( $a{with_search_path} ) {
836 ($searchpath_save) = $dbh->selectrow_array('SHOW search_path');
837
f295a73c 838 my $search_path = join ',',map {/\D/ ? $_ : $eapk_schemas[$_]} @{$a{with_search_path}};
5935c90a 839
840 $dbh->do("SET search_path = $search_path");
841 }
842
f295a73c 843 my $table_name = $a{qualify_table}
844 ? ($eapk_schemas[$a{qualify_table}] || die). ".apk"
845 : 'apk';
e25fe566 846 local $_[1]->{Warn} = 0;
f295a73c 847
848 my $id_def = $a{nextval}
ae645c61 849 ? "integer not null default nextval('$a{nextval}'::regclass)"
850 : 'serial';
5935c90a 851 $dbh->do(<<EOS);
f295a73c 852CREATE TABLE $table_name (
853 id1 serial
854 , id2 $id_def
ae645c61 855 , id3 serial primary key
5935c90a 856 , id4 serial
857)
858EOS
372da819 859
5935c90a 860 if( $searchpath_save ) {
861 $dbh->do("SET search_path = $searchpath_save");
862 }
372da819 863 });
864}
865
5935c90a 866sub eapk_drop_all {
52c53388 867 my ( $schema, $warn_exceptions ) = @_;
372da819 868
869 $schema->storage->dbh_do(sub {
870 my (undef,$dbh) = @_;
871
872 local $dbh->{Warn} = 0;
5935c90a 873
874 # drop the test schemas
f295a73c 875 for (@eapk_schemas ) {
11da4e66 876 eval{ $dbh->do("DROP SCHEMA $_ CASCADE") };
52c53388 877 diag $@ if $@ && $warn_exceptions;
11da4e66 878 }
879
372da819 880
372da819 881 });
882}
78c9596c 883
884sub eapk_find_visible_schema {
885 my ($s) = @_;
886
887 my ($schema) =
888 $s->storage->dbh_do(sub {
889 $_[1]->selectrow_array(<<EOS);
890SELECT n.nspname
891FROM pg_catalog.pg_namespace n
892JOIN pg_catalog.pg_class c ON c.relnamespace = n.oid
893WHERE c.relname = 'apk'
894 AND pg_catalog.pg_table_is_visible(c.oid)
895EOS
896 });
897 return $schema;
898}