class RSpec::Mocks::Proxy
@private
def add_message_expectation(location, method_name, opts={}, &block)
- Private: -
def add_message_expectation(location, method_name, opts={}, &block) meth_double = method_double_for(method_name) if null_object? && !block meth_double.add_default_stub(@error_generator, @expectation_ordering, location, opts) do @object end end meth_double.add_expectation @error_generator, @expectation_ordering, location, opts, &block end
def add_simple_expectation(method_name, response, location)
- Private: -
def add_simple_expectation(method_name, response, location) method_double_for(method_name).add_simple_expectation method_name, response, @error_generator, location end
def add_simple_stub(method_name, response)
- Private: -
def add_simple_stub(method_name, response) method_double_for(method_name).add_simple_stub method_name, response end
def add_stub(location, method_name, opts={}, &implementation)
- Private: -
def add_stub(location, method_name, opts={}, &implementation) method_double_for(method_name).add_stub @error_generator, @expectation_ordering, location, opts, &implementation end
def as_null_object
- Private: -
def as_null_object @null_object = true @object end
def build_expectation(method_name)
- Private: -
def build_expectation(method_name) meth_double = method_double_for(method_name) meth_double.build_expectation( @error_generator, @expectation_ordering ) end
def check_for_unexpected_arguments(expectation)
- Private: -
def check_for_unexpected_arguments(expectation) @messages_received.each do |(method_name, args, _)| if expectation.matches_name_but_not_args(method_name, *args) raise_unexpected_message_args_error(expectation, *args) end end end
def find_almost_matching_expectation(method_name, *args)
def find_almost_matching_expectation(method_name, *args) find_best_matching_expectation_for(method_name) do |expectation| expectation.matches_name_but_not_args(method_name, *args) end end
def find_almost_matching_stub(method_name, *args)
def find_almost_matching_stub(method_name, *args) method_double_for(method_name).stubs.find {|stub| stub.matches_name_but_not_args(method_name, *args)} end
def find_best_matching_expectation_for(method_name)
def find_best_matching_expectation_for(method_name) first_match = nil method_double_for(method_name).expectations.each do |expectation| next unless yield expectation return expectation unless expectation.called_max_times? first_match ||= expectation end first_match end
def find_matching_expectation(method_name, *args)
def find_matching_expectation(method_name, *args) find_best_matching_expectation_for(method_name) do |expectation| expectation.matches?(method_name, *args) end end
def find_matching_method_stub(method_name, *args)
def find_matching_method_stub(method_name, *args) method_double_for(method_name).stubs.find {|stub| stub.matches?(method_name, *args)} end
def has_negative_expectation?(message)
- Private: -
def has_negative_expectation?(message) method_double_for(message).expectations.detect {|expectation| expectation.negative_expectation_for?(message)} end
def initialize(object, order_group, name=nil, options={})
- Private: -
def initialize(object, order_group, name=nil, options={}) @object = object @order_group = order_group @name = name @error_generator = ErrorGenerator.new(object, name) @expectation_ordering = RSpec::Mocks::space.expectation_ordering @messages_received = [] @options = options @null_object = false @method_doubles = Hash.new { |h, k| h[k] = MethodDouble.new(@object, k, self) } end
def message_received(message, *args, &block)
- Private: -
def message_received(message, *args, &block) record_message_received message, *args, &block expectation = find_matching_expectation(message, *args) stub = find_matching_method_stub(message, *args) if (stub && expectation && expectation.called_max_times?) || (stub && !expectation) expectation.increase_actual_received_count! if expectation && expectation.actual_received_count_matters? if expectation = find_almost_matching_expectation(message, *args) expectation.advise(*args) unless expectation.expected_messages_received? end stub.invoke(nil, *args, &block) elsif expectation expectation.invoke(stub, *args, &block) elsif expectation = find_almost_matching_expectation(message, *args) expectation.advise(*args) if null_object? unless expectation.expected_messages_received? raise_unexpected_message_args_error(expectation, *args) unless (has_negative_expectation?(message) or null_object?) elsif stub = find_almost_matching_stub(message, *args) stub.advise(*args) raise_missing_default_stub_error(stub, *args) elsif Class === @object @object.superclass.__send__(message, *args, &block) else @object.__send__(:method_missing, message, *args, &block) end end
def method_double_for(message)
def method_double_for(message) @method_doubles[message.to_sym] end
def method_handle_for(message)
- Private: -
def method_handle_for(message) nil end
def null_object?
- Private: -
def null_object? @null_object end
def raise_missing_default_stub_error(expectation, *args)
- Private: -
def raise_missing_default_stub_error(expectation, *args) @error_generator.raise_missing_default_stub_error(expectation, *args) end
def raise_unexpected_message_args_error(expectation, *args)
- Private: -
def raise_unexpected_message_args_error(expectation, *args) @error_generator.raise_unexpected_message_args_error(expectation, *args) end
def raise_unexpected_message_error(method_name, *args)
- Private: -
def raise_unexpected_message_error(method_name, *args) @error_generator.raise_unexpected_message_error method_name, *args end
def received_message?(method_name, *args, &block)
- Private: -
def received_message?(method_name, *args, &block) @messages_received.any? {|array| array == [method_name, args, block]} end
def record_message_received(message, *args, &block)
- Private: -
def record_message_received(message, *args, &block) @order_group.invoked SpecificMessage.new(object, message) @messages_received << [message, args, block] end
def remove_single_stub(method_name, stub)
- Private: -
def remove_single_stub(method_name, stub) method_double_for(method_name).remove_single_stub(stub) end
def remove_stub(method_name)
- Private: -
def remove_stub(method_name) method_double_for(method_name).remove_stub end
def replay_received_message_on(expectation, &block)
- Private: -
def replay_received_message_on(expectation, &block) expected_method_name = expectation.message meth_double = method_double_for(expected_method_name) if meth_double.expectations.any? @error_generator.raise_expectation_on_mocked_method(expected_method_name) end unless null_object? || meth_double.stubs.any? @error_generator.raise_expectation_on_unstubbed_method(expected_method_name) end @messages_received.each do |(actual_method_name, args, _)| if expectation.matches?(actual_method_name, *args) expectation.invoke(nil) block.call(*args) if block end end end
def reset
- Private: -
def reset @method_doubles.each_value {|d| d.reset} @messages_received.clear end
def verify
- Private: -
def verify @method_doubles.each_value {|d| d.verify} ensure reset end