spec/lib/appsignal/cli_spec.rb in appsignal-1.3.6 vs spec/lib/appsignal/cli_spec.rb in appsignal-1.4.0.alpha.1

@@ -1,15 +1,18 @@
+require 'spec_helper'
 require 'appsignal/cli'
 
 describe Appsignal::CLI do
   let(:out_stream) { StringIO.new }
   let(:cli) { Appsignal::CLI }
   before do
+    @original_stdout = $stdout
+    $stdout = out_stream
     Dir.stub(:pwd => project_fixture_path)
     cli.options = {:environment => 'production'}
   end
-  around do |example|
-    capture_stdout(out_stream) { example.run }
+  after do
+    $stdout = @original_stdout
   end
 
   describe "#config" do
     subject { cli.config }