pg 1.1.4
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +7 -0
- checksums.yaml.gz.sig +3 -0
- data.tar.gz.sig +0 -0
- data/.gemtest +0 -0
- data/BSDL +22 -0
- data/ChangeLog +6595 -0
- data/Contributors.rdoc +46 -0
- data/History.rdoc +492 -0
- data/LICENSE +56 -0
- data/Manifest.txt +72 -0
- data/POSTGRES +23 -0
- data/README-OS_X.rdoc +68 -0
- data/README-Windows.rdoc +56 -0
- data/README.ja.rdoc +14 -0
- data/README.rdoc +178 -0
- data/Rakefile +215 -0
- data/Rakefile.cross +298 -0
- data/ext/errorcodes.def +968 -0
- data/ext/errorcodes.rb +45 -0
- data/ext/errorcodes.txt +478 -0
- data/ext/extconf.rb +94 -0
- data/ext/gvl_wrappers.c +17 -0
- data/ext/gvl_wrappers.h +241 -0
- data/ext/pg.c +640 -0
- data/ext/pg.h +365 -0
- data/ext/pg_binary_decoder.c +229 -0
- data/ext/pg_binary_encoder.c +162 -0
- data/ext/pg_coder.c +549 -0
- data/ext/pg_connection.c +4252 -0
- data/ext/pg_copy_coder.c +596 -0
- data/ext/pg_errors.c +95 -0
- data/ext/pg_result.c +1501 -0
- data/ext/pg_text_decoder.c +981 -0
- data/ext/pg_text_encoder.c +682 -0
- data/ext/pg_tuple.c +541 -0
- data/ext/pg_type_map.c +166 -0
- data/ext/pg_type_map_all_strings.c +116 -0
- data/ext/pg_type_map_by_class.c +239 -0
- data/ext/pg_type_map_by_column.c +312 -0
- data/ext/pg_type_map_by_mri_type.c +284 -0
- data/ext/pg_type_map_by_oid.c +355 -0
- data/ext/pg_type_map_in_ruby.c +299 -0
- data/ext/util.c +149 -0
- data/ext/util.h +65 -0
- data/ext/vc/pg.sln +26 -0
- data/ext/vc/pg_18/pg.vcproj +216 -0
- data/ext/vc/pg_19/pg_19.vcproj +209 -0
- data/lib/pg.rb +74 -0
- data/lib/pg/basic_type_mapping.rb +459 -0
- data/lib/pg/binary_decoder.rb +22 -0
- data/lib/pg/coder.rb +83 -0
- data/lib/pg/connection.rb +291 -0
- data/lib/pg/constants.rb +11 -0
- data/lib/pg/exceptions.rb +11 -0
- data/lib/pg/result.rb +31 -0
- data/lib/pg/text_decoder.rb +47 -0
- data/lib/pg/text_encoder.rb +69 -0
- data/lib/pg/tuple.rb +30 -0
- data/lib/pg/type_map_by_column.rb +15 -0
- data/spec/data/expected_trace.out +26 -0
- data/spec/data/random_binary_data +0 -0
- data/spec/helpers.rb +380 -0
- data/spec/pg/basic_type_mapping_spec.rb +508 -0
- data/spec/pg/connection_spec.rb +1872 -0
- data/spec/pg/connection_sync_spec.rb +41 -0
- data/spec/pg/result_spec.rb +491 -0
- data/spec/pg/tuple_spec.rb +280 -0
- data/spec/pg/type_map_by_class_spec.rb +138 -0
- data/spec/pg/type_map_by_column_spec.rb +222 -0
- data/spec/pg/type_map_by_mri_type_spec.rb +136 -0
- data/spec/pg/type_map_by_oid_spec.rb +149 -0
- data/spec/pg/type_map_in_ruby_spec.rb +164 -0
- data/spec/pg/type_map_spec.rb +22 -0
- data/spec/pg/type_spec.rb +949 -0
- data/spec/pg_spec.rb +50 -0
- metadata +322 -0
- metadata.gz.sig +0 -0
@@ -0,0 +1,280 @@
|
|
1
|
+
# -*- rspec -*-
|
2
|
+
# encoding: utf-8
|
3
|
+
|
4
|
+
require_relative '../helpers'
|
5
|
+
require 'pg'
|
6
|
+
require 'objspace'
|
7
|
+
|
8
|
+
describe PG::Tuple do
|
9
|
+
let!(:typemap) { PG::BasicTypeMapForResults.new(@conn) }
|
10
|
+
let!(:result2x2) { @conn.exec( "VALUES(1, 'a'), (2, 'b')" ) }
|
11
|
+
let!(:result2x3cast) { @conn.exec( "SELECT * FROM (VALUES(1, TRUE, '3'), (2, FALSE, '4')) AS m (a, b, b)" ).map_types!(typemap) }
|
12
|
+
let!(:tuple0) { result2x2.tuple(0) }
|
13
|
+
let!(:tuple1) { result2x2.tuple(1) }
|
14
|
+
let!(:tuple2) { result2x3cast.tuple(0) }
|
15
|
+
let!(:tuple3) { str = Marshal.dump(result2x3cast.tuple(1)); Marshal.load(str) }
|
16
|
+
let!(:tuple_empty) { PG::Tuple.new }
|
17
|
+
|
18
|
+
describe "[]" do
|
19
|
+
it "returns nil for invalid keys" do
|
20
|
+
expect( tuple0["x"] ).to be_nil
|
21
|
+
expect( tuple0[0.5] ).to be_nil
|
22
|
+
expect( tuple0[2] ).to be_nil
|
23
|
+
expect( tuple0[-3] ).to be_nil
|
24
|
+
expect( tuple2[-4] ).to be_nil
|
25
|
+
expect{ tuple_empty[0] }.to raise_error(TypeError)
|
26
|
+
end
|
27
|
+
|
28
|
+
it "supports array like access" do
|
29
|
+
expect( tuple0[0] ).to eq( "1" )
|
30
|
+
expect( tuple0[1] ).to eq( "a" )
|
31
|
+
expect( tuple1[0] ).to eq( "2" )
|
32
|
+
expect( tuple1[1] ).to eq( "b" )
|
33
|
+
expect( tuple2[0] ).to eq( 1 )
|
34
|
+
expect( tuple2[1] ).to eq( true )
|
35
|
+
expect( tuple2[2] ).to eq( "3" )
|
36
|
+
expect( tuple3[0] ).to eq( 2 )
|
37
|
+
expect( tuple3[1] ).to eq( false )
|
38
|
+
expect( tuple3[2] ).to eq( "4" )
|
39
|
+
end
|
40
|
+
|
41
|
+
it "supports negative indices" do
|
42
|
+
expect( tuple0[-2] ).to eq( "1" )
|
43
|
+
expect( tuple0[-1] ).to eq( "a" )
|
44
|
+
expect( tuple2[-3] ).to eq( 1 )
|
45
|
+
expect( tuple2[-2] ).to eq( true )
|
46
|
+
expect( tuple2[-1] ).to eq( "3" )
|
47
|
+
end
|
48
|
+
|
49
|
+
it "supports hash like access" do
|
50
|
+
expect( tuple0["column1"] ).to eq( "1" )
|
51
|
+
expect( tuple0["column2"] ).to eq( "a" )
|
52
|
+
expect( tuple2["a"] ).to eq( 1 )
|
53
|
+
expect( tuple2["b"] ).to eq( "3" )
|
54
|
+
expect( tuple0["x"] ).to be_nil
|
55
|
+
end
|
56
|
+
|
57
|
+
it "casts lazy and caches result" do
|
58
|
+
a = []
|
59
|
+
deco = Class.new(PG::SimpleDecoder) do
|
60
|
+
define_method(:decode) do |*args|
|
61
|
+
a << args
|
62
|
+
args.last
|
63
|
+
end
|
64
|
+
end.new
|
65
|
+
|
66
|
+
result2x2.map_types!(PG::TypeMapByColumn.new([deco, deco]))
|
67
|
+
t = result2x2.tuple(1)
|
68
|
+
|
69
|
+
# cast and cache at first call to [0]
|
70
|
+
a.clear
|
71
|
+
expect( t[0] ).to eq( 0 )
|
72
|
+
expect( a ).to eq([["2", 1, 0]])
|
73
|
+
|
74
|
+
# use cache at second call to [0]
|
75
|
+
a.clear
|
76
|
+
expect( t[0] ).to eq( 0 )
|
77
|
+
expect( a ).to eq([])
|
78
|
+
|
79
|
+
# cast and cache at first call to [1]
|
80
|
+
a.clear
|
81
|
+
expect( t[1] ).to eq( 1 )
|
82
|
+
expect( a ).to eq([["b", 1, 1]])
|
83
|
+
end
|
84
|
+
end
|
85
|
+
|
86
|
+
describe "fetch" do
|
87
|
+
it "raises proper errors for invalid keys" do
|
88
|
+
expect{ tuple0.fetch("x") }.to raise_error(KeyError)
|
89
|
+
expect{ tuple0.fetch(0.5) }.to raise_error(KeyError)
|
90
|
+
expect{ tuple0.fetch(2) }.to raise_error(IndexError)
|
91
|
+
expect{ tuple0.fetch(-3) }.to raise_error(IndexError)
|
92
|
+
expect{ tuple0.fetch(-3) }.to raise_error(IndexError)
|
93
|
+
expect{ tuple2.fetch(-4) }.to raise_error(IndexError)
|
94
|
+
expect{ tuple_empty[0] }.to raise_error(TypeError)
|
95
|
+
end
|
96
|
+
|
97
|
+
it "supports array like access" do
|
98
|
+
expect( tuple0.fetch(0) ).to eq( "1" )
|
99
|
+
expect( tuple0.fetch(1) ).to eq( "a" )
|
100
|
+
expect( tuple2.fetch(0) ).to eq( 1 )
|
101
|
+
expect( tuple2.fetch(1) ).to eq( true )
|
102
|
+
expect( tuple2.fetch(2) ).to eq( "3" )
|
103
|
+
end
|
104
|
+
|
105
|
+
it "supports default value for indices" do
|
106
|
+
expect( tuple0.fetch(2, 42) ).to eq( 42 )
|
107
|
+
expect( tuple0.fetch(2){43} ).to eq( 43 )
|
108
|
+
end
|
109
|
+
|
110
|
+
it "supports negative indices" do
|
111
|
+
expect( tuple0.fetch(-2) ).to eq( "1" )
|
112
|
+
expect( tuple0.fetch(-1) ).to eq( "a" )
|
113
|
+
expect( tuple2.fetch(-3) ).to eq( 1 )
|
114
|
+
expect( tuple2.fetch(-2) ).to eq( true )
|
115
|
+
expect( tuple2.fetch(-1) ).to eq( "3" )
|
116
|
+
end
|
117
|
+
|
118
|
+
it "supports hash like access" do
|
119
|
+
expect( tuple0.fetch("column1") ).to eq( "1" )
|
120
|
+
expect( tuple0.fetch("column2") ).to eq( "a" )
|
121
|
+
expect( tuple2.fetch("a") ).to eq( 1 )
|
122
|
+
expect( tuple2.fetch("b") ).to eq( "3" )
|
123
|
+
end
|
124
|
+
|
125
|
+
it "supports default value for name keys" do
|
126
|
+
expect( tuple0.fetch("x", "defa") ).to eq("defa")
|
127
|
+
expect( tuple0.fetch("x"){"defa"} ).to eq("defa")
|
128
|
+
end
|
129
|
+
end
|
130
|
+
|
131
|
+
describe "each" do
|
132
|
+
it "can be used as an enumerator" do
|
133
|
+
expect( tuple0.each ).to be_kind_of(Enumerator)
|
134
|
+
expect( tuple0.each.to_a ).to eq( [["column1", "1"], ["column2", "a"]] )
|
135
|
+
expect( tuple1.each.to_a ).to eq( [["column1", "2"], ["column2", "b"]] )
|
136
|
+
expect( tuple2.each.to_a ).to eq( [["a", 1], ["b", true], ["b", "3"]] )
|
137
|
+
expect( tuple3.each.to_a ).to eq( [["a", 2], ["b", false], ["b", "4"]] )
|
138
|
+
expect{ tuple_empty.each }.to raise_error(TypeError)
|
139
|
+
end
|
140
|
+
|
141
|
+
it "can be used with block" do
|
142
|
+
a = []
|
143
|
+
tuple0.each do |*v|
|
144
|
+
a << v
|
145
|
+
end
|
146
|
+
expect( a ).to eq( [["column1", "1"], ["column2", "a"]] )
|
147
|
+
end
|
148
|
+
end
|
149
|
+
|
150
|
+
describe "each_value" do
|
151
|
+
it "can be used as an enumerator" do
|
152
|
+
expect( tuple0.each_value ).to be_kind_of(Enumerator)
|
153
|
+
expect( tuple0.each_value.to_a ).to eq( ["1", "a"] )
|
154
|
+
expect( tuple1.each_value.to_a ).to eq( ["2", "b"] )
|
155
|
+
expect( tuple2.each_value.to_a ).to eq( [1, true, "3"] )
|
156
|
+
expect( tuple3.each_value.to_a ).to eq( [2, false, "4"] )
|
157
|
+
expect{ tuple_empty.each_value }.to raise_error(TypeError)
|
158
|
+
end
|
159
|
+
|
160
|
+
it "can be used with block" do
|
161
|
+
a = []
|
162
|
+
tuple0.each_value do |v|
|
163
|
+
a << v
|
164
|
+
end
|
165
|
+
expect( a ).to eq( ["1", "a"] )
|
166
|
+
end
|
167
|
+
end
|
168
|
+
|
169
|
+
it "responds to values" do
|
170
|
+
expect( tuple0.values ).to eq( ["1", "a"] )
|
171
|
+
expect( tuple3.values ).to eq( [2, false, "4"] )
|
172
|
+
expect{ tuple_empty.values }.to raise_error(TypeError)
|
173
|
+
end
|
174
|
+
|
175
|
+
it "responds to key?" do
|
176
|
+
expect( tuple1.key?("column1") ).to eq( true )
|
177
|
+
expect( tuple1.key?("other") ).to eq( false )
|
178
|
+
expect( tuple1.has_key?("column1") ).to eq( true )
|
179
|
+
expect( tuple1.has_key?("other") ).to eq( false )
|
180
|
+
end
|
181
|
+
|
182
|
+
it "responds to keys" do
|
183
|
+
expect( tuple0.keys ).to eq( ["column1", "column2"] )
|
184
|
+
expect( tuple2.keys ).to eq( ["a", "b", "b"] )
|
185
|
+
end
|
186
|
+
|
187
|
+
describe "each_key" do
|
188
|
+
it "can be used as an enumerator" do
|
189
|
+
expect( tuple0.each_key ).to be_kind_of(Enumerator)
|
190
|
+
expect( tuple0.each_key.to_a ).to eq( ["column1", "column2"] )
|
191
|
+
expect( tuple2.each_key.to_a ).to eq( ["a", "b", "b"] )
|
192
|
+
end
|
193
|
+
|
194
|
+
it "can be used with block" do
|
195
|
+
a = []
|
196
|
+
tuple0.each_key do |v|
|
197
|
+
a << v
|
198
|
+
end
|
199
|
+
expect( a ).to eq( ["column1", "column2"] )
|
200
|
+
end
|
201
|
+
end
|
202
|
+
|
203
|
+
it "responds to length" do
|
204
|
+
expect( tuple0.length ).to eq( 2 )
|
205
|
+
expect( tuple0.size ).to eq( 2 )
|
206
|
+
expect( tuple2.size ).to eq( 3 )
|
207
|
+
end
|
208
|
+
|
209
|
+
it "responds to index" do
|
210
|
+
expect( tuple0.index("column1") ).to eq( 0 )
|
211
|
+
expect( tuple0.index("column2") ).to eq( 1 )
|
212
|
+
expect( tuple0.index("x") ).to eq( nil )
|
213
|
+
expect( tuple2.index("a") ).to eq( 0 )
|
214
|
+
expect( tuple2.index("b") ).to eq( 2 )
|
215
|
+
end
|
216
|
+
|
217
|
+
it "can be used as Enumerable" do
|
218
|
+
expect( tuple0.to_a ).to eq( [["column1", "1"], ["column2", "a"]] )
|
219
|
+
expect( tuple1.to_a ).to eq( [["column1", "2"], ["column2", "b"]] )
|
220
|
+
expect( tuple2.to_a ).to eq( [["a", 1], ["b", true], ["b", "3"]] )
|
221
|
+
expect( tuple3.to_a ).to eq( [["a", 2], ["b", false], ["b", "4"]] )
|
222
|
+
end
|
223
|
+
|
224
|
+
it "can be marshaled" do
|
225
|
+
[tuple0, tuple1, tuple2, tuple3].each do |t1|
|
226
|
+
str = Marshal.dump(t1)
|
227
|
+
t2 = Marshal.load(str)
|
228
|
+
|
229
|
+
expect( t2 ).to be_kind_of(t1.class)
|
230
|
+
expect( t2 ).not_to equal(t1)
|
231
|
+
expect( t2.to_a ).to eq(t1.to_a)
|
232
|
+
end
|
233
|
+
end
|
234
|
+
|
235
|
+
it "passes instance variables when marshaled" do
|
236
|
+
t1 = tuple0
|
237
|
+
t1.instance_variable_set("@a", 4711)
|
238
|
+
str = Marshal.dump(t1)
|
239
|
+
t2 = Marshal.load(str)
|
240
|
+
|
241
|
+
expect( t2.instance_variable_get("@a") ).to eq( 4711 )
|
242
|
+
end
|
243
|
+
|
244
|
+
it "can't be marshaled when empty" do
|
245
|
+
expect{ Marshal.dump(tuple_empty) }.to raise_error(TypeError)
|
246
|
+
end
|
247
|
+
|
248
|
+
it "should give account about memory usage" do
|
249
|
+
expect( ObjectSpace.memsize_of(tuple0) ).to be > 40
|
250
|
+
expect( ObjectSpace.memsize_of(tuple_empty) ).to be > 0
|
251
|
+
end
|
252
|
+
|
253
|
+
it "should override #inspect" do
|
254
|
+
expect( tuple1.inspect ).to eq('#<PG::Tuple column1: "2", column2: "b">')
|
255
|
+
expect( tuple2.inspect ).to eq('#<PG::Tuple a: 1, b: true, b: "3">')
|
256
|
+
expect{ tuple_empty.inspect }.to raise_error(TypeError)
|
257
|
+
end
|
258
|
+
|
259
|
+
context "with cleared result" do
|
260
|
+
it "should raise an error when non-materialized fields are used" do
|
261
|
+
r = result2x2
|
262
|
+
t = r.tuple(0)
|
263
|
+
t[0] # materialize first field only
|
264
|
+
r.clear
|
265
|
+
|
266
|
+
# second column should fail
|
267
|
+
expect{ t[1] }.to raise_error(PG::Error)
|
268
|
+
expect{ t.fetch(1) }.to raise_error(PG::Error)
|
269
|
+
expect{ t.fetch("column2") }.to raise_error(PG::Error)
|
270
|
+
|
271
|
+
# first column should succeed
|
272
|
+
expect( t[0] ).to eq( "1" )
|
273
|
+
expect( t.fetch(0) ).to eq( "1" )
|
274
|
+
expect( t.fetch("column1") ).to eq( "1" )
|
275
|
+
|
276
|
+
# should fail due to the second column
|
277
|
+
expect{ t.values }.to raise_error(PG::Error)
|
278
|
+
end
|
279
|
+
end
|
280
|
+
end
|
@@ -0,0 +1,138 @@
|
|
1
|
+
# -*- rspec -*-
|
2
|
+
# encoding: utf-8
|
3
|
+
|
4
|
+
require_relative '../helpers'
|
5
|
+
|
6
|
+
require 'pg'
|
7
|
+
|
8
|
+
|
9
|
+
describe PG::TypeMapByClass do
|
10
|
+
|
11
|
+
let!(:textenc_int){ PG::TextEncoder::Integer.new name: 'INT4', oid: 23 }
|
12
|
+
let!(:textenc_float){ PG::TextEncoder::Float.new name: 'FLOAT8', oid: 701 }
|
13
|
+
let!(:textenc_string){ PG::TextEncoder::String.new name: 'TEXT', oid: 25 }
|
14
|
+
let!(:binaryenc_int){ PG::BinaryEncoder::Int8.new name: 'INT8', oid: 20, format: 1 }
|
15
|
+
let!(:pass_through_type) do
|
16
|
+
type = Class.new(PG::SimpleEncoder) do
|
17
|
+
def encode(*v)
|
18
|
+
v.inspect
|
19
|
+
end
|
20
|
+
end.new
|
21
|
+
type.oid = 25
|
22
|
+
type.format = 0
|
23
|
+
type.name = 'pass_through'
|
24
|
+
type
|
25
|
+
end
|
26
|
+
|
27
|
+
let!(:tm) do
|
28
|
+
tm = PG::TypeMapByClass.new
|
29
|
+
tm[Integer] = binaryenc_int
|
30
|
+
tm[Float] = textenc_float
|
31
|
+
tm[Symbol] = pass_through_type
|
32
|
+
tm
|
33
|
+
end
|
34
|
+
|
35
|
+
let!(:raise_class) do
|
36
|
+
Class.new
|
37
|
+
end
|
38
|
+
|
39
|
+
let!(:derived_tm) do
|
40
|
+
tm = Class.new(PG::TypeMapByClass) do
|
41
|
+
def array_type_map_for(value)
|
42
|
+
PG::TextEncoder::Array.new name: '_INT4', oid: 1007, elements_type: PG::TextEncoder::Integer.new
|
43
|
+
end
|
44
|
+
end.new
|
45
|
+
tm[Integer] = proc{|value| textenc_int }
|
46
|
+
tm[raise_class] = proc{|value| /invalid/ }
|
47
|
+
tm[Array] = :array_type_map_for
|
48
|
+
tm
|
49
|
+
end
|
50
|
+
|
51
|
+
it "should retrieve all conversions" do
|
52
|
+
expect( tm.coders ).to eq( {
|
53
|
+
Integer => binaryenc_int,
|
54
|
+
Float => textenc_float,
|
55
|
+
Symbol => pass_through_type,
|
56
|
+
} )
|
57
|
+
end
|
58
|
+
|
59
|
+
it "should retrieve particular conversions" do
|
60
|
+
expect( tm[Integer] ).to eq(binaryenc_int)
|
61
|
+
expect( tm[Float] ).to eq(textenc_float)
|
62
|
+
expect( tm[Range] ).to be_nil
|
63
|
+
expect( derived_tm[raise_class] ).to be_kind_of(Proc)
|
64
|
+
expect( derived_tm[Array] ).to eq(:array_type_map_for)
|
65
|
+
end
|
66
|
+
|
67
|
+
it "should allow deletion of coders" do
|
68
|
+
tm[Integer] = nil
|
69
|
+
expect( tm[Integer] ).to be_nil
|
70
|
+
expect( tm.coders ).to eq( {
|
71
|
+
Float => textenc_float,
|
72
|
+
Symbol => pass_through_type,
|
73
|
+
} )
|
74
|
+
end
|
75
|
+
|
76
|
+
it "forwards query param conversions to the #default_type_map" do
|
77
|
+
tm1 = PG::TypeMapByColumn.new( [textenc_int, nil, nil] )
|
78
|
+
|
79
|
+
tm2 = PG::TypeMapByClass.new
|
80
|
+
tm2[Integer] = PG::TextEncoder::Integer.new name: 'INT2', oid: 21
|
81
|
+
tm2.default_type_map = tm1
|
82
|
+
|
83
|
+
res = @conn.exec_params( "SELECT $1, $2, $3::TEXT", ['1', 2, 3], 0, tm2 )
|
84
|
+
|
85
|
+
expect( res.ftype(0) ).to eq( 23 ) # tm1
|
86
|
+
expect( res.ftype(1) ).to eq( 21 ) # tm2
|
87
|
+
expect( res.getvalue(0,2) ).to eq( "3" ) # TypeMapAllStrings
|
88
|
+
end
|
89
|
+
|
90
|
+
#
|
91
|
+
# Decoding Examples
|
92
|
+
#
|
93
|
+
|
94
|
+
it "should raise an error when used for results" do
|
95
|
+
res = @conn.exec_params( "SELECT 1", [], 1 )
|
96
|
+
expect{ res.type_map = tm }.to raise_error(NotImplementedError, /not suitable to map result values/)
|
97
|
+
end
|
98
|
+
|
99
|
+
#
|
100
|
+
# Encoding Examples
|
101
|
+
#
|
102
|
+
|
103
|
+
it "should allow mixed type conversions" do
|
104
|
+
res = @conn.exec_params( "SELECT $1, $2, $3", [5, 1.23, :TestSymbol], 0, tm )
|
105
|
+
expect( res.values ).to eq([['5', '1.23', "[:TestSymbol, #{@conn.internal_encoding.inspect}]"]])
|
106
|
+
expect( res.ftype(0) ).to eq(20)
|
107
|
+
end
|
108
|
+
|
109
|
+
it "should expire the cache after changes to the coders" do
|
110
|
+
res = @conn.exec_params( "SELECT $1", [5], 0, tm )
|
111
|
+
expect( res.ftype(0) ).to eq(20)
|
112
|
+
|
113
|
+
tm[Integer] = textenc_int
|
114
|
+
|
115
|
+
res = @conn.exec_params( "SELECT $1", [5], 0, tm )
|
116
|
+
expect( res.ftype(0) ).to eq(23)
|
117
|
+
end
|
118
|
+
|
119
|
+
it "should allow mixed type conversions with derived type map" do
|
120
|
+
res = @conn.exec_params( "SELECT $1, $2", [6, [7]], 0, derived_tm )
|
121
|
+
expect( res.values ).to eq([['6', '{7}']])
|
122
|
+
expect( res.ftype(0) ).to eq(23)
|
123
|
+
expect( res.ftype(1) ).to eq(1007)
|
124
|
+
end
|
125
|
+
|
126
|
+
it "should raise TypeError with derived type map" do
|
127
|
+
expect{
|
128
|
+
@conn.exec_params( "SELECT $1", [raise_class.new], 0, derived_tm )
|
129
|
+
}.to raise_error(TypeError, /invalid type Regexp/)
|
130
|
+
end
|
131
|
+
|
132
|
+
it "should raise error on invalid coder object" do
|
133
|
+
tm[TrueClass] = "dummy"
|
134
|
+
expect{
|
135
|
+
res = @conn.exec_params( "SELECT $1", [true], 0, tm )
|
136
|
+
}.to raise_error(NoMethodError, /undefined method.*call/)
|
137
|
+
end
|
138
|
+
end
|
@@ -0,0 +1,222 @@
|
|
1
|
+
# -*- rspec -*-
|
2
|
+
# encoding: utf-8
|
3
|
+
|
4
|
+
require_relative '../helpers'
|
5
|
+
|
6
|
+
require 'pg'
|
7
|
+
|
8
|
+
|
9
|
+
describe PG::TypeMapByColumn do
|
10
|
+
|
11
|
+
let!(:textenc_int){ PG::TextEncoder::Integer.new name: 'INT4', oid: 23 }
|
12
|
+
let!(:textdec_int){ PG::TextDecoder::Integer.new name: 'INT4', oid: 23 }
|
13
|
+
let!(:textenc_float){ PG::TextEncoder::Float.new name: 'FLOAT4', oid: 700 }
|
14
|
+
let!(:textdec_float){ PG::TextDecoder::Float.new name: 'FLOAT4', oid: 700 }
|
15
|
+
let!(:textenc_string){ PG::TextEncoder::String.new name: 'TEXT', oid: 25 }
|
16
|
+
let!(:textdec_string){ PG::TextDecoder::String.new name: 'TEXT', oid: 25 }
|
17
|
+
let!(:textdec_bytea){ PG::TextDecoder::Bytea.new name: 'BYTEA', oid: 17 }
|
18
|
+
let!(:binaryenc_bytea){ PG::BinaryEncoder::Bytea.new name: 'BYTEA', oid: 17, format: 1 }
|
19
|
+
let!(:binarydec_bytea){ PG::BinaryDecoder::Bytea.new name: 'BYTEA', oid: 17, format: 1 }
|
20
|
+
let!(:pass_through_type) do
|
21
|
+
type = Class.new(PG::SimpleDecoder) do
|
22
|
+
def decode(*v)
|
23
|
+
v
|
24
|
+
end
|
25
|
+
end.new
|
26
|
+
type.oid = 123456
|
27
|
+
type.format = 1
|
28
|
+
type.name = 'pass_through'
|
29
|
+
type
|
30
|
+
end
|
31
|
+
|
32
|
+
it "should retrieve it's conversions" do
|
33
|
+
cm = PG::TypeMapByColumn.new( [textdec_int, textenc_string, textdec_float, pass_through_type, nil] )
|
34
|
+
expect( cm.coders ).to eq( [
|
35
|
+
textdec_int,
|
36
|
+
textenc_string,
|
37
|
+
textdec_float,
|
38
|
+
pass_through_type,
|
39
|
+
nil
|
40
|
+
] )
|
41
|
+
expect( cm.inspect ).to eq( "#<PG::TypeMapByColumn INT4:0 TEXT:0 FLOAT4:0 pass_through:1 nil>" )
|
42
|
+
end
|
43
|
+
|
44
|
+
it "should retrieve it's oids" do
|
45
|
+
cm = PG::TypeMapByColumn.new( [textdec_int, textdec_string, textdec_float, pass_through_type, nil] )
|
46
|
+
expect( cm.oids ).to eq( [23, 25, 700, 123456, nil] )
|
47
|
+
end
|
48
|
+
|
49
|
+
it "should gracefully handle not initialized state" do
|
50
|
+
# PG::TypeMapByColumn is not initialized in allocate function, like other
|
51
|
+
# type maps, but in #initialize. So it might be not called by derived classes.
|
52
|
+
|
53
|
+
not_init = Class.new(PG::TypeMapByColumn) do
|
54
|
+
def initialize
|
55
|
+
# no super call
|
56
|
+
end
|
57
|
+
end.new
|
58
|
+
|
59
|
+
expect{ @conn.exec_params( "SELECT $1", [ 0 ], 0, not_init ) }.to raise_error(NotImplementedError)
|
60
|
+
|
61
|
+
res = @conn.exec( "SELECT 1" )
|
62
|
+
expect{ res.type_map = not_init }.to raise_error(NotImplementedError)
|
63
|
+
|
64
|
+
@conn.copy_data("COPY (SELECT 1) TO STDOUT") do
|
65
|
+
decoder = PG::TextDecoder::CopyRow.new(type_map: not_init)
|
66
|
+
expect{ @conn.get_copy_data(false, decoder) }.to raise_error(NotImplementedError)
|
67
|
+
@conn.get_copy_data
|
68
|
+
end
|
69
|
+
end
|
70
|
+
|
71
|
+
|
72
|
+
#
|
73
|
+
# Encoding Examples
|
74
|
+
#
|
75
|
+
|
76
|
+
it "should encode integer params" do
|
77
|
+
col_map = PG::TypeMapByColumn.new( [textenc_int]*3 )
|
78
|
+
res = @conn.exec_params( "SELECT $1, $2, $3", [ 0, nil, "-999" ], 0, col_map )
|
79
|
+
expect( res.values ).to eq( [
|
80
|
+
[ "0", nil, "-999" ],
|
81
|
+
] )
|
82
|
+
end
|
83
|
+
|
84
|
+
it "should encode bytea params" do
|
85
|
+
data = "'\u001F\\"
|
86
|
+
col_map = PG::TypeMapByColumn.new( [binaryenc_bytea]*2 )
|
87
|
+
res = @conn.exec_params( "SELECT $1, $2", [ data, nil ], 0, col_map )
|
88
|
+
res.type_map = PG::TypeMapByColumn.new( [textdec_bytea]*2 )
|
89
|
+
expect( res.values ).to eq( [
|
90
|
+
[ data, nil ],
|
91
|
+
] )
|
92
|
+
end
|
93
|
+
|
94
|
+
|
95
|
+
it "should allow hash form parameters for default encoder" do
|
96
|
+
col_map = PG::TypeMapByColumn.new( [nil, nil] )
|
97
|
+
hash_param_bin = { value: ["00ff"].pack("H*"), type: 17, format: 1 }
|
98
|
+
hash_param_nil = { value: nil, type: 17, format: 1 }
|
99
|
+
res = @conn.exec_params( "SELECT $1, $2",
|
100
|
+
[ hash_param_bin, hash_param_nil ], 0, col_map )
|
101
|
+
expect( res.values ).to eq( [["\\x00ff", nil]] )
|
102
|
+
expect( result_typenames(res) ).to eq( ['bytea', 'bytea'] )
|
103
|
+
end
|
104
|
+
|
105
|
+
it "should convert hash form parameters to string when using string encoders" do
|
106
|
+
col_map = PG::TypeMapByColumn.new( [textenc_string, textenc_string] )
|
107
|
+
hash_param_bin = { value: ["00ff"].pack("H*"), type: 17, format: 1 }
|
108
|
+
hash_param_nil = { value: nil, type: 17, format: 1 }
|
109
|
+
res = @conn.exec_params( "SELECT $1::text, $2::text",
|
110
|
+
[ hash_param_bin, hash_param_nil ], 0, col_map )
|
111
|
+
expect( res.values ).to eq( [["{:value=>\"\\x00\\xFF\", :type=>17, :format=>1}", "{:value=>nil, :type=>17, :format=>1}"]] )
|
112
|
+
end
|
113
|
+
|
114
|
+
it "shouldn't allow param mappings with different number of fields" do
|
115
|
+
expect{
|
116
|
+
@conn.exec_params( "SELECT $1", [ 123 ], 0, PG::TypeMapByColumn.new([]) )
|
117
|
+
}.to raise_error(ArgumentError, /mapped columns/)
|
118
|
+
end
|
119
|
+
|
120
|
+
it "should verify the default type map for query params as well" do
|
121
|
+
tm1 = PG::TypeMapByColumn.new([])
|
122
|
+
expect{
|
123
|
+
@conn.exec_params( "SELECT $1", [ 123 ], 0, PG::TypeMapByColumn.new([nil]).with_default_type_map(tm1) )
|
124
|
+
}.to raise_error(ArgumentError, /mapped columns/)
|
125
|
+
end
|
126
|
+
|
127
|
+
it "forwards query param conversions to the #default_type_map" do
|
128
|
+
tm1 = PG::TypeMapByClass.new
|
129
|
+
tm1[Integer] = PG::TextEncoder::Integer.new name: 'INT2', oid: 21
|
130
|
+
|
131
|
+
tm2 = PG::TypeMapByColumn.new( [textenc_int, nil, nil] ).with_default_type_map( tm1 )
|
132
|
+
res = @conn.exec_params( "SELECT $1, $2, $3::TEXT", [1, 2, :abc], 0, tm2 )
|
133
|
+
|
134
|
+
expect( res.ftype(0) ).to eq( 23 ) # tm2
|
135
|
+
expect( res.ftype(1) ).to eq( 21 ) # tm1
|
136
|
+
expect( res.getvalue(0,2) ).to eq( "abc" ) # TypeMapAllStrings
|
137
|
+
end
|
138
|
+
|
139
|
+
#
|
140
|
+
# Decoding Examples
|
141
|
+
#
|
142
|
+
|
143
|
+
class Exception_in_decode < PG::SimpleDecoder
|
144
|
+
def decode(res, tuple, field)
|
145
|
+
raise "no type decoder defined for tuple #{tuple} field #{field}"
|
146
|
+
end
|
147
|
+
end
|
148
|
+
|
149
|
+
it "should raise an error from decode method of type converter" do
|
150
|
+
res = @conn.exec( "SELECT now()" )
|
151
|
+
types = Array.new( res.nfields, Exception_in_decode.new )
|
152
|
+
res.type_map = PG::TypeMapByColumn.new( types )
|
153
|
+
expect{ res.values }.to raise_error(/no type decoder defined/)
|
154
|
+
end
|
155
|
+
|
156
|
+
it "should raise an error for invalid params" do
|
157
|
+
expect{ PG::TypeMapByColumn.new( :WrongType ) }.to raise_error(TypeError, /wrong argument type/)
|
158
|
+
expect{ PG::TypeMapByColumn.new( [123] ) }.to raise_error(ArgumentError, /invalid/)
|
159
|
+
end
|
160
|
+
|
161
|
+
it "shouldn't allow result mappings with different number of fields" do
|
162
|
+
res = @conn.exec( "SELECT 1" )
|
163
|
+
expect{ res.type_map = PG::TypeMapByColumn.new([]) }.to raise_error(ArgumentError, /mapped columns/)
|
164
|
+
end
|
165
|
+
|
166
|
+
it "should verify the default type map for result values as well" do
|
167
|
+
res = @conn.exec( "SELECT 1" )
|
168
|
+
tm1 = PG::TypeMapByColumn.new([])
|
169
|
+
expect{
|
170
|
+
res.type_map = PG::TypeMapByColumn.new([nil]).with_default_type_map(tm1)
|
171
|
+
}.to raise_error(ArgumentError, /mapped columns/)
|
172
|
+
end
|
173
|
+
|
174
|
+
it "forwards result value conversions to a TypeMapByOid as #default_type_map" do
|
175
|
+
# One run with implicit built TypeMapByColumn and another with online lookup
|
176
|
+
[0, 10].each do |max_rows|
|
177
|
+
tm1 = PG::TypeMapByOid.new
|
178
|
+
tm1.add_coder PG::TextDecoder::Integer.new name: 'INT2', oid: 21
|
179
|
+
tm1.max_rows_for_online_lookup = max_rows
|
180
|
+
|
181
|
+
tm2 = PG::TypeMapByColumn.new( [textdec_int, nil, nil] ).with_default_type_map( tm1 )
|
182
|
+
res = @conn.exec( "SELECT '1'::INT4, '2'::INT2, '3'::INT8" ).map_types!( tm2 )
|
183
|
+
|
184
|
+
expect( res.getvalue(0,0) ).to eq( 1 ) # tm2
|
185
|
+
expect( res.getvalue(0,1) ).to eq( 2 ) # tm1
|
186
|
+
expect( res.getvalue(0,2) ).to eq( "3" ) # TypeMapAllStrings
|
187
|
+
end
|
188
|
+
end
|
189
|
+
|
190
|
+
it "forwards get_copy_data conversions to another TypeMapByColumn as #default_type_map" do
|
191
|
+
tm1 = PG::TypeMapByColumn.new( [textdec_int, nil, nil] )
|
192
|
+
tm2 = PG::TypeMapByColumn.new( [nil, textdec_int, nil] ).with_default_type_map( tm1 )
|
193
|
+
decoder = PG::TextDecoder::CopyRow.new(type_map: tm2)
|
194
|
+
@conn.copy_data("COPY (SELECT 1, 2, 3) TO STDOUT", decoder) do
|
195
|
+
expect( @conn.get_copy_data ).to eq( [1, 2, '3'] )
|
196
|
+
@conn.get_copy_data
|
197
|
+
end
|
198
|
+
end
|
199
|
+
|
200
|
+
it "will deny copy queries with different column count" do
|
201
|
+
[[2, 2], [2, 3], [3, 2]].each do |cols1, cols2|
|
202
|
+
tm1 = PG::TypeMapByColumn.new( [textdec_int, nil, nil][0, cols1] )
|
203
|
+
tm2 = PG::TypeMapByColumn.new( [nil, textdec_int, nil][0, cols2] ).with_default_type_map( tm1 )
|
204
|
+
decoder = PG::TextDecoder::CopyRow.new(type_map: tm2)
|
205
|
+
@conn.copy_data("COPY (SELECT 1, 2, 3) TO STDOUT", decoder) do
|
206
|
+
expect{ @conn.get_copy_data }.to raise_error(ArgumentError, /number of copy fields/)
|
207
|
+
@conn.get_copy_data
|
208
|
+
end
|
209
|
+
end
|
210
|
+
end
|
211
|
+
|
212
|
+
#
|
213
|
+
# Decoding Examples text format
|
214
|
+
#
|
215
|
+
|
216
|
+
it "should allow mixed type conversions" do
|
217
|
+
res = @conn.exec( "SELECT 1, 'a', 2.0::FLOAT, '2013-06-30'::DATE, 3" )
|
218
|
+
res.type_map = PG::TypeMapByColumn.new( [textdec_int, textdec_string, textdec_float, pass_through_type, nil] )
|
219
|
+
expect( res.values ).to eq( [[1, 'a', 2.0, ['2013-06-30', 0, 3], '3' ]] )
|
220
|
+
end
|
221
|
+
|
222
|
+
end
|