class RSpec::Mocks::Proxy

def add_message_expectation(method_name, 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

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)
  method_double_for(method_name).add_simple_stub method_name, response
end

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

def as_null_object
  @null_object = true
  @object
end

def build_expectation(method_name)
  meth_double = method_double_for(method_name)
  meth_double.build_expectation(
    @error_generator,
    @order_group
  )
end

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 ensure_implemented(*args)
  # noop for basic proxies, see VerifyingProxy for behaviour.
end

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)
  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)
  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)
  find_best_matching_expectation_for(method_name) do |expectation|
    expectation.matches?(method_name, *args)
  end
end

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)
  method_double_for(message).expectations.detect {|expectation| expectation.negative_expectation_for?(message)}
end

def initialize(object, order_group, name=nil, options={})
  @object = object
  @order_group = order_group
  @name = name
  @error_generator = ErrorGenerator.new(object, name)
  @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)
  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)
  @method_doubles[message.to_sym]
end

def null_object?
  @null_object
end

def original_method_handle_for(message)
  nil
end

def prepended_modules_of_singleton_class
  @prepended_modules_of_singleton_class ||= begin
    singleton_class = @object.singleton_class
    singleton_class.ancestors.take_while do |mod|
      !(Class === mod || @object.equal?(mod))
    end
  end
end

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)
  @error_generator.raise_unexpected_message_args_error(expectation, *args)
end

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)
  @messages_received.any? {|array| array == [method_name, args, block]}
end

def record_message_received(message, *args, &block)
  @order_group.invoked SpecificMessage.new(object, message, args)
  @messages_received << [message, args, block]
end

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

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

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
  @messages_received.clear
end

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

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