lib/capy.rb in capy-1.4.0 vs lib/capy.rb in capy-1.4.1

@@ -101,11 +101,11 @@
           return
         else
           begin
             result = evaluater.eval_script(buf, mode)
             puts "=> #{result.inspect}".cyan
-          rescue => e
+          rescue Exception => e
             error e
           end
         end
       end
 
@@ -146,9 +146,27 @@
     end
     alias_method :js, :javascript
 
     def capybara(script)
       instance_eval script
+    end
+
+    def iframe(id)
+      raise 'no block given' unless block_given?
+      browser.switch_to.frame(id)
+      begin
+        yield
+      ensure
+        browser.switch_to.default_content
+      end
+    end
+
+    def driver
+      page.driver
+    end
+
+    def browser
+      driver.browser
     end
 
     def host(app_host)
       Capybara.app_host = app_host
     end