Delegate #present? to Element. Closes #216

watir · Sep 12, 2013 · 231002d · jarmo · Sep 12, 2013 · p0deje
1 parent 1fc91c9
commit 231002d
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/lib/watir-webdriver/wait.rb b/lib/watir-webdriver/wait.rb
@@ -1,4 +1,6 @@
 # encoding: utf-8
+require 'forwardable'
+
 module Watir
   module Wait
 
@@ -80,6 +82,10 @@ def wait_while(*args, &blk)
   #
 
   class WhenPresentDecorator
+    extend Forwardable
+
+    def_delegator :@element, :present?
+
     def initialize(element, timeout, message = nil)
       @element = element
       @timeout = timeout

diff --git a/spec/wait_spec.rb b/spec/wait_spec.rb
@@ -83,6 +83,16 @@
         decorator.should respond_to(:present?)
         decorator.should respond_to(:click)
       end
+
+      it "delegates present? to element" do
+        Object.class_eval do
+          def present?
+            false
+          end
+        end
+        element = browser.a(:id, "show_bar").when_present(1)
+        element.should be_present
+      end
     end
 
     describe "#wait_until_present" do