1 files changed, 28 insertions, 0 deletions
diff --git a/activeresource/test/abstract_unit.rb b/activeresource/test/abstract_unit.rb
index f19635826c..08b41329a7 100644
--- a/activeresource/test/abstract_unit.rb
+++ b/activeresource/test/abstract_unit.rb
@@ -6,3 +6,31 @@ require 'active_resource/http_mock'
 require 'active_support/breakpoint'
 
 ActiveResource::Base.logger = Logger.new("#{File.dirname(__FILE__)}/debug.log")
+
+class SetterTrap < Builder::BlankSlate
+  class << self
+    def rollback_sets(obj)
+      returning yield(setter_trap = new(obj)) do
+        setter_trap.rollback_sets
+      end
+    end
+  end
+  
+  def initialize(obj)
+    @cache = {}
+    @obj = obj
+  end
+  
+  def respond_to?(method)
+    @obj.respond_to?(method)
+  end
+  
+  def method_missing(method, *args, &proc)
+    @cache[method] = @obj.send(method.to_s[0 ... -1]) if method.to_s[-1 .. -1] == "=" unless @cache[method]
+    @obj.send method, *args, &proc
+  end
+  
+  def rollback_sets
+    @cache.each { |k, v| @obj.send k, v }
+  end
+end
+\ No newline at end of file