Pass block args via method_missing

brynary · Jul 26, 2008 · 88a9a4c · 88a9a4c
1 parent 592f709
commit 88a9a4c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 12 deletions.
diff --git a/lib/webrat/core/session.rb b/lib/webrat/core/session.rb
@@ -44,13 +44,9 @@ def save_and_open_page
       current_page.save_and_open
     end
 
-    def within(selector, &block)
-      current_page.within(selector, &block)
-    end
-
-    def method_missing(name, *args)
+    def method_missing(name, *args, &block)
       if current_page.respond_to?(name)
-        current_page.send(name, *args)
+        current_page.send(name, *args, &block)
       else
         super
       end

diff --git a/lib/webrat/rails/session.rb b/lib/webrat/rails/session.rb
@@ -18,14 +18,10 @@ def visits(*args)
       def respond_to?(name)
         super || webrat_session.respond_to?(name)
       end
-
-      def within(selector, &block)
-        webrat_session.within(selector, &block)
-      end
 
-      def method_missing(name, *args)
+      def method_missing(name, *args, &block)
         if webrat_session.respond_to?(name)
-          webrat_session.send(name, *args)
+          webrat_session.send(name, *args, &block)
         else
           super
         end