pg 0.17.1 → 0.18.4

data/spec/pg/type_map_in_ruby_spec.rb

@@ -0,0 +1,164 @@
+#!/usr/bin/env rspec
+# encoding: utf-8
+
+require_relative '../helpers'
+
+require 'pg'
+
+
+describe PG::TypeMapInRuby do
+
+	context "result values" do
+		it "should be usable non-derived" do
+			tm = PG::TypeMapInRuby.new
+			res = @conn.exec("select 5").map_types!(tm)
+			expect( res.getvalue(0,0) ).to eq( "5" )
+		end
+
+		it "should call derived result mapping methods" do
+			tm = Class.new(PG::TypeMapInRuby) do
+				attr_reader :fit_to_result_args
+
+				def fit_to_result(*args)
+					@fit_to_result_args = args
+					self
+				end
+
+				def typecast_result_value(*args)
+					[args, super]
+				end
+			end.new
+
+			res = @conn.exec("select 5,6").map_types!(tm)
+			expect( res.getvalue(0,1) ).to eq( [[res, 0, 1], "6"] )
+			expect( tm.fit_to_result_args ).to eq( [res] )
+		end
+
+		it "should accept only a type map object from fit_to_result" do
+			tm = Class.new(PG::TypeMapInRuby) do
+				def fit_to_result(*args)
+					:invalid
+				end
+			end.new
+
+			res = @conn.exec("select 5,6")
+			expect{ res.map_types!(tm) }.to raise_error(TypeError, /kind of PG::TypeMap/)
+		end
+	end
+
+	context "query bind params" do
+		it "should be usable non-derived" do
+			tm = PG::TypeMapInRuby.new
+			res = @conn.exec_params("select $1::int, $2::text", [5, 6], 0, tm)
+			expect( res.values ).to eq( [["5", "6"]] )
+		end
+
+		it "should call derived param mapping methods" do
+			tm = Class.new(PG::TypeMapInRuby) do
+				attr_reader :fit_to_query_args
+				attr_reader :typecast_query_param_args
+
+				def fit_to_query(params)
+					@fit_to_query_args = params
+					@typecast_query_param_args = []
+					self
+				end
+
+				def typecast_query_param(*args)
+					@typecast_query_param_args << [args, super]
+					PG::TextEncoder::Integer.new name: 'INT4', oid: 23
+				end
+			end.new
+
+			res = @conn.exec_params("select $1, $2", [5, 6], 0, tm)
+			expect( res.ftype(0) ).to eq( 23 )
+			expect( tm.fit_to_query_args ).to eq( [5, 6] )
+			expect( tm.typecast_query_param_args ).to eq( [[[5, 0], nil], [[6, 1], nil]] )
+		end
+	end
+
+	context "put_copy_data" do
+		it "should be usable non-derived" do
+			tm = PG::TypeMapInRuby.new
+			ce = PG::TextEncoder::CopyRow.new type_map: tm
+			res = ce.encode([5, 6])
+			expect( res ).to eq( "5\t6\n" )
+		end
+
+		it "should call derived data mapping methods" do
+			tm = Class.new(PG::TypeMapInRuby) do
+				attr_reader :fit_to_query_args
+				attr_reader :typecast_query_param_args
+
+				def fit_to_query(params)
+					@fit_to_query_args = params
+					@typecast_query_param_args = []
+					self
+				end
+
+				def typecast_query_param(*args)
+					@typecast_query_param_args << [args, super]
+					PG::TextEncoder::Integer.new name: 'INT4', oid: 23
+				end
+			end.new
+
+			ce = PG::TextEncoder::CopyRow.new type_map: tm
+			res = ce.encode([5, 6])
+			expect( res ).to eq( "5\t6\n" )
+			expect( tm.fit_to_query_args ).to eq( [5, 6] )
+			expect( tm.typecast_query_param_args ).to eq( [[[5, 0], nil], [[6, 1], nil]] )
+		end
+
+		it "shouldn't accept invalid return from typecast_query_param" do
+			tm = Class.new(PG::TypeMapInRuby) do
+				def typecast_query_param(*args)
+					:invalid
+				end
+			end.new
+
+			ce = PG::TextEncoder::CopyRow.new type_map: tm
+			expect{ ce.encode([5, 6]) }.to raise_error(TypeError, /nil or kind of PG::Coder/)
+		end
+	end
+
+	context "get_copy_data" do
+		it "should be usable non-derived" do
+			tm = PG::TypeMapInRuby.new
+			ce = PG::TextDecoder::CopyRow.new type_map: tm
+			res = ce.decode("5\t6\n")
+			expect( res ).to eq( ["5", "6"] )
+		end
+
+		it "should call derived data mapping methods" do
+			tm = Class.new(PG::TypeMapInRuby) do
+				attr_reader :fit_to_copy_get_args
+
+				def fit_to_copy_get(*args)
+					@fit_to_copy_get_args = args
+					0
+				end
+
+				def typecast_copy_get(field_str, fieldno, format, enc)
+					[field_str, fieldno, format, enc, super]
+				end
+			end.new
+
+			ce = PG::TextDecoder::CopyRow.new type_map: tm
+			res = ce.decode("5\t6\n")
+			expect( tm.fit_to_copy_get_args ).to eq( [] )
+			expect( res ).to eq( [["5", 0, 0, Encoding::UTF_8, "5"], ["6", 1, 0, Encoding::UTF_8, "6"]] )
+		end
+
+		it "shouldn't accept invalid return from fit_to_copy_get" do
+			tm = Class.new(PG::TypeMapInRuby) do
+				def fit_to_copy_get
+					:invalid
+				end
+			end.new
+
+			ce = PG::TextDecoder::CopyRow.new type_map: tm
+			expect{ ce.decode("5\t6\n") }.to raise_error(TypeError, /kind of Integer/)
+		end
+	end
+
+end

data/spec/pg/type_map_spec.rb

@@ -0,0 +1,22 @@
+#!/usr/bin/env rspec
+# encoding: utf-8
+
+require_relative '../helpers'
+
+require 'pg'
+
+
+describe PG::TypeMap do
+	let!(:tm){ PG::TypeMap.new }
+
+	it "should raise an error when used for param type casts" do
+		expect{
+			@conn.exec_params( "SELECT $1", [5], 0, tm )
+		}.to raise_error(NotImplementedError, /not suitable to map query params/)
+	end
+
+	it "should raise an error when used for result type casts" do
+		res = @conn.exec( "SELECT 1" )
+		expect{ res.map_types!(tm) }.to raise_error(NotImplementedError, /not suitable to map result values/)
+	end
+end