module RSpec::Mocks::VerifyingProxyMethods

def add_message_expectation(method_name, opts={}, &block)
  ensure_implemented(method_name)
  super
end

def add_simple_stub(method_name, *args)
  ensure_implemented(method_name)
  super
end

def add_stub(method_name, opts={}, &implementation)
  ensure_implemented(method_name)
  super
end

def ensure_implemented(method_name)
  if method_reference[method_name].unimplemented?
    @error_generator.raise_unimplemented_error(
      @doubled_module,
      method_name
    )
  end
end

def ensure_publicly_implemented(method_name, object)
  ensure_implemented(method_name)
  visibility = method_reference[method_name].visibility
  unless visibility == :public
    @error_generator.raise_non_public_error(method_name, visibility)
  end
end