spec/unit/messaging_spec.rb in caricature-0.7.0 vs spec/unit/messaging_spec.rb in caricature-0.7.1

@@ -1,8 +1,9 @@
 require File.dirname(__FILE__) + "/../bacon_helper"
 
 class EmptyExpectations
+
   def find(method_name, *args, &b)
     nil
   end
 end
 
@@ -12,10 +13,14 @@
     @super_before_called = false
     @call_super_called = false
     @called_super_after = false
   end
 
+  def block
+    nil
+  end
+
   def super_before?
     @super_before_called = true
     false
   end
 
@@ -37,20 +42,91 @@
     5
   end
 end
 
 class ReturnValueExpectations
+
   def expectation
     @expectation ||= ReturnValueExpecation.new
     @expectation
   end
   def find(method_name, *args, &b)
     return expectation  if method_name == :a_message
     nil
   end
 end
 
+class PassThruBlockExpectation
+
+  def block
+    @block
+  end
+
+  def block=(val)
+    @block=val
+  end
+
+  def block_args
+    [5,6,7]
+  end
+
+  def execute(*args, &b)
+    b.call *block_args
+  end
+
+  def super_before?
+    false
+  end
+
+  def call_super?
+    false
+  end
+
+end
+
+class CallingBlock
+  def a_message(*args, &b)
+    b.call(7,8,9)
+  end
+end
+
+class BlockExpectation
+  def block
+    lambda { |*args| return args }
+  end
+
+  def super_before?
+    false
+  end
+
+  def call_super?
+    true
+  end
+
+  def execute(*args, &b)
+
+  end
+
+end
+
+class BlockExpectations
+
+  def initialize(pass_thru=true)
+    @pass_thru = pass_thru
+  end
+
+  def expectation
+    @expectation ||= (@pass_thru ? PassThruBlockExpectation.new : BlockExpectation.new)
+    @expectation
+  end
+  def find(method_name, *args, &b)
+    return expectation  if method_name == :a_message
+    nil
+  end
+end
+
+
 describe "Caricature::Messenger strategies" do
 
   describe "Caricature::RubyMessenger" do
 
     it "should return nil for any method name not in the expectation collection" do
@@ -82,13 +158,35 @@
         @messenger.expectations.expectation.should.have_called_call_super
       end
 
     end
 
+    describe "when an expectation with a block has been defined" do
 
+      it "should invoke the block that is passed with the args from the expectation only once"  do
+        messenger = Caricature::RubyMessenger.new BlockExpectations.new
+        messenger.recorder = Caricature::MethodCallRecorder.new
+        counter = 0
+        arguments = []
+        messenger.deliver(:a_message, nil) do |*args|
+          counter += 1
+          arguments = args
+        end
+        counter.should == 1
+        [5,6,7].should == arguments
+      end
 
+      it "should call the block that is defined on the expectation by super when call super is enabled" do
+        exp =  BlockExpectations.new(false)
+        messenger = Caricature::RubyMessenger.new exp, CallingBlock.new
+        result = messenger.deliver(:a_message, nil, exp.expectation)
+        [7,8,9].should == result
+      end
 
+    end
+
+
   end
 
   describe "Caricature::ClrClassMessenger" do
 
     it "should return nil for any method name not in the expectation collection" do
@@ -125,10 +223,33 @@
         @messenger.expectations.expectation.should.have_called_call_super
       end
 
     end
 
+    describe "when an expectation with a block has been defined" do
+
+      it "should invoke the block that is passed with the args from the expectation only once"  do
+        messenger = Caricature::ClrClassMessenger.new BlockExpectations.new
+        messenger.recorder = Caricature::MethodCallRecorder.new
+        counter = 0
+        arguments = []
+        messenger.deliver(:a_message, nil) do |*args|
+          counter += 1
+          arguments = args
+        end
+        counter.should == 1
+        [5,6,7].should == arguments
+      end
+
+      it "should call the block that is defined on the expectation by super when call super is enabled" do
+        messenger = Caricature::ClrClassMessenger.new BlockExpectations.new(false), CallingBlock.new
+        result = messenger.deliver(:a_message, nil)
+        [7,8,9].should == result
+      end
+
+    end
+
   end
 
   describe "Caricature::ClrInterfaceMessenger" do
 
     it "should return nil for any method name not in the expectation collection" do
@@ -161,9 +282,27 @@
       end
 
       it "should not call call_super? after executing the expectation" do
         @messenger.deliver(:a_message, nil)
         @messenger.expectations.expectation.should.not.have_called_call_super
+      end
+
+    end
+
+    describe "when an expectation with a block has been defined" do
+
+      it "should invoke the block that is passed with the args from the expectation only once"  do
+        exp = BlockExpectations.new
+        messenger = Caricature::ClrInterfaceMessenger.new exp
+        messenger.recorder = Caricature::MethodCallRecorder.new
+        counter = 0
+        arguments = []
+        messenger.deliver(:a_message, nil, exp.expectation) do |*args|
+          counter += 1
+          arguments = args
+        end
+        counter.should == 1
+        [5,6,7].should == arguments
       end
 
     end
 
   end
\ No newline at end of file