lib/rufus.rb in rufus-0.4 vs lib/rufus.rb in rufus-0.5

@@ -29,14 +29,30 @@
   end
 
   def enabled_after_wait?(view, timeout = 5)
     wait = Selenium::WebDriver::Wait.new :timeout => timeout
     begin
-      wait.until{view.displayed?}
+      wait.until{view.enabled?}
     rescue Selenium::WebDriver::Error::TimeOutError
       return false
     end
     true
+  end
+
+  def find(locator)
+    selenium.find(locator)
+  end
+
+  def click(locator)
+    selenium.find(locator).click
+  end
+
+  def page_source
+    selenium.page_source
+  end
+
+  def elements_of_type(type)
+    selenium.elements_by_tag(type)
   end
 
 end