ruby--ruby/spec/bundler/support/streams.rb

16 lines
298 B
Ruby
Raw Normal View History

# frozen_string_literal: true
require "stringio"
def capture(*streams)
streams.map!(&:to_s)
begin
result = StringIO.new
streams.each {|stream| eval "$#{stream} = result" }
yield
ensure
streams.each {|stream| eval("$#{stream} = #{stream.upcase}") }
end
result.string
end