From 0e9f4731bb0a8aa3ea5a3ca04b87efd6d40a6f93 Mon Sep 17 00:00:00 2001
From: Bryan Helmkamp <bryan@brynary.com>
Date: Wed, 3 Dec 2008 22:23:13 -0500
Subject: [PATCH] Adding webrat.simulate and webrat.automate for using the same
 step mathers in multiple modes

---
 lib/webrat/core/methods.rb |  8 +++++++-
 lib/webrat/core/session.rb | 10 ++++++++++
 2 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/lib/webrat/core/methods.rb b/lib/webrat/core/methods.rb
index 648d8b28..0d993e80 100644
--- a/lib/webrat/core/methods.rb
+++ b/lib/webrat/core/methods.rb
@@ -11,6 +11,10 @@ def #{meth}(*args, &blk)
       end
     end
     
+    def webrat
+      webrat_session
+    end
+    
     def webrat_session
       @_webrat_session ||= ::Webrat.session_class.new(self)
     end
@@ -47,7 +51,9 @@ def webrat_session
       :select_date, :select_time, :select_datetime,
       :field_by_xpath,
       :field_with_id,
-      :selenium
+      :selenium,
+      :simulate, :automate
+      
       
     
   end
diff --git a/lib/webrat/core/session.rb b/lib/webrat/core/session.rb
index 3f789d70..0ca3ffa1 100644
--- a/lib/webrat/core/session.rb
+++ b/lib/webrat/core/session.rb
@@ -208,6 +208,16 @@ def xml_content_type?
       false
     end
     
+    def simulate
+      return if Webrat.configuration.mode == :selenium
+      yield
+    end
+    
+    def automate
+      return unless Webrat.configuration.mode == :selenium
+      yield
+    end
+    
     def_delegators :current_scope, :fill_in,            :fills_in
     def_delegators :current_scope, :set_hidden_field
     def_delegators :current_scope, :submit_form