summaryrefslogtreecommitdiff
path: root/spec/mspec/lib/mspec/matchers/output.rb
blob: 551e7506cfeafbd406a120ae5a143abdc9bd8ac0 (plain)
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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
require 'mspec/helpers/io'

class OutputMatcher
  def initialize(stdout, stderr)
    @out = stdout
    @err = stderr
  end

  def matches?(proc)
    @saved_out = $stdout
    @saved_err = $stderr
    @stdout = $stdout = IOStub.new
    @stderr = $stderr = IOStub.new

    proc.call

    unless @out.nil?
      case @out
      when Regexp
        return false unless $stdout =~ @out
      else
        return false unless $stdout == @out
      end
    end

    unless @err.nil?
      case @err
      when Regexp
        return false unless $stderr =~ @err
      else
        return false unless $stderr == @err
      end
    end

    return true
  ensure
    $stdout = @saved_out
    $stderr = @saved_err
  end

  def failure_message
    expected_out = "\n"
    actual_out = "\n"
    unless @out.nil?
      expected_out += "  $stdout: #{@out.inspect}\n"
      actual_out += "  $stdout: #{@stdout.inspect}\n"
    end
    unless @err.nil?
      expected_out += "  $stderr: #{@err.inspect}\n"
      actual_out += "  $stderr: #{@stderr.inspect}\n"
    end
    ["Expected:#{expected_out}", "     got:#{actual_out}"]
  end

  def negative_failure_message
    out = ""
    out += "  $stdout: #{@stdout.chomp.dump}\n" unless @out.nil?
    out += "  $stderr: #{@stderr.chomp.dump}\n" unless @err.nil?
    ["Expected output not to be:\n", out]
  end
end

class Object
  def output(stdout=nil, stderr=nil)
    OutputMatcher.new(stdout, stderr)
  end
end