Class: RSpec::Mocks::Proxy Private

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 45

attr_reader :object

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 82

def add_simple_expectation(method_name, response, location)
  method_double_for(method_name).add_simple_expectation method_name, response, @error_generator, location
end

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 143

def add_simple_stub(method_name, response)
  method_double_for(method_name).add_simple_stub method_name, response
end

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 137

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

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 55

def as_null_object
  @null_object = true
  @object
end

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 87

def build_expectation(method_name)
  meth_double = method_double_for(method_name)

  meth_double.build_expectation(
    @error_generator,
    @order_group
  )
end

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 120

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

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 37

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

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 18

def ensure_implemented(*_args)
  # noop for basic proxies, see VerifyingProxy for behaviour.
end

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 280

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

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 304

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

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 287

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

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 273

def find_matching_expectation(method_name, *args)
  find_best_matching_expectation_for(method_name) do |expectation|
    expectation.matches?(method_name, *args)
  end
end

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 299

def find_matching_method_stub(method_name, *args)
  method_double_for(method_name).stubs.find { |stub| stub.matches?(method_name, *args) }
end

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 184

def has_negative_expectation?(message)
  method_double_for(message).expectations.find { |expectation| expectation.negative_expectation_for?(message) }
end

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 198

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

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 177

def messages_arg_list
  @messages_received_mutex.synchronize do
    @messages_received.map { |_, args, _| args }
  end
end

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 269

def method_double_for(message)
  @method_doubles[message.to_sym]
end

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 263

def method_double_if_exists_for_message(message)
  method_double_for(message) if @method_doubles.key?(message.to_sym)
end

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 61

def original_method_handle_for(_message)
  nil
end

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 257

def prepended_modules_of_singleton_class
  @prepended_modules_of_singleton_class ||= RSpec::Mocks::Proxy.prepended_modules_of(@object.singleton_class)
end

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 236

def raise_missing_default_stub_error(expectation, args_for_multiple_calls)
  @error_generator.raise_missing_default_stub_error(expectation, args_for_multiple_calls)
end

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 231

def raise_unexpected_message_error(method_name, args)
  @error_generator.raise_unexpected_message_error method_name, args
end

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 170

def received_message?(method_name, *args, &block)
  @messages_received_mutex.synchronize do
    @messages_received.any? { |array| array == [method_name, args, block] }
  end
end

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 189

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

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 148

def remove_stub(method_name)
  method_double_for(method_name).remove_stub
end

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 153

def remove_stub_if_present(method_name)
  method_double_for(method_name).remove_stub_if_present
end

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 97

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

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 163

def reset
  @messages_received_mutex.synchronize do
    @messages_received.clear
  end
end

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 158

def verify
  @method_doubles.each_value { |d| d.verify }
end

# File 'rspec-mocks/lib/rspec/mocks/proxy.rb', line 241

def visibility_for(_method_name)
  # This is the default (for test doubles). Subclasses override this.
  :public
end