1 files changed, 15 insertions, 11 deletions

diff --git a/actionpack/lib/action_dispatch/testing/integration.rb b/actionpack/lib/action_dispatch/testing/integration.rb
index 9b00616968..f7f898288b 100644
--- a/actionpack/lib/action_dispatch/testing/integration.rb
+++ b/actionpack/lib/action_dispatch/testing/integration.rb
@@ -369,7 +369,7 @@ module ActionDispatch
           @request_count += 1
           @request  = ActionDispatch::Request.new(session.last_request.env)
           response = _mock_session.last_response
-          @response = ActionDispatch::TestResponse.new(response.status, response.headers, response.body)
+          @response = ActionDispatch::TestResponse.from_response(response)
           @html_document = nil
           @url_options = nil
 
@@ -388,16 +388,8 @@ module ActionDispatch
 
       APP_SESSIONS = {}
 
-      attr_reader :app
-
-      def before_setup
-        super
-        @app = nil
-        @integration_session = nil
-      end
-
-      def integration_session
-        @integration_session ||= create_session(app)
+      def app
+        @app ||= nil
       end
 
       # Reset the current session. This is useful for testing multiple sessions
@@ -425,6 +417,8 @@ module ActionDispatch
       %w(get post patch put head delete cookies assigns
          xml_http_request xhr get_via_redirect post_via_redirect).each do |method|
         define_method(method) do |*args|
+          reset! unless integration_session
+
           # reset the html_document variable, except for cookies/assigns calls
           unless method == 'cookies' || method == 'assigns'
             @html_document = nil
@@ -456,16 +450,19 @@ module ActionDispatch
       # Copy the instance variables from the current session instance into the
       # test instance.
       def copy_session_variables! #:nodoc:
+        return unless integration_session
         @controller = @integration_session.controller
         @response   = @integration_session.response
         @request    = @integration_session.request
       end
 
       def default_url_options
+        reset! unless integration_session
         integration_session.default_url_options
       end
 
       def default_url_options=(options)
+        reset! unless integration_session
         integration_session.default_url_options = options
       end
 
@@ -475,6 +472,7 @@ module ActionDispatch
 
       # Delegate unhandled messages to the current session instance.
       def method_missing(sym, *args, &block)
+        reset! unless integration_session
         if integration_session.respond_to?(sym)
           integration_session.__send__(sym, *args, &block).tap do
             copy_session_variables!
@@ -483,6 +481,11 @@ module ActionDispatch
           super
         end
       end
+
+      private
+        def integration_session
+          @integration_session ||= nil
+        end
     end
   end
 
@@ -659,6 +662,7 @@ module ActionDispatch
     end
 
     def url_options
+      reset! unless integration_session
       integration_session.url_options
     end