[rb] update spec helper methods to allow returning located elements

Author: titusfortner

rb/spec/integration/selenium/webdriver/spec_support/helpers.rb

@@ -56,10 +56,22 @@ def fix_windows_path(path)
 
         def long_wait
           @long_wait ||= Wait.new(timeout: 30)
+
+          return @long_wait unless block_given?
+
+          result = nil
+          @long_wait.until { result = yield }
+          result
         end
 
         def short_wait
           @short_wait ||= Wait.new(timeout: 3)
+
+          return @short_wait unless block_given?
+
+          result = nil
+          @short_wait.until { result = yield }
+          result
         end
 
         def wait_for_alert