File: output.rb

package info (click to toggle)
ruby-test-unit 3.2.8-1
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 1,028 kB
  • sloc: ruby: 13,799; makefile: 12
file content (31 lines) | stat: -rw-r--r-- 741 bytes parent folder | download | duplicates (9)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
module Test
  module Unit
    module Util
      module Output
        ##
        # Returns output for standard output and standard
        # error as string.
        #
        # Example:
        #   capture_output do
        #     puts("stdout")
        #     warn("stderr")
        #   end # -> ["stdout\n", "stderr\n"]
        def capture_output
          require 'stringio'

          output = StringIO.new
          error = StringIO.new
          stdout_save, stderr_save = $stdout, $stderr
          $stdout, $stderr = output, error
          begin
            yield
            [output.string, error.string]
          ensure
            $stdout, $stderr = stdout_save, stderr_save
          end
        end
      end
    end
  end
end