@private Object that is yielded to `expect` when one of the yield matchers is used. Provides information about the yield behavior of the object-under-test.
# File lib/rspec/matchers/built_in/yield.rb, line 20 def initialize(block) @block = block @used = false self.num_yields, self.yielded_args = 0, [] end
# File lib/rspec/matchers/built_in/yield.rb, line 9 def self.probe(block) probe = new(block) return probe unless probe.has_block? probe.assert_valid_expect_block! block.call(probe) probe.assert_used! probe end
# File lib/rspec/matchers/built_in/yield.rb, line 62 def assert_used! return if @used raise "You must pass the argument yielded to your expect block on " + "to the method-under-test as a block. It acts as a probe that " + "allows the matcher to detect whether or not the method-under-test " + "yields, and, if so, how many times, and what the yielded arguments " + "are." end
# File lib/rspec/matchers/built_in/yield.rb, line 71 def assert_valid_expect_block! return if @block.arity == 1 raise "Your expect block must accept an argument to be used with this " + "matcher. Pass the argument as a block on to the method you are testing." end
# File lib/rspec/matchers/built_in/yield.rb, line 26 def has_block? Proc === @block end
# File lib/rspec/matchers/built_in/yield.rb, line 41 def single_yield_args yielded_args.first end
# File lib/rspec/matchers/built_in/yield.rb, line 56 def successive_yield_args yielded_args.map do |arg_array| arg_array.size == 1 ? arg_array.first : arg_array end end
# File lib/rspec/matchers/built_in/yield.rb, line 30 def to_proc @used = true probe = self Proc.new do |*args| probe.num_yields += 1 probe.yielded_args << args nil # to indicate the block does not return a meaningful value end end
# File lib/rspec/matchers/built_in/yield.rb, line 45 def yielded_once?(matcher_name) case num_yields when 1 then true when 0 then false else raise "The #{matcher_name} matcher is not designed to be used with a " + "method that yields multiple times. Use the yield_successive_args " + "matcher for that case." end end