spec_helper.rb reading STDIO

spec_helper.rb

require "#{File.dirname(__FILE__)}/../lib/transactions.rb"
include Transactions

def capture_stdout(&block)
  original_stdout = $stdout
  $stdout = fake = StringIO.new
  begin
    yield
  ensure
    $stdout = original_stdout
  end
  fake.string
end

class TransactionInput
  def initialize output
    @output = output
  end

  def should_output output
    @output.split("\n").should == output.split("\n")[1..-2].collect{|line| line.strip}
  end
end

def transaction_input input
  TransactionInput.new capture_stdout{ input.split("\n")[1..-2].each{|line| Message.new(line).process} }
end