diff --git a/lib/webrat/rails.rb b/lib/webrat/rails.rb index 48a008a..3488e41 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 @@ end 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 @@ module ActionController #:nodoc: end end - class IntegrationTest + IntegrationTest.class_eval do include Webrat::Methods end end