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

@@ -1,18 +1,15 @@
-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
-  after do
-    $stdout = @original_stdout
+  around do |example|
+    capture_stdout(out_stream) { example.run }
   end
 
   describe "#config" do
     subject { cli.config }