This doesn't feel like it belongs in `mocks/proxy.rb`.  Notice that you are asking `expectation` something and then doing something with `expectation.orig_object`.  This is all knowledge that should be internal to `MessageExpectation`.  (And same with the `stub` code above).