Contents

MAX_TIMES = 20
WAIT_TIME = 0.1

def _wait_for_exceptions
  exceptions = [ Capybara::ElementNotFound ]
  if defined?(Capybara::Driver::Webkit::Node::ElementNotDisplayedError)
    exceptions << Capybara::Driver::Webkit::Node::ElementNotDisplayedError
  end
  if defined?(Selenium::WebDriver::Error::StaleElementReferenceError)
    exceptions << Selenium::WebDriver::Error::StaleElementReferenceError
  end

  exceptions
end

def _wait_for_not_exceptions
  exceptions = [ Capybara::ElementNotFound ]
  if defined?(Capybara::Driver::Webkit::NodeNotAttachedError)
    exceptions << Capybara::Driver::Webkit::NodeNotAttachedError
  end

  exceptions
end

def _wait_for_not_continue_exceptions
  exceptions = []
  if defined?(Selenium::WebDriver::Error::StaleElementReferenceError)
    exceptions << Selenium::WebDriver::Error::StaleElementReferenceError
  end

  exceptions
end

class WaitingForElementFailure < StandardError
  def initialize(code)
    @code = code
  end

  def message
    @code
  end

  def backtrace
    []
  end
end

def wait_for(times = MAX_TIMES, &block)
  $stderr.puts "wait_for deprecated. Just use find."

  1.upto(times) do
    ok = false

    begin
      ok = block.()
    rescue *_wait_for_exceptions
      ok = false
    end

    if ok
      return
    else
      sleep WAIT_TIME
    end
  end

  raise WaitingForElementFailure.new(block)
end

def wait_for_not(times = MAX_TIMES, &block)
  $stderr.puts "wait_for_not deprecated. Just use find."

  original_time = Capybara.default_wait_time
  Capybara.default_wait_time = 0

  1.upto(times) do
    ok = false

    begin
      block.()
    rescue *_wait_for_not_continue_exceptions
      ok = false
    rescue *_wait_for_not_exceptions
      ok = true
    end

    if ok
      Capybara.default_wait_time = original_time

      return
    else
      sleep WAIT_TIME
    end
  end

  raise WaitingForElementFailure.new(block)
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
cuke-pack-0.0.3	lib/cuke-pack/support/wait_for.rb