spec/support/helpers/stdout_helper.rb in kata-1.3.1 vs spec/support/helpers/stdout_helper.rb in kata-1.3.2

@@ -1,11 +1,14 @@
-require 'stringio'
-
 def capture_stdout
-  $stdout = StringIO.new
-  $stdin = StringIO.new("y\n")
+  require 'stringio'
+
+  orig_stdout, orig_stdin         = $stdout, $stdin
+  captured_stdout, captured_stdin = StringIO.new, StringIO.new("y\n")
+  $stdout, $stdin                 = captured_stdout, captured_stdin
+
   yield
-  $stdout.string.strip
+
+  return captured_stdout.string.chomp
 ensure
-  $stdout = STDOUT
-  $stdin = STDIN
-end
+  $stdout = orig_stdout
+  $stdin = orig_stdin
+end
\ No newline at end of file