class RSpec::Mocks::AnyInstance::StubChain

def create_message_expectation_on(instance)
  super do |proxy, expected_from|
    stub = proxy.add_stub(expected_from, *@expectation_args, &@expectation_block)
    @recorder.stubs[stub.message] << stub
    stub
  end
end

def expectation_fulfilled?
  true
end

def invocation_order
  @invocation_order ||= {
    :with => [nil],
    :and_return => [:with, nil],
    :and_raise => [:with, nil],
    :and_yield => [:with, nil],
    :and_call_original => [:with, nil]
  }
end

def verify_invocation_order(rspec_method_name, *args, &block)
  unless invocation_order[rspec_method_name].include?(last_message)
    raise(NoMethodError, "Undefined method #{rspec_method_name}")
  end
end