Sha256: 97d2daea21504af23db07aaf2ed4afae0e343434cefc8dfa5b08e0acbd46e125
Contents?: true
Size: 1.5 KB
Versions: 5
Compression:
Stored size: 1.5 KB
Contents
require 'time'
require 'timeout'
require 'socket'
# Borrowed from Webrat and Selenium client, watches for TCP port
# liveness of the spawned server.
# @private
class TCPSocket
def self.wait_for_service(options)
verbose_wait until listening_service?(options)
end
def self.wait_for_service_termination(options)
verbose_wait while listening_service?(options)
end
def self.listening_service?(options)
Timeout::timeout(options[:timeout] || 20) do
begin
socket = TCPSocket.new(options[:host], options[:port])
socket.close unless socket.nil?
true
rescue Errno::ECONNREFUSED,
Errno::EBADF # Windows
false
end
end
end
def self.verbose_wait
# Removed the puts call so as not to clutter up test output.
sleep 2
end
def self.wait_for_service_with_timeout(options)
start_time = Time.now
until listening_service?(options)
verbose_wait
if options[:timeout] && (Time.now > start_time + options[:timeout])
raise SocketError.new("Socket did not open within #{options[:timeout]} seconds")
end
end
end
def self.wait_for_service_termination_with_timeout(options)
start_time = Time.now
while listening_service?(options)
verbose_wait
if options[:timeout] && (Time.now > start_time + options[:timeout])
raise SocketError.new("Socket did not terminate within #{options[:timeout]} seconds")
end
end
end
end
Version data entries
5 entries across 5 versions & 1 rubygems