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
|
require 'dbd/Pg'
require 'pg'
LEN = 50
class TestPostgresByteA < DBDConfig.testbase(:postgresql)
# FIXME the 'pg' module is broken and doesn't encode/decode properly.
# this test should prove that the 'pg' module works so we can back out our
# hacks.
def skip_underlying_driver
str = generate_random_string
encoded = PGconn.escape_bytea(str.dup)
decoded = PGconn.unescape_bytea(encoded)
assert_equal str, decoded
end
def test_encode_decode
bytea = DBI::DBD::Pg::Type::ByteA
# some specific cases that were failing intermittenly
# poor \\ handling
str = "\236\000\257\202G<\371\035TPEO\211\005*AH'H\3136\360\004\245\261\037\340u\003s\\772X\231\002\200\n\327\202\217\353\177r\317o\341\237\341"
encoded = bytea.escape_bytea(str)
decoded = bytea.parse(encoded)
assert_equal str, decoded
# the split hack not working
str = "\343\336e\260\337\373\314\026\323#\237i\035\0302\024\346X\274\016\324\371\206\036\230\374\206#rA\n\214\272\316\330\025\374\000\2663\244M\255x\360\002\266q\336\231"
encoded = bytea.escape_bytea(str)
decoded = bytea.parse(encoded)
assert_equal str, decoded
# delimiter at the end
str = "\343\336e\260\337\373\314\026\323#\237i\035\0302\024\346X\274\016\324\371\206\036\230\374\206#rA\n\214\272\316\330\025\374\000\2663\244M\255x\360\002\266q\336\231\\\\\\\\"
encoded = bytea.escape_bytea(str)
decoded = bytea.parse(encoded)
assert_equal str, decoded
# a huge test to weed out all the stragglers
5_000.times do
str = generate_random_string
encoded = bytea.escape_bytea(str)
decoded = bytea.parse(encoded)
assert_equal str, decoded
end
end
def test_insert_retrieve
assert_nothing_raised do
str = generate_random_string
sth = @dbh.prepare('insert into bytea_test (foo) values (?)')
sth.execute(DBI::DBD::Pg::Type::ByteA.new(str))
sth.finish
sth = @dbh.prepare('select * from bytea_test')
sth.execute
assert_equal([str], sth.fetch)
sth.finish
end
end
def generate_random_string
# random string test
str = " " * LEN
for i in 0...LEN
str[i] = (rand * 256).to_i.chr
end
return str
end
end
|
