diff --git a/lib/webrat/rails.rb b/lib/webrat/rails.rb index 48a008a2..3488e41c 100644 --- a/lib/webrat/rails.rb +++ b/lib/webrat/rails.rb @@ -1,4 +1,5 @@ require "webrat" +require "action_controller/integration" module Webrat class RailsSession < Session #:nodoc: @@ -71,7 +72,7 @@ def response #:nodoc: module ActionController #:nodoc: module Integration #:nodoc: - class Session #:nodoc: + Session.class_eval do unless instance_methods.include?("put_via_redirect") require "webrat/rails/redirect_actions" include Webrat::RedirectActions @@ -79,7 +80,7 @@ class Session #:nodoc: end end - class IntegrationTest + IntegrationTest.class_eval do include Webrat::Methods end end