Skip error/warn frames within CAG - saner callsite error messages this way
[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 {
94 my $v = DBICTest::Schema->connect($dsn, $user, $pass)
95 ->storage
96 ->_get_dbh
97 ->get_info(18);
98 $v =~ /^(\d+)\.(\d+)/
99 or die "Unparseable Pg server version: $v\n";
100
101 ( sprintf ('%d.%d', $1, $2) >= 8.2 ) ? 1 : 0;
102};
103is (
104 DBICTest::Schema->connect($dsn, $user, $pass)->storage->_use_insert_returning,
105 $test_server_supports_insert_returning,
106 'insert returning capability guessed correctly'
107);
108
109my $schema;
110for my $use_insert_returning ($test_server_supports_insert_returning
111 ? (0,1)
112 : (0)
113) {
114
115 no warnings qw/once/;
116 local *DBICTest::Schema::connection = subname 'DBICTest::Schema::connection' => sub {
117 my $s = shift->next::method (@_);
118 $s->storage->_use_insert_returning ($use_insert_returning);
119 $s;
120 };
121
122### test capability override
123 {
124 my $s = DBICTest::Schema->connect($dsn, $user, $pass);
125
126 ok (!$s->storage->_dbh, 'definitely not connected');
127
128 ok (
129 ! ($s->storage->_use_insert_returning xor $use_insert_returning),
130 'insert returning capability set correctly',
131 );
132 ok (!$s->storage->_dbh, 'still not connected (capability override works)');
133 }
134
70a201a5 135### connect, create postgres-specific test schema
c3af542a 136
bab40dee 137 $schema = DBICTest::Schema->connect($dsn, $user, $pass);
1f5aae08 138 $schema->storage->ensure_connected;
139
bab40dee 140 drop_test_schema($schema);
141 create_test_schema($schema);
609c5f1b 142
70a201a5 143### begin main tests
4617b792 144
372da819 145# run a BIG bunch of tests for last-insert-id / Auto-PK / sequence
146# discovery
bab40dee 147 run_apk_tests($schema); #< older set of auto-pk tests
148 run_extended_apk_tests($schema); #< new extended set of auto-pk tests
70a201a5 149
bab40dee 150### type_info tests
52c53388 151
bab40dee 152 my $test_type_info = {
153 'artistid' => {
154 'data_type' => 'integer',
155 'is_nullable' => 0,
156 'size' => 4,
157 },
158 'name' => {
159 'data_type' => 'character varying',
160 'is_nullable' => 1,
161 'size' => 100,
162 'default_value' => undef,
163 },
164 'rank' => {
165 'data_type' => 'integer',
166 'is_nullable' => 0,
167 'size' => 4,
168 'default_value' => 13,
52c53388 169
bab40dee 170 },
171 'charfield' => {
172 'data_type' => 'character',
173 'is_nullable' => 1,
174 'size' => 10,
175 'default_value' => undef,
176 },
177 'arrayfield' => {
178 'data_type' => 'integer[]',
179 'is_nullable' => 1,
180 'size' => undef,
181 'default_value' => undef,
182 },
183 };
52c53388 184
bab40dee 185 my $type_info = $schema->storage->columns_info_for('dbic_t_schema.artist');
186 my $artistid_defval = delete $type_info->{artistid}->{default_value};
187 like($artistid_defval,
188 qr/^nextval\('([^\.]*\.){0,1}artist_artistid_seq'::(?:text|regclass)\)/,
189 'columns_info_for - sequence matches Pg get_autoinc_seq expectations');
190 is_deeply($type_info, $test_type_info,
191 'columns_info_for - column data types');
52c53388 192
70a201a5 193
a953d8d9 194
a953d8d9 195
bab40dee 196####### Array tests
70a201a5 197
bab40dee 198 BEGIN {
199 package DBICTest::Schema::ArrayTest;
200 push @main::test_classes, __PACKAGE__;
70a201a5 201
bab40dee 202 use strict;
203 use warnings;
41519379 204 use base 'DBICTest::BaseResult';
70a201a5 205
bab40dee 206 __PACKAGE__->table('dbic_t_schema.array_test');
207 __PACKAGE__->add_columns(qw/id arrayfield/);
208 __PACKAGE__->column_info_from_storage(1);
209 __PACKAGE__->set_primary_key('id');
70a201a5 210
bab40dee 211 }
212 SKIP: {
0647e0cc 213 skip "Need DBD::Pg 2.9.2 or newer for array tests", 4 if $DBD::Pg::VERSION < 2.009002;
70a201a5 214
41519379 215 my $arr_rs = $schema->resultset('ArrayTest');
216
bab40dee 217 lives_ok {
41519379 218 $arr_rs->create({
bab40dee 219 arrayfield => [1, 2],
220 });
221 } 'inserting arrayref as pg array data';
70a201a5 222
bab40dee 223 lives_ok {
41519379 224 $arr_rs->update({
bab40dee 225 arrayfield => [3, 4],
226 });
227 } 'updating arrayref as pg array data';
ac45262b 228
41519379 229 $arr_rs->create({
0647e0cc 230 arrayfield => [5, 6],
9ba627e3 231 });
c224117b 232
f6faeab8 233 lives_ok {
234 $schema->populate('ArrayTest', [
235 [ qw/arrayfield/ ],
236 [ [0,0] ],
237 ]);
238 } 'inserting arrayref using void ctx populate';
239
41519379 240 # Search using arrays
241 lives_ok {
242 is_deeply (
243 $arr_rs->search({ arrayfield => { -value => [3,4] } })->first->arrayfield,
244 [3,4],
245 'Array value matches'
246 );
247 } 'searching by arrayref';
a780a0ff 248
bab40dee 249 lives_ok {
41519379 250 is_deeply (
251 $arr_rs->search({ arrayfield => { '=' => { -value => [3,4] }} })->first->arrayfield,
252 [3,4],,
253 'Array value matches explicit equal'
254 );
255 } 'searching by arrayref (explicit equal sign)';
a780a0ff 256
41519379 257 lives_ok {
258 is_deeply (
259 $arr_rs->search({ arrayfield => { '>' => { -value => [3,1] }} })->first->arrayfield,
260 [3,4],
261 'Array value matches greater than'
262 );
263 } 'searching by arrayref (greater than)';
264
265 lives_ok {
266 is (
267 $arr_rs->search({ arrayfield => { '>' => { -value => [3,7] }} })->count,
268 1,
269 'Greater than search found [5,6]',
270 );
271 } 'searching by arrayref (greater than)';
272
273 # Find using arrays
274 lives_ok {
275 is_deeply (
276 $arr_rs->find({ arrayfield => { -value => [3,4] } })->arrayfield,
277 [3,4],
278 'Array value matches implicit equal'
279 );
280 } 'find by arrayref';
281
282 lives_ok {
283 is_deeply (
284 $arr_rs->find({ arrayfield => { '=' => { -value => [3,4] }} })->arrayfield,
285 [3,4],
286 'Array value matches explicit equal'
287 );
288 } 'find by arrayref (equal)';
289
290 # test inferred condition for creation
291 TODO: for my $cond (
292 { -value => [3,4] },
293 \[ '= ?' => [arrayfield => [3, 4]] ],
294 ) {
295 local $TODO = 'No introspection of complex conditions :(';
296 my $arr_rs_cond = $arr_rs->search({ arrayfield => $cond });
297
298 my $row = $arr_rs_cond->create({});
a780a0ff 299 is_deeply ($row->arrayfield, [3,4], 'Array value taken from $rs condition');
300 $row->discard_changes;
301 is_deeply ($row->arrayfield, [3,4], 'Array value made it to storage');
302 }
bab40dee 303 }
9ba627e3 304
70a201a5 305########## Case check
306
bab40dee 307 BEGIN {
308 package DBICTest::Schema::Casecheck;
309 push @main::test_classes, __PACKAGE__;
70a201a5 310
bab40dee 311 use strict;
312 use warnings;
313 use base 'DBIx::Class::Core';
70a201a5 314
bab40dee 315 __PACKAGE__->table('dbic_t_schema.casecheck');
316 __PACKAGE__->add_columns(qw/id name NAME uc_name/);
317 __PACKAGE__->column_info_from_storage(1);
318 __PACKAGE__->set_primary_key('id');
319 }
70a201a5 320
bab40dee 321 my $name_info = $schema->source('Casecheck')->column_info( 'name' );
322 is( $name_info->{size}, 1, "Case sensitive matching info for 'name'" );
ae515736 323
bab40dee 324 my $NAME_info = $schema->source('Casecheck')->column_info( 'NAME' );
325 is( $NAME_info->{size}, 2, "Case sensitive matching info for 'NAME'" );
ae515736 326
bab40dee 327 my $uc_name_info = $schema->source('Casecheck')->column_info( 'uc_name' );
328 is( $uc_name_info->{size}, 3, "Case insensitive matching info for 'uc_name'" );
ae515736 329
70a201a5 330
ecccae8a 331## Test ResultSet->update
332my $artist = $schema->resultset('Artist')->first;
333my $cds = $artist->cds_unordered->search({
334 year => { '!=' => 2010 }
335}, { prefetch => 'liner_notes' });
67b79ae1 336lives_ok { $cds->update({ year => '2010' }) } 'Update on prefetched rs';
70a201a5 337
338## Test SELECT ... FOR UPDATE
339
bab40dee 340 SKIP: {
67b79ae1 341 if(eval { require Sys::SigAction }) {
bab40dee 342 Sys::SigAction->import( 'set_sig_handler' );
343 }
344 else {
345 skip "Sys::SigAction is not available", 6;
346 }
347
348 my ($timed_out, $artist2);
349
350 for my $t (
351 {
352 # Make sure that an error was raised, and that the update failed
353 update_lock => 1,
354 test_sub => sub {
355 ok($timed_out, "update from second schema times out");
356 ok($artist2->is_column_changed('name'), "'name' column is still dirty from second schema");
357 },
52c53388 358 },
bab40dee 359 {
360 # Make sure that an error was NOT raised, and that the update succeeded
361 update_lock => 0,
362 test_sub => sub {
363 ok(! $timed_out, "update from second schema DOES NOT timeout");
364 ok(! $artist2->is_column_changed('name'), "'name' column is NOT dirty from second schema");
365 },
52c53388 366 },
bab40dee 367 ) {
368 # create a new schema
369 my $schema2 = DBICTest::Schema->connect($dsn, $user, $pass);
370 $schema2->source("Artist")->name("dbic_t_schema.artist");
371
372 $schema->txn_do( sub {
d2408067 373 my $rs = $schema->resultset('Artist')->search(
bab40dee 374 {
375 artistid => 1
376 },
377 $t->{update_lock} ? { for => 'update' } : {}
d2408067 378 );
379 ok ($rs->count, 'Count works');
380
381 my $artist = $rs->next;
bab40dee 382 is($artist->artistid, 1, "select returns artistid = 1");
383
384 $timed_out = 0;
385 eval {
386 my $h = set_sig_handler( 'ALRM', sub { die "DBICTestTimeout" } );
387 alarm(2);
388 $artist2 = $schema2->resultset('Artist')->find(1);
389 $artist2->name('fooey');
390 $artist2->update;
391 alarm(0);
392 };
393 $timed_out = $@ =~ /DBICTestTimeout/;
394 });
395
396 $t->{test_sub}->();
397 }
398 }
95ba7ee4 399
70a201a5 400
372da819 401######## other older Auto-pk tests
70a201a5 402
bab40dee 403 $schema->source("SequenceTest")->name("dbic_t_schema.sequence_test");
404 for (1..5) {
405 my $st = $schema->resultset('SequenceTest')->create({ name => 'foo' });
406 is($st->pkid1, $_, "Auto-PK for sequence without default: First primary key");
407 is($st->pkid2, $_ + 9, "Auto-PK for sequence without default: Second primary key");
408 is($st->nonpkid, $_ + 19, "Auto-PK for sequence without default: Non-primary key");
409 }
410 my $st = $schema->resultset('SequenceTest')->create({ name => 'foo', pkid1 => 55 });
411 is($st->pkid1, 55, "Auto-PK for sequence without default: First primary key set manually");
4d4dc518 412
413
38d5ea9f 414######## test non-serial auto-pk
4d4dc518 415
bbdda281 416 if ($schema->storage->_use_insert_returning) {
bab40dee 417 $schema->source('TimestampPrimaryKey')->name('dbic_t_schema.timestamp_primary_key_test');
418 my $row = $schema->resultset('TimestampPrimaryKey')->create({});
419 ok $row->id;
420 }
d093d3eb 421
be860760 422######## test with_deferred_fk_checks
423
bab40dee 424 $schema->source('CD')->name('dbic_t_schema.cd');
425 $schema->source('Track')->name('dbic_t_schema.track');
426 lives_ok {
427 $schema->storage->with_deferred_fk_checks(sub {
428 $schema->resultset('Track')->create({
429 trackid => 999, cd => 999, position => 1, title => 'deferred FK track'
430 });
431 $schema->resultset('CD')->create({
432 artist => 1, cdid => 999, year => '2003', title => 'deferred FK cd'
433 });
be860760 434 });
bab40dee 435 } 'with_deferred_fk_checks code survived';
be860760 436
bab40dee 437 is eval { $schema->resultset('Track')->find(999)->title }, 'deferred FK track',
438 'code in with_deferred_fk_checks worked';
be860760 439
bab40dee 440 throws_ok {
441 $schema->resultset('Track')->create({
442 trackid => 1, cd => 9999, position => 1, title => 'Track1'
443 });
444 } qr/constraint/i, 'with_deferred_fk_checks is off';
445}
be860760 446
2fbf50ff 447done_testing;
609c5f1b 448
5935c90a 449END {
f66c9ba8 450 return unless $schema;
5935c90a 451 drop_test_schema($schema);
6a6dd46a 452 eapk_drop_all( $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}