-
Notifications
You must be signed in to change notification settings - Fork 151
/
72pg_bytea.t
128 lines (100 loc) · 3.63 KB
/
72pg_bytea.t
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
BEGIN { do "./t/lib/ANFANG.pm" or die ( $@ || $! ) }
use DBIx::Class::Optional::Dependencies -skip_all_without => qw(test_rdbms_pg binary_data);
use strict;
use warnings;
use Test::More;
use DBIx::Class::_Util 'modver_gt_or_eq';
use DBICTest;
my ($dsn, $dbuser, $dbpass) = @ENV{map { "DBICTEST_PG_${_}" } qw/DSN USER PASS/};
my $schema = DBICTest::Schema->connect($dsn, $dbuser, $dbpass, { AutoCommit => 1 });
plan skip_all => 'DBD::Pg < 2.17.2 does not work with Pg >= 9.0 BYTEA columns' if (
! modver_gt_or_eq('DBD::Pg', '2.17.2')
and
$schema->storage->_server_info->{normalized_dbms_version} >= 9.0
);
my $dbh = $schema->storage->dbh;
{
local $SIG{__WARN__} = sub {};
$dbh->do('DROP TABLE IF EXISTS bindtype_test');
# the blob/clob are for reference only, will be useful when we switch to SQLT and can test Oracle along the way
$dbh->do(qq[
CREATE TABLE bindtype_test
(
id serial NOT NULL PRIMARY KEY,
bytea bytea NULL,
blob bytea NULL,
clob text NULL,
a_memo text NULL
);
],{ RaiseError => 1, PrintError => 1 });
}
$schema->storage->debug(0); # these tests spew up way too much stuff, disable trace
my $big_long_string = "\x00\x01\x02 abcd" x 125000;
my $new;
# test inserting a row
{
$new = $schema->resultset('BindType')->create({ bytea => $big_long_string });
ok($new->id, "Created a bytea row");
ok($new->bytea eq $big_long_string, "Set the blob correctly.");
}
# test retrieval of the bytea column
{
my $row = $schema->resultset('BindType')->find({ id => $new->id });
ok($row->get_column('bytea') eq $big_long_string, "Created the blob correctly.");
}
{
my $rs = $schema->resultset('BindType')->search({ bytea => $big_long_string });
# search on the bytea column (select)
{
my $row = $rs->first;
is($row ? $row->id : undef, $new->id, "Found the row searching on the bytea column.");
}
# search on the bytea column (update)
{
my $new_big_long_string = $big_long_string . "2";
$schema->txn_do(sub {
$rs->update({ bytea => $new_big_long_string });
my $row = $schema->resultset('BindType')->find({ id => $new->id });
ok( ($row ? $row->get_column('bytea') : '') eq $new_big_long_string,
"Updated the row correctly (searching on the bytea column)."
);
$schema->txn_rollback;
});
}
# search on the bytea column (delete)
{
$schema->txn_do(sub {
$rs->delete;
my $row = $schema->resultset('BindType')->find({ id => $new->id });
is($row, undef, "Deleted the row correctly (searching on the bytea column).");
$schema->txn_rollback;
});
}
# create with blob from $rs
$new = $rs->create({});
ok($new->bytea eq $big_long_string, 'Object has bytea value from $rs');
$new->discard_changes;
ok($new->bytea eq $big_long_string, 'bytea value made it to db');
}
# test inserting a row via populate() (bindtype propagation through execute_for_fetch)
# use a new $dbh to ensure no leakage due to prepare_cached
{
my $cnt = 4;
$schema->storage->_dbh(undef);
my $rs = $schema->resultset('BindType');
$rs->delete;
$rs->populate([
[qw/id bytea/],
map { [
\[ '?', [ {} => $_ ] ],
"pop_${_}_" . $big_long_string,
]} (1 .. $cnt)
]);
is($rs->count, $cnt, 'All rows were correctly inserted');
for (1..$cnt) {
my $r = $rs->find({ bytea => "pop_${_}_" . $big_long_string });
is ($r->id, $_, "Row $_ found after find() on the blob");
}
}
done_testing;
eval { $schema->storage->dbh_do(sub { $_[1]->do("DROP TABLE bindtype_test") } ) };