Contents

require "selenium/webdriver/common/platform"
require "socket"

module Selenium
  module WebDriver
    class SocketPoller

      def initialize(host, port, timeout = 0, interval = 0.25)
        @host     = host
        @port     = Integer(port)
        @timeout  = Integer(timeout)
        @interval = interval
      end

      #
      # Returns true if the server is listening within the given timeout,
      # false otherwise.
      #
      # @return [Boolean]
      #

      def connected?
        with_timeout { listening? }
      end

      #
      # Returns true if the server has stopped listening within the given timeout,
      # false otherwise.
      #
      # @return [Boolean]
      #

      def closed?
        with_timeout { not listening? }
      end

      private

      SOCKET_ERRORS = [Errno::ECONNREFUSED, Errno::ENOTCONN, SocketError]
      SOCKET_ERRORS << Errno::EPERM if Platform.cygwin?

      def listening?
        # There's a bug in 1.9.1 on Windows where this will succeed even if no
        # one is listening. Users who hit that should upgrade their Ruby.
        TCPSocket.new(@host, @port).close
        true
      rescue *SOCKET_ERRORS => e
        $stderr.puts [@host, @port].inspect if $DEBUG
        false
      end

      def with_timeout(&blk)
        max_time = Time.now + @timeout

        (
          return true if yield
          wait
        ) until Time.now > max_time

        false
      end

      def wait
        sleep @interval
      end

    end # SocketPoller
  end # WebDriver
end # Selenium

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
selenium-webdriver-0.2.0	lib/selenium/webdriver/common/socket_poller.rb
selenium-webdriver-0.2.0.dev	lib/selenium/webdriver/common/socket_poller.rb
selenium-webdriver-0.1.4	lib/selenium/webdriver/common/socket_poller.rb
selenium-webdriver-0.1.3	lib/selenium/webdriver/common/socket_poller.rb
selenium-webdriver-0.1.3.dev	lib/selenium/webdriver/common/socket_poller.rb
selenium-webdriver-0.1.2	lib/selenium/webdriver/common/socket_poller.rb