spec/spec_helper.rb in work-bench-1.0.7 vs spec/spec_helper.rb in work-bench-1.0.8

@@ -2,5 +2,26 @@
 
 require 'rubygems'
 
 require 'work_bench'
 require 'rspec'
+require 'stringio'
+
+RSpec.configure do |config|
+
+  config.before :each do
+    ARGV.replace []
+  end
+
+  def capture(stream)
+    begin
+      stream = stream.to_s
+      eval "$#{stream} = StringIO.new"
+      yield
+      result = eval("$#{stream}").string
+    ensure
+      eval("$#{stream} = #{stream.upcase}")
+    end
+    result
+  end
+
+end