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