spec/lib/appsignal/cli_spec.rb in appsignal-1.3.5 vs spec/lib/appsignal/cli_spec.rb in appsignal-1.3.6.beta.1

@@ -2,16 +2,14 @@
 
 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 }