class RSpec::Mocks::Proxy

@private

Constants

DEFAULT_MESSAGE_EXPECTATION_OPTS
SpecificMessage

@private

Attributes

@private

Public Class Methods

@private

# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 20
def initialize(object, order_group, options={})
  ensure_can_be_proxied!(object)

  @object = object
  @order_group = order_group
  @error_generator = ErrorGenerator.new(object)
  @messages_received = []
  @messages_received_mutex = Support::Mutex.new
  @options = options
  @null_object = false
  @method_doubles = Hash.new { |h, k| h[k] = MethodDouble.new(@object, k, self) }
end
# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 244
def self.prepended_modules_of(klass)
  ancestors = klass.ancestors

  # `|| 0` is necessary for Ruby 2.0, where the singleton class
  # is only in the ancestor list when there are prepended modules.
  singleton_index = ancestors.index(klass) || 0

  ancestors[0, singleton_index]
end

Public Instance Methods

@private

# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 65
def add_message_expectation(method_name, opts=DEFAULT_MESSAGE_EXPECTATION_OPTS, &block)
  location = opts.fetch(:expected_from) { CallerFilter.first_non_rspec_line }
  meth_double = method_double_for(method_name)

  if null_object? && !block
    meth_double.add_default_stub(@error_generator, @order_group, location, opts) do
      @object
    end
  end

  meth_double.add_expectation @error_generator, @order_group, location, opts, &block
end

@private

# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 79
def add_simple_expectation(method_name, response, location)
  method_double_for(method_name).add_simple_expectation method_name, response, @error_generator, location
end

@private

# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 140
def add_simple_stub(method_name, response)
  method_double_for(method_name).add_simple_stub method_name, response
end

@private

# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 134
def add_stub(method_name, opts={}, &implementation)
  location = opts.fetch(:expected_from) { CallerFilter.first_non_rspec_line }
  method_double_for(method_name).add_stub @error_generator, @order_group, location, opts, &implementation
end

@private Tells the object to ignore any messages that aren’t explicitly set as stubs or message expectations.

# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 52
def as_null_object
  @null_object = true
  @object
end

@private

# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 84
def build_expectation(method_name)
  meth_double = method_double_for(method_name)

  meth_double.build_expectation(
    @error_generator,
    @order_group
  )
end

@private

# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 117
def check_for_unexpected_arguments(expectation)
  @messages_received_mutex.synchronize do
    return if @messages_received.empty?

    return if @messages_received.any? { |method_name, args, _| expectation.matches?(method_name, *args) }

    name_but_not_args, others = @messages_received.partition do |(method_name, args, _)|
      expectation.matches_name_but_not_args(method_name, *args)
    end

    return if name_but_not_args.empty? && !others.empty?

    expectation.raise_unexpected_message_args_error(name_but_not_args.map { |args| args[1] })
  end
end

@private

# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 34
def ensure_can_be_proxied!(object)
  return unless object.is_a?(Symbol)

  msg = "Cannot proxy frozen objects. Symbols such as #{object} cannot be mocked or stubbed."
  raise ArgumentError, msg
end

@private

# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 15
def ensure_implemented(*_args)
  # noop for basic proxies, see VerifyingProxy for behaviour.
end

@private

# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 181
def has_negative_expectation?(message)
  method_double_for(message).expectations.find { |expectation| expectation.negative_expectation_for?(message) }
end

@private

# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 195
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.unadvise(messages_arg_list)
    expectation.invoke(stub, *args, &block)
  elsif (expectation = find_almost_matching_expectation(message, *args))
    expectation.advise(*args) if null_object? unless expectation.expected_messages_received?

    if null_object? || !has_negative_expectation?(message)
      expectation.raise_unexpected_message_args_error([args])
    end
  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

@private

# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 174
def messages_arg_list
  @messages_received_mutex.synchronize do
    @messages_received.map { |_, args, _| args }
  end
end

@private

# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 260
def method_double_if_exists_for_message(message)
  method_double_for(message) if @method_doubles.key?(message.to_sym)
end

@private

# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 45
def null_object?
  @null_object
end

@private

# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 58
def original_method_handle_for(_message)
  nil
end
# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 254
def prepended_modules_of_singleton_class
  @prepended_modules_of_singleton_class ||= RSpec::Mocks::Proxy.prepended_modules_of(@object.singleton_class)
end

@private

# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 233
def raise_missing_default_stub_error(expectation, args_for_multiple_calls)
  @error_generator.raise_missing_default_stub_error(expectation, args_for_multiple_calls)
end

@private

# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 228
def raise_unexpected_message_error(method_name, args)
  @error_generator.raise_unexpected_message_error method_name, args
end

@private

# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 167
def received_message?(method_name, *args, &block)
  @messages_received_mutex.synchronize do
    @messages_received.any? { |array| array == [method_name, args, block] }
  end
end

@private

# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 186
def record_message_received(message, *args, &block)
  @order_group.invoked SpecificMessage.new(object, message, args)
  @messages_received_mutex.synchronize do
    @messages_received << [message, args, block]
  end
end

@private

# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 145
def remove_stub(method_name)
  method_double_for(method_name).remove_stub
end

@private

# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 150
def remove_stub_if_present(method_name)
  method_double_for(method_name).remove_stub_if_present
end

@private

# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 94
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_mutex.synchronize do
    @messages_received.each do |(actual_method_name, args, received_block)|
      next unless expectation.matches?(actual_method_name, *args)

      expectation.safe_invoke(nil)
      block.call(*args, &received_block) if block
    end
  end
end

@private

# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 160
def reset
  @messages_received_mutex.synchronize do
    @messages_received.clear
  end
end

@private

# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 155
def verify
  @method_doubles.each_value { |d| d.verify }
end

@private

# File rspec-mocks/lib/rspec/mocks/proxy.rb, line 238
def visibility_for(_method_name)
  # This is the default (for test doubles). Subclasses override this.
  :public
end