examples/rr/double/double_reset_example.rb in rr-0.1.14 vs examples/rr/double/double_reset_example.rb in rr-0.1.15

@@ -1,10 +1,10 @@
 require "examples/example_helper"
 
 module RR
 describe Double, "#reset", :shared => true do
-  it "cleans up by removing the __rr__ method" do
+  it "cleans up by removing the __rr__method" do
     @double.bind
     @object.methods.should include("__rr__foobar")
 
     @double.reset
     @object.methods.should_not include("__rr__foobar")
@@ -43,17 +43,55 @@
       :original_foobar
     end
     @object.methods.should include(@method_name.to_s)
     @original_method = @object.method(@method_name)
     @double = Double.new(@space, @object, @method_name)
-  end
 
-  it "removes the method" do
     @double.bind
     @object.methods.should include(@method_name.to_s)
+  end
 
+  it "rebind original method" do
     @double.reset
     @object.methods.should include(@method_name.to_s)
     @object.foobar.should == :original_foobar
+  end
+
+  it "cleans up by removing the __rr__original_method" do
+    @double.bind
+    @object.methods.should include("__rr__foobar")
+
+    @double.reset
+    @object.methods.should_not include("__rr__foobar")
+  end
+end
+
+describe Double, "#reset when method with block exists" do
+  it_should_behave_like "RR::Double#reset"
+
+  before do
+    @space = Space.new
+    @object = Object.new
+    @method_name = :foobar
+    def @object.foobar
+      yield(:original_argument)
+    end
+    @object.methods.should include(@method_name.to_s)
+    @original_method = @object.method(@method_name)
+    @double = Double.new(@space, @object, @method_name)
+
+    @double.bind
+    @object.methods.should include(@method_name.to_s)
+  end
+
+  it "rebinds original method with block" do
+    @double.reset
+    @object.methods.should include(@method_name.to_s)
+    
+    original_argument = nil
+    @object.foobar do |arg|
+      original_argument = arg
+    end
+    original_argument.should == :original_argument
   end
 end
 end