module RSpec::Mocks::Methods

def __mock_proxy
  @mock_proxy ||= begin
    mp = if Mock === self
      Proxy.new(self, @name, @options)
    else
      Proxy.new(self)
    end
    Serialization.fix_for(self)
    mp
  end
end

def as_null_object
  __mock_proxy.as_null_object
end

def format_chain(*chain, &blk)
  if Hash === chain.last
    hash = chain.pop
    hash.each do |k,v|
      chain << k
      blk = lambda { v }
    end
  end
  return chain.join('.').split('.'), blk
end

def null_object?
  __mock_proxy.null_object?
end

def received_message?(sym, *args, &block) #:nodoc:
  __mock_proxy.received_message?(sym.to_sym, *args, &block)
end

def rspec_reset #:nodoc:
  __mock_proxy.reset
end

def rspec_verify #:nodoc:
  __mock_proxy.verify
end

def should_not_receive(sym, &block)
  __mock_proxy.add_negative_message_expectation(caller(1)[0], sym.to_sym, &block)
end

def should_receive(sym, opts={}, &block)
  __mock_proxy.add_message_expectation(opts[:expected_from] || caller(1)[0], sym.to_sym, opts, &block)
end

def stub(sym_or_hash, opts={}, &block)
  if Hash === sym_or_hash
    sym_or_hash.each {|method, value| stub!(method).and_return value }
  else
    __mock_proxy.add_stub(caller(1)[0], sym_or_hash.to_sym, opts, &block)
  end
end

def stub_chain(*chain, &blk)
  chain, blk = format_chain(*chain, &blk)
  if chain.length > 1
    if matching_stub = __mock_proxy.__send__(:find_matching_method_stub, chain[0].to_sym)
      chain.shift
      matching_stub.invoke.stub_chain(*chain)
    else
      next_in_chain = Object.new
      stub(chain.shift) { next_in_chain }
      next_in_chain.stub_chain(*chain, &blk)
    end
  else
    stub(chain.shift, &blk)
  end
end

def unstub(sym)
  __mock_proxy.remove_stub(sym)
end