pg 0.18.1 → 1.5.6

data/spec/helpers.rb

@@ -1,355 +0,0 @@
-#!/usr/bin/env ruby
-
-require 'pathname'
-require 'rspec'
-require 'shellwords'
-require 'pg'
-
-TEST_DIRECTORY = Pathname.getwd + "tmp_test_specs"
-
-module PG::TestingHelpers
-
-	### Automatically set up the database when it's used, and wrap a transaction around
-	### examples that don't disable it.
-	def self::included( mod )
-		super
-
-		if mod.respond_to?( :around )
-
-			mod.before( :all ) { @conn = setup_testing_db(described_class ? described_class.name : mod.description) }
-
-			mod.around( :each ) do |example|
-				begin
-					@conn.exec( 'BEGIN' ) unless example.metadata[:without_transaction]
-					if PG.respond_to?( :library_version )
-						desc = example.source_location.join(':')
-						@conn.exec_params %Q{SET application_name TO '%s'} %
-							[@conn.escape_string(desc.slice(-60))]
-					end
-					example.run
-				ensure
-					@conn.exec( 'ROLLBACK' ) unless example.metadata[:without_transaction]
-				end
-			end
-
-			mod.after( :all ) { teardown_testing_db(@conn) }
-		end
-
-	end
-
-
-	#
-	# Examples
-	#
-
-	# Set some ANSI escape code constants (Shamelessly stolen from Perl's
-	# Term::ANSIColor by Russ Allbery <rra@stanford.edu> and Zenin <zenin@best.com>
-	ANSI_ATTRIBUTES = {
-		'clear'      => 0,
-		'reset'      => 0,
-		'bold'       => 1,
-		'dark'       => 2,
-		'underline'  => 4,
-		'underscore' => 4,
-		'blink'      => 5,
-		'reverse'    => 7,
-		'concealed'  => 8,
-
-		'black'      => 30,   'on_black'   => 40,
-		'red'        => 31,   'on_red'     => 41,
-		'green'      => 32,   'on_green'   => 42,
-		'yellow'     => 33,   'on_yellow'  => 43,
-		'blue'       => 34,   'on_blue'    => 44,
-		'magenta'    => 35,   'on_magenta' => 45,
-		'cyan'       => 36,   'on_cyan'    => 46,
-		'white'      => 37,   'on_white'   => 47
-	}
-
-
-	###############
-	module_function
-	###############
-
-	### Create a string that contains the ANSI codes specified and return it
-	def ansi_code( *attributes )
-		attributes.flatten!
-		attributes.collect! {|at| at.to_s }
-
-		return '' unless /(?:vt10[03]|xterm(?:-color)?|linux|screen)/i =~ ENV['TERM']
-		attributes = ANSI_ATTRIBUTES.values_at( *attributes ).compact.join(';')
-
-		# $stderr.puts "  attr is: %p" % [attributes]
-		if attributes.empty?
-			return ''
-		else
-			return "\e[%sm" % attributes
-		end
-	end
-
-
-	### Colorize the given +string+ with the specified +attributes+ and return it, handling
-	### line-endings, color reset, etc.
-	def colorize( *args )
-		string = ''
-
-		if block_given?
-			string = yield
-		else
-			string = args.shift
-		end
-
-		ending = string[/(\s)$/] || ''
-		string = string.rstrip
-
-		return ansi_code( args.flatten ) + string + ansi_code( 'reset' ) + ending
-	end
-
-
-	### Output a message with highlighting.
-	def message( *msg )
-		$stderr.puts( colorize(:bold) { msg.flatten.join(' ') } )
-	end
-
-
-	### Output a logging message if $VERBOSE is true
-	def trace( *msg )
-		return unless $VERBOSE
-		output = colorize( msg.flatten.join(' '), 'yellow' )
-		$stderr.puts( output )
-	end
-
-
-	### Return the specified args as a string, quoting any that have a space.
-	def quotelist( *args )
-		return args.flatten.collect {|part| part.to_s =~ /\s/ ? part.to_s.inspect : part.to_s }
-	end
-
-
-	### Run the specified command +cmd+ with system(), failing if the execution
-	### fails.
-	def run( *cmd )
-		cmd.flatten!
-
-		if cmd.length > 1
-			trace( quotelist(*cmd) )
-		else
-			trace( cmd )
-		end
-
-		system( *cmd )
-		raise "Command failed: [%s]" % [cmd.join(' ')] unless $?.success?
-	end
-
-
-	### Run the specified command +cmd+ after redirecting stdout and stderr to the specified
-	### +logpath+, failing if the execution fails.
-	def log_and_run( logpath, *cmd )
-		cmd.flatten!
-
-		if cmd.length > 1
-			trace( quotelist(*cmd) )
-		else
-			trace( cmd )
-		end
-
-		# Eliminate the noise of creating/tearing down the database by
-		# redirecting STDERR/STDOUT to a logfile
-		logfh = File.open( logpath, File::WRONLY|File::CREAT|File::APPEND )
-		system( *cmd, [STDOUT, STDERR] => logfh )
-
-		raise "Command failed: [%s]" % [cmd.join(' ')] unless $?.success?
-	end
-
-
-	### Check the current directory for directories that look like they're
-	### testing directories from previous tests, and tell any postgres instances
-	### running in them to shut down.
-	def stop_existing_postmasters
-		# tmp_test_0.22329534700318
-		pat = Pathname.getwd + 'tmp_test_*'
-		Pathname.glob( pat.to_s ).each do |testdir|
-			datadir = testdir + 'data'
-			pidfile = datadir + 'postmaster.pid'
-			if pidfile.exist? && pid = pidfile.read.chomp.to_i
-				$stderr.puts "pidfile (%p) exists: %d" % [ pidfile, pid ]
-				begin
-					Process.kill( 0, pid )
-				rescue Errno::ESRCH
-					$stderr.puts "No postmaster running for %s" % [ datadir ]
-					# Process isn't alive, so don't try to stop it
-				else
-					$stderr.puts "Stopping lingering database at PID %d" % [ pid ]
-					run 'pg_ctl', '-D', datadir.to_s, '-m', 'fast', 'stop'
-				end
-			else
-				$stderr.puts "No pidfile (%p)" % [ pidfile ]
-			end
-		end
-	end
-
-
-	### Set up a PostgreSQL database instance for testing.
-	def setup_testing_db( description )
-		require 'pg'
-		stop_existing_postmasters()
-
-		puts "Setting up test database for #{description}"
-		@test_pgdata = TEST_DIRECTORY + 'data'
-		@test_pgdata.mkpath
-
-		@port = 54321
-		ENV['PGPORT'] = @port.to_s
-		ENV['PGHOST'] = 'localhost'
-		@conninfo = "host=localhost port=#{@port} dbname=test"
-
-		@logfile = TEST_DIRECTORY + 'setup.log'
-		trace "Command output logged to #{@logfile}"
-
-		begin
-			unless (@test_pgdata+"postgresql.conf").exist?
-				FileUtils.rm_rf( @test_pgdata, :verbose => $DEBUG )
-				$stderr.puts "Running initdb"
-				log_and_run @logfile, 'initdb', '-E', 'UTF8', '--no-locale', '-D', @test_pgdata.to_s
-			end
-
-			trace "Starting postgres"
-			log_and_run @logfile, 'pg_ctl', '-w', '-o', "-k #{TEST_DIRECTORY.to_s.dump}",
-				'-D', @test_pgdata.to_s, 'start'
-			sleep 2
-
-			$stderr.puts "Creating the test DB"
-			log_and_run @logfile, 'psql', '-e', '-c', 'DROP DATABASE IF EXISTS test', 'postgres'
-			log_and_run @logfile, 'createdb', '-e', 'test'
-
-		rescue => err
-			$stderr.puts "%p during test setup: %s" % [ err.class, err.message ]
-			$stderr.puts "See #{@logfile} for details."
-			$stderr.puts *err.backtrace if $DEBUG
-			fail
-		end
-
-		conn = PG.connect( @conninfo )
-		conn.set_notice_processor do |message|
-			$stderr.puts( description + ':' + message ) if $DEBUG
-		end
-
-		return conn
-	end
-
-
-	def teardown_testing_db( conn )
-		puts "Tearing down test database"
-
-		if conn
-			check_for_lingering_connections( conn )
-			conn.finish
-		end
-
-		log_and_run @logfile, 'pg_ctl', '-D', @test_pgdata.to_s, 'stop'
-	end
-
-
-	def check_for_lingering_connections( conn )
-		conn.exec( "SELECT * FROM pg_stat_activity" ) do |res|
-			conns = res.find_all {|row| row['pid'].to_i != conn.backend_pid }
-			unless conns.empty?
-				puts "Lingering connections remain:"
-				conns.each do |row|
-					puts "  [%d] {%s} %s -- %s" % row.values_at( 'pid', 'state', 'application_name', 'query' )
-				end
-			end
-		end
-	end
-
-
-	# Retrieve the names of the column types of a given result set.
-	def result_typenames(res)
-		@conn.exec( "SELECT " + res.nfields.times.map{|i| "format_type($#{i*2+1},$#{i*2+2})"}.join(","),
-				res.nfields.times.map{|i| [res.ftype(i), res.fmod(i)] }.flatten ).
-				values[0]
-	end
-
-
-	# A matcher for checking the status of a PG::Connection to ensure it's still
-	# usable.
-	class ConnStillUsableMatcher
-
-		def initialize
-			@conn = nil
-			@problem = nil
-		end
-
-		def matches?( conn )
-			@conn = conn
-			@problem = self.check_for_problems
-			return @problem.nil?
-		end
-
-		def check_for_problems
-			return "is finished" if @conn.finished?
-			return "has bad status" unless @conn.status == PG::CONNECTION_OK
-			return "has bad transaction status (%d)" % [ @conn.transaction_status ] unless
-				@conn.transaction_status.between?( PG::PQTRANS_IDLE, PG::PQTRANS_INTRANS )
-			return "is not usable." unless self.can_exec_query?
-			return nil
-		end
-
-		def can_exec_query?
-			@conn.send_query( "VALUES (1)" )
-			@conn.get_last_result.values == [["1"]]
-		end
-
-		def failure_message
-			return "expected %p to be usable, but it %s" % [ @conn, @problem ]
-		end
-
-		def failure_message_when_negated
-			"expected %p not to be usable, but it still is" % [ @conn ]
-		end
-
-	end
-
-
-	### Return a ConnStillUsableMatcher to be used like:
-	###
-	###    expect( pg_conn ).to still_be_usable
-	###
-	def still_be_usable
-		return ConnStillUsableMatcher.new
-	end
-
-end
-
-
-RSpec.configure do |config|
-	config.include( PG::TestingHelpers )
-
-	config.run_all_when_everything_filtered = true
-	config.filter_run :focus
-	config.order = 'random'
-	config.mock_with( :rspec ) do |mock|
-		mock.syntax = :expect
-	end
-
-	if RUBY_PLATFORM =~ /mingw|mswin/
-		config.filter_run_excluding :unix
-	else
-		config.filter_run_excluding :windows
-	end
-	config.filter_run_excluding :socket_io unless
-		PG::Connection.instance_methods.map( &:to_sym ).include?( :socket_io )
-
-	config.filter_run_excluding :postgresql_90 unless
-		PG::Connection.instance_methods.map( &:to_sym ).include?( :escape_literal )
-
-	if !PG.respond_to?( :library_version )
-		config.filter_run_excluding( :postgresql_91, :postgresql_92, :postgresql_93, :postgresql_94 )
-	elsif PG.library_version < 90200
-		config.filter_run_excluding( :postgresql_92, :postgresql_93, :postgresql_94 )
-	elsif PG.library_version < 90300
-		config.filter_run_excluding( :postgresql_93, :postgresql_94 )
-	elsif PG.library_version < 90400
-		config.filter_run_excluding( :postgresql_94 )
-	end
-end
-

data/spec/pg/basic_type_mapping_spec.rb

@@ -1,251 +0,0 @@
-#!/usr/bin/env rspec
-# encoding: utf-8
-
-require_relative '../helpers'
-
-require 'pg'
-
-describe 'Basic type mapping' do
-
-	describe PG::BasicTypeMapForQueries do
-		let!(:basic_type_mapping) do
-			PG::BasicTypeMapForQueries.new @conn
-		end
-
-		#
-		# Encoding Examples
-		#
-
-		it "should do basic param encoding", :ruby_19 do
-			res = @conn.exec_params( "SELECT $1::int8,$2::float,$3,$4::TEXT",
-				[1, 2.1, true, "b"], nil, basic_type_mapping )
-
-			expect( res.values ).to eq( [
-					[ "1", "2.1", "t", "b" ],
-			] )
-
-			expect( result_typenames(res) ).to eq( ['bigint', 'double precision', 'boolean', 'text'] )
-		end
-
-		it "should do array param encoding" do
-			res = @conn.exec_params( "SELECT $1,$2,$3,$4", [
-					[1, 2, 3], [[1, 2], [3, nil]],
-					[1.11, 2.21],
-					['/,"'.gsub("/", "\\"), nil, 'abcäöü'],
-				], nil, basic_type_mapping )
-
-			expect( res.values ).to eq( [[
-					'{1,2,3}', '{{1,2},{3,NULL}}',
-					'{1.11,2.21}',
-					'{"//,/"",NULL,abcäöü}'.gsub("/", "\\"),
-			]] )
-
-			expect( result_typenames(res) ).to eq( ['bigint[]', 'bigint[]', 'double precision[]', 'text[]'] )
-		end
-	end
-
-
-
-	describe PG::BasicTypeMapForResults do
-		let!(:basic_type_mapping) do
-			PG::BasicTypeMapForResults.new @conn
-		end
-
-		#
-		# Decoding Examples
-		#
-
-		it "should do OID based type conversions", :ruby_19 do
-			res = @conn.exec( "SELECT 1, 'a', 2.0::FLOAT, TRUE, '2013-06-30'::DATE, generate_series(4,5)" )
-			expect( res.map_types!(basic_type_mapping).values ).to eq( [
-					[ 1, 'a', 2.0, true, Time.new(2013,6,30), 4 ],
-					[ 1, 'a', 2.0, true, Time.new(2013,6,30), 5 ],
-			] )
-		end
-
-		#
-		# Decoding Examples text+binary format converters
-		#
-
-		describe "connection wide type mapping" do
-			before :each do
-				@conn.type_map_for_results = basic_type_mapping
-			end
-
-			after :each do
-				@conn.type_map_for_results = PG::TypeMapAllStrings.new
-			end
-
-			it "should do boolean type conversions" do
-				[1, 0].each do |format|
-					res = @conn.exec( "SELECT true::BOOLEAN, false::BOOLEAN, NULL::BOOLEAN", [], format )
-					expect( res.values ).to eq( [[true, false, nil]] )
-				end
-			end
-
-			it "should do binary type conversions" do
-				[1, 0].each do |format|
-					res = @conn.exec( "SELECT E'\\\\000\\\\377'::BYTEA", [], format )
-					expect( res.values ).to eq( [[["00ff"].pack("H*")]] )
-					expect( res.values[0][0].encoding ).to eq( Encoding::ASCII_8BIT ) if Object.const_defined? :Encoding
-				end
-			end
-
-			it "should do integer type conversions" do
-				[1, 0].each do |format|
-					res = @conn.exec( "SELECT -8999::INT2, -899999999::INT4, -8999999999999999999::INT8", [], format )
-					expect( res.values ).to eq( [[-8999, -899999999, -8999999999999999999]] )
-				end
-			end
-
-			it "should do string type conversions" do
-				@conn.internal_encoding = 'utf-8' if Object.const_defined? :Encoding
-				[1, 0].each do |format|
-					res = @conn.exec( "SELECT 'abcäöü'::TEXT", [], format )
-					expect( res.values ).to eq( [['abcäöü']] )
-					expect( res.values[0][0].encoding ).to eq( Encoding::UTF_8 ) if Object.const_defined? :Encoding
-				end
-			end
-
-			it "should do float type conversions" do
-				[1, 0].each do |format|
-					res = @conn.exec( "SELECT -8.999e3::FLOAT4,
-														8.999e10::FLOAT4,
-														-8999999999e-99::FLOAT8,
-														NULL::FLOAT4,
-														'NaN'::FLOAT4,
-														'Infinity'::FLOAT4,
-														'-Infinity'::FLOAT4
-													", [], format )
-					expect( res.getvalue(0,0) ).to be_within(1e-2).of(-8.999e3)
-					expect( res.getvalue(0,1) ).to be_within(1e5).of(8.999e10)
-					expect( res.getvalue(0,2) ).to be_within(1e-109).of(-8999999999e-99)
-					expect( res.getvalue(0,3) ).to be_nil
-					expect( res.getvalue(0,4) ).to be_nan
-					expect( res.getvalue(0,5) ).to eq( Float::INFINITY )
-					expect( res.getvalue(0,6) ).to eq( -Float::INFINITY )
-				end
-			end
-
-			it "should do datetime without time zone type conversions" do
-				[0].each do |format|
-					res = @conn.exec( "SELECT CAST('2013-12-31 23:58:59+02' AS TIMESTAMP WITHOUT TIME ZONE),
-																		CAST('1913-12-31 23:58:59.123-03' AS TIMESTAMP WITHOUT TIME ZONE),
-																		CAST('infinity' AS TIMESTAMP WITHOUT TIME ZONE),
-																		CAST('-infinity' AS TIMESTAMP WITHOUT TIME ZONE)", [], format )
-					expect( res.getvalue(0,0) ).to eq( Time.new(2013, 12, 31, 23, 58, 59) )
-					expect( res.getvalue(0,1) ).to be_within(1e-3).of(Time.new(1913, 12, 31, 23, 58, 59.123))
-					expect( res.getvalue(0,2) ).to eq( 'infinity' )
-					expect( res.getvalue(0,3) ).to eq( '-infinity' )
-				end
-			end
-
-			it "should do datetime with time zone type conversions" do
-				[0].each do |format|
-					res = @conn.exec( "SELECT CAST('2013-12-31 23:58:59+02' AS TIMESTAMP WITH TIME ZONE),
-																		CAST('1913-12-31 23:58:59.123-03' AS TIMESTAMP WITH TIME ZONE),
-																		CAST('infinity' AS TIMESTAMP WITH TIME ZONE),
-																		CAST('-infinity' AS TIMESTAMP WITH TIME ZONE)", [], format )
-					expect( res.getvalue(0,0) ).to eq( Time.new(2013, 12, 31, 23, 58, 59, "+02:00") )
-					expect( res.getvalue(0,1) ).to be_within(1e-3).of(Time.new(1913, 12, 31, 23, 58, 59.123, "-03:00"))
-					expect( res.getvalue(0,2) ).to eq( 'infinity' )
-					expect( res.getvalue(0,3) ).to eq( '-infinity' )
-				end
-			end
-
-			it "should do date type conversions" do
-				[0].each do |format|
-					res = @conn.exec( "SELECT CAST('2113-12-31' AS DATE),
-																		CAST('1913-12-31' AS DATE),
-																		CAST('infinity' AS DATE),
-																		CAST('-infinity' AS DATE)", [], format )
-					expect( res.getvalue(0,0) ).to eq( Time.new(2113, 12, 31) )
-					expect( res.getvalue(0,1) ).to eq( Time.new(1913, 12, 31) )
-					expect( res.getvalue(0,2) ).to eq( 'infinity' )
-					expect( res.getvalue(0,3) ).to eq( '-infinity' )
-				end
-			end
-
-			it "should do array type conversions" do
-				[0].each do |format|
-					res = @conn.exec( "SELECT CAST('{1,2,3}' AS INT2[]), CAST('{{1,2},{3,4}}' AS INT2[][]),
-															CAST('{1,2,3}' AS INT4[]),
-															CAST('{1,2,3}' AS INT8[]),
-															CAST('{1,2,3}' AS TEXT[]),
-															CAST('{1,2,3}' AS VARCHAR[]),
-															CAST('{1,2,3}' AS FLOAT4[]),
-															CAST('{1,2,3}' AS FLOAT8[])
-														", [], format )
-					expect( res.getvalue(0,0) ).to eq( [1,2,3] )
-					expect( res.getvalue(0,1) ).to eq( [[1,2],[3,4]] )
-					expect( res.getvalue(0,2) ).to eq( [1,2,3] )
-					expect( res.getvalue(0,3) ).to eq( [1,2,3] )
-					expect( res.getvalue(0,4) ).to eq( ['1','2','3'] )
-					expect( res.getvalue(0,5) ).to eq( ['1','2','3'] )
-					expect( res.getvalue(0,6) ).to eq( [1.0,2.0,3.0] )
-					expect( res.getvalue(0,7) ).to eq( [1.0,2.0,3.0] )
-				end
-			end
-		end
-
-		context "with usage of result oids for copy decoder selection" do
-			it "can type cast #copy_data output with explicit decoder" do
-				@conn.exec( "CREATE TEMP TABLE copytable (t TEXT, i INT, ai INT[])" )
-				@conn.exec( "INSERT INTO copytable VALUES ('a', 123, '{5,4,3}'), ('b', 234, '{2,3}')" )
-
-				# Retrieve table OIDs per empty result.
-				res = @conn.exec( "SELECT * FROM copytable LIMIT 0" )
-				tm = basic_type_mapping.build_column_map( res )
-				row_decoder = PG::TextDecoder::CopyRow.new type_map: tm
-
-				rows = []
-				@conn.copy_data( "COPY copytable TO STDOUT", row_decoder ) do |res|
-					while row=@conn.get_copy_data
-						rows << row
-					end
-				end
-				expect( rows ).to eq( [['a', 123, [5,4,3]], ['b', 234, [2,3]]] )
-			end
-		end
-	end
-
-
-	describe PG::BasicTypeMapBasedOnResult do
-		let!(:basic_type_mapping) do
-			PG::BasicTypeMapBasedOnResult.new @conn
-		end
-
-		context "with usage of result oids for bind params encoder selection" do
-			it "can type cast query params" do
-				@conn.exec( "CREATE TEMP TABLE copytable (t TEXT, i INT, ai INT[])" )
-
-				# Retrieve table OIDs per empty result.
-				res = @conn.exec( "SELECT * FROM copytable LIMIT 0" )
-				tm = basic_type_mapping.build_column_map( res )
-
-				@conn.exec_params( "INSERT INTO copytable VALUES ($1, $2, $3)", ['a', 123, [5,4,3]], 0, tm )
-				@conn.exec_params( "INSERT INTO copytable VALUES ($1, $2, $3)", ['b', 234, [2,3]], 0, tm )
-				res = @conn.exec( "SELECT * FROM copytable" )
-				expect( res.values ).to eq( [['a', '123', '{5,4,3}'], ['b', '234', '{2,3}']] )
-			end
-		end
-
-		context "with usage of result oids for copy encoder selection" do
-			it "can type cast #copy_data input with explicit encoder" do
-				@conn.exec( "CREATE TEMP TABLE copytable (t TEXT, i INT, ai INT[])" )
-
-				# Retrieve table OIDs per empty result set.
-				res = @conn.exec( "SELECT * FROM copytable LIMIT 0" )
-				tm = basic_type_mapping.build_column_map( res )
-				row_encoder = PG::TextEncoder::CopyRow.new type_map: tm
-
-				@conn.copy_data( "COPY copytable FROM STDIN", row_encoder ) do |res|
-					@conn.put_copy_data ['a', 123, [5,4,3]]
-					@conn.put_copy_data ['b', 234, [2,3]]
-				end
-				res = @conn.exec( "SELECT * FROM copytable" )
-				expect( res.values ).to eq( [['a', '123', '{5,4,3}'], ['b', '234', '{2,3}']] )
-			end
-		end
-	end
-end