Use modules for webrat extension

lib/webrat.rb

@@ -1,6 +1,6 @@
 require File.join(File.dirname(__FILE__), "webrat", "rails_extensions")
 require File.join(File.dirname(__FILE__), "webrat", "session")
 
-class Webrat
+module Webrat
   VERSION = '0.2.1'
 end

lib/webrat/rails_extensions.rb

@@ -1,27 +1,25 @@
-module ActionController
-  module Integration
-
-    class Session
-      
-      unless instance_methods.include?("put_via_redirect")
-        # Waiting for http://dev.rubyonrails.org/ticket/10497 to be committed.
+module Webrat
+  module RedirectActions
     def put_via_redirect(path, parameters = {}, headers = {})
       put path, parameters, headers
       follow_redirect! while redirect?
       status
     end
-      end
 
-      unless instance_methods.include?("delete_via_redirect")
-        # Waiting for http://dev.rubyonrails.org/ticket/10497 to be committed.
     def delete_via_redirect(path, parameters = {}, headers = {})
       delete path, parameters, headers
       follow_redirect! while redirect?
       status
     end
   end
-
 end
 
+# Waiting for http://dev.rubyonrails.org/ticket/10497 to be committed.
+  
+module ActionController
+  module Integration
+    class Session
+      include Webrat::RedirectActions unless instance_methods.include?("put_via_redirect")
+    end
   end
 end

lib/webrat/session.rb

@@ -1,10 +1,8 @@
 require "hpricot"
 require "English"
 
-module ActionController
-  module Integration
-   
-    class Session
+module Webrat
+  module Session
     # Issues a GET request for a page, follows any redirects, and verifies the final page
     # load was successful.
     #
@@ -516,8 +514,13 @@ module ActionController
         nil
       end
     end
-      
+  end
 end
 
+module ActionController
+  module Integration
+    class Session
+      include Webrat::Session
+    end
   end
 end