Switching #within to use a stack of scopes

lib/webrat/core/session.rb

@@ -87,7 +87,8 @@ module Webrat
       save_and_open_page if exception_caught?
       flunk("Page load was not successful (Code: #{response_code.inspect}):\n#{formatted_error}") unless success_code?
       
-      @scope        = nil
+      @_scopes      = nil
+      @_page_scope  = nil
       @current_url  = url
       @http_method  = http_method
       @data         = data
@@ -104,7 +105,7 @@ module Webrat
     end
     
     def current_scope
-      @scope ||= Scope.new(self, response_body)
+      scopes.last || page_scope
     end
     
     # Reloads the last page requested. Note that this will resubmit forms
@@ -132,7 +133,10 @@ module Webrat
     alias_method :clicks_link_within, :click_link_within
     
     def within(selector)
-      yield Scope.new(self, response_body, selector)
+      scopes.push(Scope.new(self, response_body, selector))
+      ret = yield(current_scope)
+      scopes.pop
+      return ret
     end
     
     # Issues a GET request for a page, follows any redirects, and verifies the final page
@@ -160,6 +164,14 @@ module Webrat
       response_body
     end
 
+    def scopes
+      @_scopes ||= []
+    end
+
+    def page_scope
+      @_page_scope ||= Scope.new(self, response_body)
+    end
+    
     def_delegators :current_scope, :fill_in,            :fills_in
     def_delegators :current_scope, :check,              :checks
     def_delegators :current_scope, :uncheck,            :unchecks

spec/api/within_spec.rb

@@ -14,8 +14,8 @@ describe "within" do
     EOS
     
     @session.should_receive(:get).with("/page2", {})
-    @session.within "#container" do |scope|
-      scope.click_link "Link"
+    @session.within "#container" do
+      @session.click_link "Link"
     end
   end
   
@@ -32,9 +32,9 @@ describe "within" do
     EOS
     
     @session.should_receive(:get).with("/form2", "email" => "test@example.com")
-    @session.within "#form2" do |scope|
-      scope.fill_in "Email", :with => "test@example.com"
-      scope.click_button
+    @session.within "#form2" do
+      @session.fill_in "Email", :with => "test@example.com"
+      @session.click_button
     end
   end
   
@@ -47,9 +47,9 @@ describe "within" do
       </form>
     EOS
     
-    @session.within "#form2" do |scope|
+    @session.within "#form2" do
       lambda {
-        scope.click_button
+        @session.click_button
       }.should raise_error
     end
   end