Capybara 2.0 Wait for ajax

wait_for_ajax.rb

require "timeout"

module WaitForAjax
  def self.delay
    @delay || 1
  end

  def self.delay= val
    @delay = val
  end

  def self.max_time
    @max_time || 10
  end

  def self.max_time= val
    @max_time = val
  end  

  def wait_for_ajax &block
    begin
      Timeout.timeout(WaitForAjax.max_time) do
        sleep(WaitForAjax.delay) and print '.' until value = yield
        print "\n"
        value
      end
    rescue TimeoutError
      false
    end
  end
end

# WaitForAjax.max_time = 20
# WaitForAjax.delay    = 0.5
# wait_for_ajax{ [true, false, false, false, false, false, false, false, false].sample }