class RSpec::Mocks::Proxy

def add_message_expectation(location, method_name, opts={}, &block)        
  method_double[method_name].add_expectation @error_generator, @expectation_ordering, location, opts, &block
end

def add_negative_message_expectation(location, method_name, &implementation)
  method_double[method_name].add_negative_expectation @error_generator, @expectation_ordering, location, &implementation
end

def add_stub(location, method_name, opts={}, &implementation)
  method_double[method_name].add_stub @error_generator, @expectation_ordering, location, opts, &implementation
end

def allow_message_expectations_on_nil
  @warn_about_expectations_on_nil = false
  
  # ensure nil.rspec_verify is called even if an expectation is not set in the example
  # otherwise the allowance would effect subsequent examples
  $rspec_mocks.add(nil) unless $rspec_mocks.nil?
end

def allow_message_expectations_on_nil?
  !warn_about_expectations_on_nil
end

def already_proxied_respond_to # :nodoc:
  @already_proxied_respond_to = true
end

def already_proxied_respond_to? # :nodoc:
  @already_proxied_respond_to
end

def as_null_object
  @null_object = true
  @object
end

def find_almost_matching_expectation(method_name, *args)
  method_double[method_name].expectations.find {|expectation| expectation.matches_name_but_not_args(method_name, args)}
end

def find_matching_expectation(method_name, *args)
  method_double[method_name].expectations.find {|expectation| expectation.matches(method_name, args) && !expectation.called_max_times?} || 
  method_double[method_name].expectations.find {|expectation| expectation.matches(method_name, args)}
end

def find_matching_method_stub(method_name, *args)
  method_double[method_name].stubs.find {|stub| stub.matches(method_name, args)}
end

def has_negative_expectation?(method_name)
  method_double[method_name].expectations.detect {|expectation| expectation.negative_expectation_for?(method_name)}
end

def initialize(object, name=nil, options={})
  @object = object
  @name = name
  @error_generator = ErrorGenerator.new object, name, options
  @expectation_ordering = OrderGroup.new @error_generator
  @messages_received = []
  @options = options
  @already_proxied_respond_to = false
  @null_object = false
end

def message_received(method_name, *args, &block)
  expectation = find_matching_expectation(method_name, *args)
  stub = find_matching_method_stub(method_name, *args)
  if (stub && expectation && expectation.called_max_times?) || (stub && !expectation)
    if expectation = find_almost_matching_expectation(method_name, *args)
      expectation.advise(args, block) unless expectation.expected_messages_received?
    end
    stub.invoke(args, block)
  elsif expectation
    expectation.invoke(args, block)
  elsif expectation = find_almost_matching_expectation(method_name, *args)
    expectation.advise(args, block) if null_object? unless expectation.expected_messages_received?
    raise_unexpected_message_args_error(expectation, *args) unless (has_negative_expectation?(method_name) or null_object?)
  elsif @object.is_a?(Class)
    @object.superclass.send(method_name, *args, &block)
  else
    @object.__send__ :method_missing, method_name, *args, &block
  end
end

def method_double
  @method_double ||= Hash.new {|h,k|
    h[k] = MethodDouble.new(@object, k, self)
  }
end

def method_doubles
  method_double.values
end

def null_object?
  @null_object
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(method_name, args, block)
  @messages_received << [method_name, args, block]
end

def remove_stub(method_name)
  method_double[method_name].remove_stub
end

def reset
  method_doubles.each {|d| d.reset}
end

def verify #:nodoc:
  method_doubles.each {|d| d.verify}
ensure
  reset
end

def warn_about_expectations_on_nil
  defined?(@warn_about_expectations_on_nil) ? @warn_about_expectations_on_nil : true
end

def warn_about_expectations_on_nil=(new_value)
  @warn_about_expectations_on_nil = new_value
end