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