extend the controller with special logic on recycle

For some reason, this special logic is only supposed to be executed on
the second request.  For now, we'll extend on recycle, but we should
figure out why this functionality can't be run on every request

1 files changed, 25 insertions, 10 deletions

diff --git a/actionpack/lib/action_controller/metal/testing.rb b/actionpack/lib/action_controller/metal/testing.rb
index a4718376aa..838da3322d 100644
--- a/actionpack/lib/action_controller/metal/testing.rb
+++ b/actionpack/lib/action_controller/metal/testing.rb
@@ -4,18 +4,33 @@ module ActionController
 
     include RackDelegation
 
-    def recycle!
-      @_url_options = nil
-    end
+    # This gets included on the second request. We only want to modify this
+    # behavior on the second request. Ugh.
+    module Recycled # :nodoc:
+      def set_response!(request)
+      end
+
+      def process(name)
+        ret = super
+        if cookies = @_request.env['action_dispatch.cookies']
+          cookies.write(@_response)
+        end
+        @_response.prepare!
+        ret
+      end
 
-    # TODO: Clean this up
-    def process_with_new_base_test(request, response)
-      ret = process(request.parameters[:action])
-      if cookies = @_request.env['action_dispatch.cookies']
-        cookies.write(@_response)
+      def recycled?
+        true
       end
-      @_response.prepare!
-      ret
+    end
+
+    def recycled? # :nodoc:
+      false
+    end
+
+    def recycle!
+      @_url_options = nil
+      extend Recycled unless recycled?
     end
 
     # TODO : Rewrite tests using controller.headers= to use Rack env