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
68
69
70
71
72
73
74
75
76
77
78

require 'rubygems'
require 'spec'

module SpyMatchers
  class MethodRecorder
    def method_missing(sym, *args)
      MethodVerifier.new({sym => args})
    end
  end

  def have_received
    MethodRecorder.new
  end

  class MethodVerifier
    def initialize(expected)
      @expected = expected
    end

    def matches?(target)
      @target = target
      @target.was_method_called?(@expected)
    end

    def failure_message
      "expected #{@target.__subject__.inspect} to have received #{@expected.keys[0]}(#{@expected.values[0]})"
    end

    def negative_failure_message
      "expected #{@target.__subject__.inspect} not to have received #{@expected.keys[0]}(#{@expected.values[0]})"
    end
  end
end

class Spy
  def self.on(subject)
    Spy.new(subject)
  end

  def initialize(subject)
    @subject = subject
    @method_calls = []
  end

  def was_method_called?(method_with_args)
    return @method_calls.include?(method_with_args)
  end

  def __subject__
    @subject
  end

  def method_missing(sym, *args)
    @method_calls << {sym => args}
    @subject.send(sym, *args)
  end
end

describe "ubiquitous assertion syntax" do
  include SpyMatchers

  it "should test state" do
    2.should == 1+1
  end

  it "should test an error" do
    lambda { [].not_a_method }.should raise_error(NoMethodError)
  end

  it "should test behavior" do
    array = Spy.on []
    [1, 2].each do |number|
      array << number
    end
    array.should have_received << 1
    array.should have_received << 2
  end
end