module RBS::UnitTest::Spy

def self.wrap(object, method_name)
  spy = WrapSpy.new(object: object, method_name: method_name)
  if block_given?
    begin
      yield spy, spy.wrapped_object
    end
  else
    spy
  end
end