class TCPSocket

# File lib/tcp_socket_extension.rb, line 14
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

# File lib/tcp_socket_extension.rb, line 27
def self.verbose_wait
  puts ".\n"
  sleep 2
end

# File lib/tcp_socket_extension.rb, line 6
def self.wait_for_service(options)
  verbose_wait until listening_service?(options)
end

# File lib/tcp_socket_extension.rb, line 10
def self.wait_for_service_termination(options)
  verbose_wait while listening_service?(options)
end