Wrap RAILS_ROOT to support Merb also

brynary · Apr 17, 2008 · 2e03051 · 2e03051
1 parent ede0073
commit 2e03051
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/lib/webrat.rb b/lib/webrat.rb
@@ -4,6 +4,9 @@
 
 module Webrat
   VERSION = '0.2.1'
+  def self.root
+    defined?(RAILS_ROOT) ? RAILS_ROOT : Merb.root
+  end
 end
 
 if defined?(Merb)

diff --git a/lib/webrat/page.rb b/lib/webrat/page.rb
@@ -94,10 +94,10 @@ def selects(option_text, options = {})
     # Example:
     #   save_and_open
     def save_and_open
-      return unless File.exist?(RAILS_ROOT + "/tmp")
+      return unless File.exist?(Webrat.root + "/tmp")
 
       filename = "webrat-#{Time.now.to_i}.html"
-      File.open(RAILS_ROOT + "/tmp/#{filename}", "w") do |f|
+      File.open(Webrat.root + "/tmp/#{filename}", "w") do |f|
         f.write response.body
       end
       `open tmp/#{filename}`