diff options
Diffstat (limited to 'activeresource/test/setter_trap.rb')
-rw-r--r-- | activeresource/test/setter_trap.rb | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/activeresource/test/setter_trap.rb b/activeresource/test/setter_trap.rb new file mode 100644 index 0000000000..284938823a --- /dev/null +++ b/activeresource/test/setter_trap.rb @@ -0,0 +1,29 @@ +require 'builder' + +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($`) if method.to_s =~ /=$/ + @obj.send method, *args, &proc + end + + def rollback_sets + @cache.each { |k, v| @obj.send k, v } + end +end |