diff --git a/lib/webrat.rb b/lib/webrat.rb index 744fe10d..2419dba6 100644 --- 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 index 65e50567..7887aa70 100644 --- 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}`