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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
require 'mocha/api'
require 'bourne/mockery'
module Mocha # :nodoc:
module API
# Asserts that the given mock received the given method.
#
# Examples:
#
# assert_received(mock, :to_s)
# assert_received(Radio, :new) {|expect| expect.with(1041) }
# assert_received(radio, :volume) {|expect| expect.with(11).twice }
def assert_received(mock, expected_method_name)
matcher = have_received(expected_method_name)
yield(matcher) if block_given?
assert matcher.matches?(mock), matcher.failure_message
end
class InvalidHaveReceived < StandardError
end
class HaveReceived #:nodoc:
def initialize(expected_method_name)
@expected_method_name = expected_method_name
@expectations = []
end
def method_missing(method, *args, &block)
@expectations << [method, args, block]
self
end
def matches?(mock)
if mock.respond_to?(:mocha)
@mock = mock.mocha
else
@mock = mock
end
@expectation = Expectation.new(@mock, @expected_method_name)
@expectations.each do |method, args, block|
@expectation.send(method, *args, &block)
end
invocation_count.times { @expectation.invoke }
@expectation.verified?
end
def does_not_match?(mock)
raise InvalidHaveReceived.new("should_not have_received(:#{@expected_method_name}) is invalid, please use" +
" should have_received(:#{@expected_method_name}).never")
end
def failure_message
message = ""
if matching_stubs.empty?
message << "unstubbed, "
end
message << @expectation.mocha_inspect
end
private
def invocation_count
matching_invocations.size
end
def matching_invocations
invocations.select do |invocation|
@expectation.match?(invocation.method_name, *invocation.arguments)
end
end
def invocations
Mockery.instance.invocations.select do |invocation|
invocation.mock.equal?(@mock) || invocation.mock.mocha.equal?(@mock)
end
end
def matching_stubs
Mockery.instance.stubba.stubba_methods.select do |method|
matching_stub?(method)
end
end
def matching_stub?(method)
method.mock.equal?(@mock) && method.method == @expected_method_name
end
end
# :call-seq:
# should have_received(method).with(arguments).times(times)
#
# Ensures that the given mock received the given method.
#
# Examples:
#
# mock.should have_received(:to_s)
# Radio.should have_received(:new).with(1041)
# radio.should have_received(:volume).with(11).twice
def have_received(expected_method_name)
HaveReceived.new(expected_method_name)
end
end
end
|
