Cleanup session initializaton in Webrat::Methods

brynary · Aug 13, 2009 · 66251b6 · 66251b6
1 parent ec43bb9
commit 66251b6
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 8 deletions.
diff --git a/lib/webrat/core/methods.rb b/lib/webrat/core/methods.rb
@@ -16,11 +16,11 @@ def webrat
     end
 
     def webrat_session
-      @_webrat_session ||= ::Webrat::Session.new(webrat_adapter)
-    end
-
-    def webrat_adapter
-      @_webrat_adapter ||= Webrat.adapter_class.new(self)
+      @_webrat_session ||= begin
+        session = Webrat.session_class.new
+        session.adapter = Webrat.adapter_class.new(self) if session.respond_to?(:adapter=)
+        session
+      end
     end
 
     # all of these methods delegate to the @session, which should

diff --git a/lib/webrat/core/session.rb b/lib/webrat/core/session.rb
@@ -59,20 +59,22 @@ class Session
     extend Forwardable
     include Logging
     include SaveAndOpenPage
+
+    attr_accessor :adapter
+
     attr_reader :current_url
     attr_reader :elements
 
     def_delegators :@adapter, :response, :response_code, :response_body,
       :response_body=, :response_code=,
       :get, :post, :put, :delete
 
-    def initialize(adapter=nil)
+    def initialize(adapter = nil)
+      @adapter         = adapter
       @http_method     = :get
       @data            = {}
       @default_headers = {}
       @custom_headers  = {}
-      @adapter         = adapter
-
       reset
     end