Adding require and switching to class_eval

brynary · Nov 28, 2008 · ff2dd5d · ff2dd5d
1 parent 97bb64f
commit ff2dd5d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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,15 +72,15 @@ 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
       end
     end
   end
 
-  class IntegrationTest
+  IntegrationTest.class_eval do
     include Webrat::Methods
   end
 end