rwebunit 1.0.3 → 1.3.0

data/lib/rwebunit/driver.rb

@@ -6,113 +6,163 @@
 #    click_button("submit")
 #
 require File.join(File.dirname(__FILE__), 'itest_plugin')
+require File.join(File.dirname(__FILE__), 'popup')
 require 'timeout'
+require 'uri'
 
 module RWebUnit
   module Driver
     include RWebUnit::ITestPlugin
+    include RWebUnit::Popup
 
-    def browser
-      @web_browser
-    end
+    @@default_polling_interval = 1 # second
+    @@default_timeout = 30 # seconds
 
-    # Verify the next page following an operation.
+    # open a browser, and set base_url via hash, but does not acually
     #
-    # Typical usage:
-    #   login_page.click_login
-    #   expect_page HomePage
-    def expect_page(page_clazz, argument = nil)
-      if argument
-        page_clazz.new(@web_browser, argument)
+    # example:
+    #   open_browser :base_url => http://localhost:8080
+    #
+    # There are 3 ways to set base url
+    #   1. pass as first argument
+    #   2. If running using iTest2, used as confiured
+    #   3. Use default value set
+    def open_browser(base_url = nil, options = {})
+      base_url ||= $ITEST2_PROJECT_BASE_URL
+      base_url ||= $BASE_URL
+      raise "base_url must be set" if base_url.nil?
+
+      default_options = {:speed => "fast",
+        :visible => true,
+        :highlight_colour => 'yellow',
+        :close_others => true,
+        :start_new => false, 	# start a new browser always
+      :go => true}
+
+      options = default_options.merge options
+      options[:firefox] = true if "Firefox" == $ITEST2_BROWSER || "Firefox" == $BROWSER
+      ($ITEST2_HIDE_BROWSER) ? $HIDE_IE = true : $HIDE_IE = false
+
+      if base_url =~ /^file:/
+        uri_base = base_url
       else
-        page_clazz.new(@web_browser)
+        uri = URI.parse(base_url)
+        uri_base = "#{uri.scheme}://#{uri.host}:#{uri.port}"
       end
-    end
 
-    # Using Ruby block syntax to create interesting domain specific language,
-    # may be appeal to someone.
+      if options[:start_new]
+        @web_browser = WebBrowser.new(uri_base, nil, options)
+      else
+        @web_browser = WebBrowser.reuse(uri_base, options) # Reuse existing browser
+      end
 
-    # Example:
-    #  on @page do |i|
-    #    i.enter_text('btn1')
-    #    i.click_button('btn1')
-    #  end
-    def on(page, &block)
-      yield page
+      if base_url =~ /^file:/
+        goto_url(base_url) # for files, no base url
+      else
+        (uri.path.length == 0) ?  begin_at("/") :  begin_at(uri.path) if options[:go]
+      end
+
+      return @web_browser
     end
+    alias open_browser_with open_browser
 
-    # fail the test if user can perform the operation
+    # return the underlying RWebUnit::Browser object
+    def browser
+      @web_browser
+    end
+
+
+    # Close the current browser window (started by the script). If no browser started, then close
+    # all browser windows.
     #
-    # Example:
-    #  shall_not_allow { 1/0 }
-    def shall_not_allow(&block)
-      operation_performed_ok = false
-      begin
-        yield
-        operation_performed_ok  = true
-      rescue
+    def close_browser
+      if @web_browser
+        # Old iTest2 version
+        # @web_browser.close_browser unless $ITEST2_LEAVE_BROWSER_OPEN_AFTER_RUN
+        @web_browser.close_browser
+      else
+        WebBrowser.close_all_browsers
       end
-      raise "Operation shall not be allowed" if operation_performed_ok
     end
-    alias do_not_allow shall_not_allow
+    alias close_ie close_browser
 
-    def allow(&block)
-      operation_performed_ok = false
-      begin
-        yield
-        operation_performed_ok  = true
-      rescue
+
+    # Close all opening browser windows
+    #
+    def close_all_browsers
+      if is_firefox?
+        FireWatir::Firefox.close_all
+      else
+        Watir::IE.close_all
       end
-      operation_performed_ok
     end
 
-    # try operation, ignore if errors occur
+    # Verify the next page following an operation.
     #
-    # Example:
-    #   failsafe { click_link("Logout") }  # try logout, but it still OK if not being able to (already logout))
-    def failsafe(&block)
-      begin
-        yield
-      rescue =>e
+    # Typical usage:
+    #   login_page.click_login
+    #   expect_page HomePage
+    def expect_page(page_clazz, argument = nil)
+      if argument
+        page_clazz.new(@web_browser, argument)
+      else
+        page_clazz.new(@web_browser)
       end
     end
-    alias fail_safe failsafe
 
     def context
       @web_browser.context
     end
 
+    # Starting browser with a URL
+    #
+    # Example:
+    #    begin_at("http://www.itest2.com")
     def begin_at(url)
       dump_caller_stack
       @web_browser.begin_at(url)
     end
 
+    # Return the Watir::IE instance
+    #
     def ie
       @web_browser.ie
     end
 
-    def is_firefox?
-      @web_browser.is_firefox?
-    end
-
+    # Return the FireWatir::Firefox instance
+    #
     def firefox
       @web_browser.firefox
     end
 
-    def close_browser
-      dump_caller_stack
-      @web_browser.close_browser unless $ITEST2_LEAVE_BROWSER_OPEN_AFTER_RUN
+    def is_firefox?
+      @web_browser.is_firefox? if @web_browser
     end
-    alias close_ie close_browser
 
 
+    # Go to another page on the testing site.
+    #
+    #  open_browser("http://www.itest2.com")
+    #  goto_page("/demo")  # visit page http://www.itest2.com/demo
+    #
     def goto_page(page)
       operation_delay
       dump_caller_stack
-      @web_browser.goto_page(page);
+      @web_browser.goto_page(page) if @web_browser
     end
     alias visit goto_page
 
+    # Go to another web site, normally different site being tested on
+    #
+    #  open_browser("http://www.itest2.com")
+    #  goto_url("http://myorganized.info")
+    def goto_url(url)
+      @web_browser.goto_url url
+    end
+
+    # Attach to existinb browser window
+    #
+    #  attach_browser(:title,    )
     def attach_browser(how, what, options = {})
       options.merge!(:browser => is_firefox? ? "Firefox" : "IE")
       begin
@@ -123,10 +173,21 @@ module RWebUnit
       WebBrowser.attach_browser(how, what, options)
     end
 
+    # Reuse current an opened browser window instead of opening a new one
+    #	example:
+    #     use_current_browser(:title, /.*/) # use what ever browser window
+    #     use_current_browser(:title, "iTest2") # use browser window with title "iTest2"
+    def use_current_browser(how = :title, what = /.*/)
+      @web_browser = WebBrowser.attach_browser(how, what)
+    end
+
     ##
     #  Delegate to WebTester
     #
-
+    # Note:
+    #   label(:id, "abc") # OK
+    #   label(:id, :abc)  # Error
+    #
     # Depends on which object type, you can use following attribute
     # More details: http://wiki.openqa.org/display/WTR/Methods+supported+by+Element
     #
@@ -166,7 +227,8 @@ module RWebUnit
     [:area, :button, :cell, :checkbox, :div, :form, :frame, :h1, :h2, :h3, :h4, :h5, :h6, :hidden, :image, :li, :link, :map, :pre, :row, :radio, :select_list, :span, :table, :text_field, :paragraph, :file_field, :label].each do |method|
       define_method method do |*args|
         dump_caller_stack
-        @web_browser.send(method, *args)
+        # add check for @web_browser, in case the moudule included without init browser
+        @web_browser.send(method, *args) if @web_browser
       end
     end
     alias td cell
@@ -176,7 +238,8 @@ module RWebUnit
     [:back, :forward, :refresh].each do |method|
       define_method(method) do
         dump_caller_stack
-        @web_browser.send(method)
+        operation_delay
+        @web_browser.send(method)  if @web_browser
       end
     end
     alias refresh_page refresh
@@ -186,7 +249,7 @@ module RWebUnit
     [:images, :links, :buttons, :select_lists, :checkboxes, :radios, :text_fields].each do |method|
       define_method method do
         dump_caller_stack
-        @web_browser.send(method)
+        @web_browser.send(method)  if @web_browser
       end
     end
 
@@ -195,15 +258,16 @@ module RWebUnit
     # page.check_checkbox('bad_ones', 'Chicken Little')
     # page.check_checkbox('good_ones', ['Cars', 'Toy Story'])
     #
-    [:set_form_element, :click_link_with_text, :click_link_with_id, :submit, :click_button_with_id, :click_button_with_caption, :click_button_with_value, :click_radio_option, :clear_radio_option, :select_file_for_upload, :check_checkbox, :uncheck_checkbox, :select_option].each do |method|
+    [:set_form_element, :click_link_with_text, :click_link_with_id, :submit, :click_button_with_id, :click_button_with_name, :click_button_with_caption, :click_button_with_value, :click_radio_option, :clear_radio_option, :select_file_for_upload, :check_checkbox, :uncheck_checkbox, :select_option].each do |method|
       define_method method do |*args|
         dump_caller_stack
         operation_delay
-        @web_browser.send(method, *args)
+        @web_browser.send(method, *args)  if @web_browser
       end
     end
 
     alias enter_text set_form_element
+    alias set_hidden_field set_form_element
     alias click_link click_link_with_text
     alias click_button_with_text click_button_with_caption
     alias click_button click_button_with_caption
@@ -218,9 +282,13 @@ module RWebUnit
     end
 
     def contains_text(text)
-      @web_browser.contains_text(text);
+      @web_browser.contains_text(text)
     end
 
+    # Click image buttion with image source name
+    #
+    # For an image submit button <input name="submit" type="image" src="/images/search_button.gif">
+    #  click_button_with_image("search_button.gif")
     def click_button_with_image_src_contains(image_filename)
       dump_caller_stack
       operation_delay
@@ -241,6 +309,251 @@ module RWebUnit
       @web_browser.new_popup_window(options)
     end
 
+
+    # Warning: this does not work well with Firefox yet.
+    def element_text(elem_id)
+      @web_browser.element_value(elem_id)
+    end
+
+    # Identify DOM element by ID
+    # Warning: it is only supported on IE
+    def element_by_id(elem_id)
+      @web_browser.element_by_id(elem_id)
+    end
+
+    # ---
+    # For debugging
+    # ---
+    def dump_response(stream = nil)
+      @web_browser.dump_response(stream)
+    end
+
+    # For current page souce to a file in specified folder for inspection
+    #
+    #   save_current_page(:dir => "C:\\mysite", filename => "abc", :replacement => true)
+    def save_current_page(options = {})
+      default_options = {:replacement => true}
+      options = default_options.merge(options)
+      if options[:dir]
+        # already defined the dir
+        to_dir = options[:dir]
+      elsif $ITEST2_RUNNING_SPEC_ID
+
+        $ITEST2_DUMP_DIR = File.join($ITEST2_WORKING_DIR, "dump")
+        FileUtils.mkdir($ITEST2_DUMP_DIR) unless File.exists?($ITEST2_DUMP_DIR)
+
+        spec_run_id = $ITEST2_RUNNING_SPEC_ID
+        spec_run_dir_name = spec_run_id.to_s.rjust(4, "0") unless spec_run_id == "unknown"
+        to_dir = File.join($ITEST2_DUMP_DIR, spec_run_dir_name)
+      else
+        to_dir = ENV['TEMP_DIR'] || "C:\\temp"
+      end
+
+      if options[:filename]
+        file_name = options[:filename]
+      else
+        file_name = Time.now.strftime("%m%d%H%M%S") + ".html"
+      end
+
+      Dir.mkdir(to_dir) unless File.exists?(to_dir)
+      file = File.join(to_dir, file_name)
+
+      content = page_source
+      base_url = @web_browser.context.base_url
+      current_url = @web_browser.url
+      current_url =~ /(.*\/).*$/
+      current_url_parent = $1
+      if options[:replacement] && base_url =~ /^http:/
+
+        # <link rel="stylesheet" type="text/css" href="/stylesheets/default.css" />
+        # '<script type="text/javascript" src="http://www.jeroenwijering.com/embed/swfobject.js"></script>'
+        # <script type="text/javascript" src="/javascripts/prototype.js"></script>
+        # <script type="text/javascript" src="/javascripts/scriptaculous.js?load=effects,builder"></script>
+        # <script type="text/javascript" src="/javascripts/extensions/gallery/lightbox.js"></script>
+        # <link href="/stylesheets/extensions/gallery/lightbox.css" rel="stylesheet" type="text/css" />
+        # <img src="images/mission_48.png" />
+
+        modified_content = ""
+
+        content.each_line do |line|
+          if line =~ /<script\s+.*src=["'']?(.*)["'].*/i then
+            script_src = $1
+            substitute_relative_path_in_src_line(line, script_src, base_url, current_url_parent)
+          elsif line =~ /<link\s+.*href=["'']?(.*)["'].*/i then
+            link_href = $1
+            substitute_relative_path_in_src_line(line, link_href, base_url, current_url_parent)
+          elsif line =~ /<img\s+.*src=["'']?(.*)["'].*/i then
+            img_src = $1
+            substitute_relative_path_in_src_line(line, img_src, base_url, current_url_parent)
+          end
+
+          modified_content += line
+        end
+
+        File.new(file, "w").puts modified_content
+      else
+        File.new(file, "w").puts content
+
+      end
+
+
+    end
+
+    # substut
+    def substitute_relative_path_in_src_line(line, script_src, host_url, page_parent_url)
+      unless script_src =~ /^["']?http:/
+        host_url.slice!(-1) if ends_with?(host_url, "/")
+        if script_src =~ /^\s*\// # absolute_path
+          line.gsub!(script_src, "#{host_url}#{script_src}")
+        else  #relative_path
+          line.gsub!(script_src, "#{page_parent_url}#{script_src}")
+        end
+      end
+    end
+
+    def ends_with?(str, suffix)
+      suffix = suffix.to_s
+      str[-suffix.length, suffix.length] == suffix
+    end
+
+    # current web page title
+    def page_title
+      @web_browser.page_title
+    end
+
+    # current page source (in HTML)
+    def page_source
+      @web_browser.page_source
+    end
+
+    # return plain text view of page
+    def page_text
+      @web_browser.text
+    end
+
+    # return the text of specific (identified by attribute "id") label tag
+    # For page containing
+    #     <label id="preferred_ide">iTest2</label>
+    # label_with_id("preferred_ids") # => iTest2
+    def label_with_id(label_id)
+      label(:id, label_id.to_s).text
+    end
+
+    # return the text of specific (identified by attribute "id") span tag
+    # For page containing
+    #     <span id="preferred_recorder">iTest2/Watir Recorder</span>
+    # span_with_id("preferred_recorder") # => iTest2/Watir Recorder
+    def span_with_id(span_id)
+      span(:id, span_id).text
+    end
+
+    # return the text of specific (identified by attribute "id") ta tag
+    # For page containing
+    #     <td id="preferred_recorder">iTest2/Watir Recorder</span>
+    # td_with_id("preferred_recorder") # => iTest2/Watir Recorder
+    def cell_with_id(cell_id)
+      cell(:id, cell_id).text
+    end
+    alias table_data_with_id cell_with_id
+
+
+    def is_mac?
+      RUBY_PLATFORM.downcase.include?("darwin")
+    end
+
+    def is_windows?
+      RUBY_PLATFORM.downcase.include?("mswin") or RUBY_PLATFORM.downcase.include?("mingw32")
+    end
+
+    # Support browser (IE) operations using unicode
+    #  Example:
+    #   click_button("Google 搜索")
+    def support_utf8
+      if is_windows?
+        require 'win32ole'
+        WIN32OLE.codepage = WIN32OLE::CP_UTF8
+      end
+    end
+    alias support_unicode support_utf8
+
+    def is_linux?
+      RUBY_PLATFORM.downcase.include?("linux")
+    end
+
+    #= Convenient functions
+    #
+
+    # Using Ruby block syntax to create interesting domain specific language,
+    # may be appeal to someone.
+
+    # Example:
+    #  on @page do |i|
+    #    i.enter_text('btn1')
+    #    i.click_button('btn1')
+    #  end
+    def on(page, &block)
+      yield page
+    end
+
+    # fail the test if user can perform the operation
+    #
+    # Example:
+    #  shall_not_allow { 1/0 }
+    def shall_not_allow(&block)
+      operation_performed_ok = false
+      begin
+        yield
+        operation_performed_ok  = true
+      rescue
+      end
+      raise "Operation shall not be allowed" if operation_performed_ok
+    end
+    alias do_not_allow shall_not_allow
+
+    # Does not provide real function, other than make enhancing test syntax
+    #
+    # Example:
+    #   allow { click_button('Register') }
+    def allow(&block)
+      operation_performed_ok = false
+      begin
+        yield
+        operation_performed_ok  = true
+      rescue
+      end
+      operation_performed_ok
+    end
+    alias shall_allow  allow
+    alias allowing  allow
+
+    # try operation, ignore if errors occur
+    #
+    # Example:
+    #   failsafe { click_link("Logout") }  # try logout, but it still OK if not being able to (already logout))
+    def failsafe(&block)
+      begin
+        yield
+      rescue =>e
+      end
+    end
+    alias fail_safe failsafe
+
+
+    # Execute the provided block until either (1) it returns true, or
+    # (2) the timeout (in seconds) has been reached. If the timeout is reached,
+    # a TimeOutException will be raised. The block will always
+    # execute at least once.
+    #
+    # This does not handle error, if the given block raise error, the statement finish with error
+    # Examples:
+    #   wait_until {puts 'hello'}
+    #   wait_until { div(:id, :receipt_date).exists? }
+    #
+    def wait_until(timeout = @@default_timeout || 30, polling_interval = @@default_polling_interval || 1, &block)
+      waiter = Watir::Waiter.new(timeout, polling_interval)
+      waiter.wait_until { yield }
+    end
+
     # Wait for specific seconds for an Ajax update finish.
     # Trick: In your Rails application,
     #     :loading 	=> "Element.show('search_indicator');
@@ -256,7 +569,7 @@ module RWebUnit
     #
     # Warning: this method has not been fully tested, if you are not using Rails, change your parameter accordingly.
     #
-    def ajax_wait_for_element(element_id, seconds, status='show', check_interval=2)
+    def ajax_wait_for_element(element_id, seconds, status='show', check_interval = @@default_polling_interval)
       count = 0
       check_interval = 1 if check_interval < 1 or check_interval > seconds
       while count < (seconds / check_interval) do
@@ -273,8 +586,12 @@ module RWebUnit
       return false
     end
 
-    def wait_for_element(element_id, timeout = 30, interval = 0.5)
+    #Wait the element with given id to be present in web page
+    #
+    # Warning: this not working in Firefox, try use wait_util or try instead
+    def wait_for_element(element_id, timeout = @@default_timeout, interval = @@default_polling_interval)
       start_time = Time.now
+      #TODO might not work with Firefox
       until @web_browser.element_by_id(element_id) do
         sleep(interval)
         if (Time.now - start_time) > timeout
@@ -282,6 +599,7 @@ module RWebUnit
         end
       end
     end
+=begin
 
     # TODO: Firewatir does not suport retrieving style or outerHtml
     #    http://jira.openqa.org/browse/WTR-260
@@ -313,231 +631,78 @@ module RWebUnit
       end
     end
 
-    # Warning: this does not work well with Firefox yet.
-    def element_text(elem_id)
-      @web_browser.element_value(elem_id)
-    end
-
-    def element_by_id(elem_id)
-      @web_browser.element_by_id(elem_id)
-    end
-
-    # ---
-    # For debugging
-    # ---
-    def dump_response(stream = nil)
-      @web_browser.dump_response(stream)
-    end
-
-    def save_current_page(to_dir = ENV['TEMP_DIR'] || "C:\\temp")
-      Dir.mkdir(to_dir) unless File.exists?(to_dir)
-      file_name = Time.now.strftime("%m%d%H%M%S") + ".html"
-      file = File.join(to_dir, file_name)
-      File.new(file, "w").puts page_source
-    end
-
-    def click_popup_window(button, wait_time= 9, user_input=nil )
-      @web_browser.start_clicker(button, wait_time, user_input)
-      sleep 0.5
-    end
-
-    # return plain text view of page
-    def page_text
-      Hpricot(page_source).to_plain_text
-    end
-
-    def label_with_id(label_id)
-      label(:id, label_id).text
-    end
-
-    def span_with_id(span_id)
-      span(:id, span_id).text
-    end
-
-    def cell_with_id(cell_id)
-      cell(:id, cell_id).text
-    end
-    alias table_data_with_id cell_with_id
-
-    # run a separate process waiting for the popup window to click
-    #
-    #
-    def prepare_to_click_button_in_popup(button = "OK", wait_time = 3)
-      #  !@web_browser.is_firefox?
-      # TODO: firefox is OK
-      if RUBY_PLATFORM =~ /mswin/  then
-        start_checking_js_dialog(button, wait_time)
-      else
-        raise "this only support on Windows and on IE"
-      end
-    end
-
-    # Start a background process to click the button on a javascript popup window
-    def start_checking_js_dialog(button = "OK", wait_time = 3)
-      w = WinClicker.new
-      longName = File.expand_path(File.dirname(__FILE__)).gsub("/", "\\" )
-      shortName = w.getShortFileName(longName)
-      c = "start ruby #{shortName}\\clickJSDialog.rb #{button} #{wait_time} "
-      w.winsystem(c)
-      w = nil
-    end
-
-    # Click the button in javascript popup dialog
-    # Usage:
-    #   click_button_in_popup_after { click_link('Cancel')}
-    #   click_button_in_popup_after("OK") { click_link('Cancel')}
-    #
-    def click_button_in_popup_after(options = {:button => "OK", :wait_time => 3}, &block)
-      if RUBY_PLATFORM =~ /mswin/  then
-        start_checking_js_dialog(options[:button], options[:wait_time])
-        yield
-      else
-        raise "this only support on Windows and on IE"
+=end
+
+    # Try the operation up to specified times, and sleep given interval (in seconds)
+    # Error will be ignored until timeout
+    # Example
+    #    repeat_try(3, 2) { click_button('Search' } # 3 times, 6 seconds in total
+    #    repeat_try { click_button('Search' } # using default 5 tries, 2 second interval
+    def repeat_try(num_tries = @@default_timeout || 30, interval = @@default_polling_interval || 1, &block)
+      num_tries ||= 1
+      (num_tries - 1).times do |num|
+        begin
+          yield
+          return
+        rescue => e
+          # puts "debug: #{num} failed: #{e}"
+          sleep interval
+        end
       end
-    end
 
-
-    # Support of iTest to ajust the intervals between keystroke/mouse operations
-    def operation_delay
+      # last try, throw error if still fails
       begin
-        if $ITEST2_OPERATION_DELAY && $ITEST2_OPERATION_DELAY > 0 &&
-            $ITEST2_OPERATION_DELAY && $ITEST2_OPERATION_DELAY < 30000  then # max 30 seconds
-          sleep($ITEST2_OPERATION_DELAY / 1000)
-        end
-
-        while $ITEST2_PAUSE
-          debug("Paused, waiting ...")
-          sleep 1
-        end
+        yield
       rescue => e
-        puts "Error on delaying: #{e}"
-        # ignore
+        raise e.to_s + " after trying #{num_tries} times every #{interval} seconds"
       end
+      yield
     end
 
+    # TODO: syntax
 
-    def close_all_browsers
-      Watir::IE.close_all
-    end
-
-    def is_mac?
-      RUBY_PLATFORM.downcase.include?("darwin")
-    end
-
-    def is_windows?
-      RUBY_PLATFORM.downcase.include?("mswin")
-    end
-
-    def is_linux?
-      RUBY_PLATFORM.downcase.include?("linux")
-    end
-
-    # Start background thread to click popup windows
-    #  Warning:
-    #    Make browser window active
-    #    Don't mouse your mouse to focus other window during test execution
-    def check_for_popups
-      autoit = WIN32OLE.new('AutoItX3.Control')
-      #
-      # Do forever - assumes popups could occur anywhere/anytime in your
-      # application.
-      loop do
-        # Look for window with given title. Give up after 1 second.
-        ret = autoit.WinWait('Windows Internet Explorer', '', 1)
-        #
-        # If window found, send appropriate keystroke (e.g. {enter}, {Y}, {N}).
-        if (ret==1) then
-          autoit.Send('{enter}')
-        end
-        #
-        # Take a rest to avoid chewing up cycles and give another thread a go.
-        # Then resume the loop.
-        sleep(3)
-      end
-    end
+    # Try the operation up to specified timeout (in seconds), and sleep given interval (in seconds).
+    # Error will be ignored until timeout
+    # Example
+    #    try { click_link('waiting')}
+    #    try(10, 2) { click_button('Search' } # try to click the 'Search' button upto 10 seconds, try every 2 seconds
+    #    try { click_button('Search' }
+    def try(timeout = @@default_timeout, polling_interval = @@default_polling_interval || 1, &block)
+      start_time = Time.now
 
-    ##
-    #  Check for "Security Information" and "Security Alert" alert popup, click 'Yes'
-    #
-    # Usage: For individual test suite
-    #
-    # before(:all) do
-    #  $popup = Thread.new { check_for_alerts }
-    #  open_in_browser
-    #  ...
-    # end
-    #
-    # after(:all) do
-    #   close_browser
-    #   Thread.kill($popup)
-    # end
-    #
-    # or for all tests,
-    #  $popup = Thread.new { check_for_alerts }
-    #  at_exit{ Thread.kill($popup) }
-    def check_for_security_alerts
-      autoit = WIN32OLE.new('AutoItX3.Control')
-      loop do
-        ["Security Alert", "Security Information"].each do |win_title|
-          ret = autoit.WinWait(win_title, '', 1)
-          if (ret==1) then
-            autoit.Send('{Y}')
-          end
+      last_error = nil
+      until (duration = Time.now - start_time) > timeout
+        begin
+          return if yield
+          last_error = nil
+        rescue => e
+          last_error = e
         end
-        sleep(3)
-      end
-    end
-
-    def verify_alert(title = "Microsoft Internet Explorer", button = "OK")
-      if is_windows? && !is_firefox?
-        WIN32OLE.new('AutoItX3.Control').ControlClick(title, '', button)
-      else
-        raise "This function only supports IE"
+        sleep polling_interval
       end
-    end
-
-    def click_button_in_security_information_popup(button = "&Yes")
-      verify_alert("Security Information", "", button)
-    end
-    alias click_security_information_popup click_button_in_security_information_popup
-
-    def click_button_in_security_alert_popup(button = "&Yes")
-      verify_alert("Security Alert", "", button)
-    end
-    alias click_security_alert_popup click_button_in_security_alert_popup
 
-    def click_button_in_javascript_popup(button = "OK")
-      verify_alert()
-    end
-    alias click_javascript_popup click_button_in_javascript_popup
-
-    ##
-    # This only works for IEs
-    #   Cons:
-    #     - Slow
-    #     - only works in IE
-    #     - does not work for security alert ?
-    def ie_popup_clicker(button_name = "OK", max_wait = 15)
-      require 'watir/contrib/enabled_popup'
-      require 'win32ole'
-      hwnd = ie.enabled_popup(15)
-      if (hwnd)  #yeah! a popup
-        popup = WinClicker.new
-        popup.makeWindowActive(hwnd) #Activate the window.
-        popup.clickWindowsButton_hwnd(hwnd, button_name) #Click the button
-        #popup.clickWindowsButton(/Internet/,button_name,30)
-        popup = nil
-      end
+      raise "Timeout after #{duration.to_i} seconds with error: #{last_error}." if last_error
+      raise "Timeout after #{duration.to_i} seconds."
     end
+    alias try_upto try
 
     ##
     #  Convert :first to 1, :second to 2, and so on...
     def symbol_to_sequence(symb)
-      value = { :zero => 0, :first => 1, :second => 2, :third => 3,
-        :fourth => 4, :fifth => 5, :sixth => 6, :seventh => 7,
-        :eighth => 8, :ninth => 9, :tenth => 10 }[symb]
+      value = { :zero => 0,
+        :first => 1,
+        :second => 2,
+        :third => 3,
+        :fourth => 4,
+        :fifth => 5,
+        :sixth => 6,
+        :seventh => 7,
+        :eighth => 8,
+        :ninth => 9,
+      :tenth => 10 }[symb]
       return value || symb.to_i
     end
 
   end
+
 end