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
|
require 'rspec/mocks/framework'
require 'rspec/mocks/version'
module RSpec
module Mocks
class << self
attr_accessor :space
def setup(host)
(class << host; self; end).class_eval do
include RSpec::Mocks::ExampleMethods
end
self.space ||= RSpec::Mocks::Space.new
end
def verify
space.verify_all
end
def teardown
space.reset_all
end
def proxy_for(object)
space.proxy_for(object)
end
def proxies_of(klass)
space.proxies_of(klass)
end
def any_instance_recorder_for(klass)
space.any_instance_recorder_for(klass)
end
# Adds an allowance (stub) on `subject`
#
# @param subject the subject to which the message will be added
# @param message a symbol, representing the message that will be
# added.
# @param opts a hash of options, :expected_from is used to set the
# original call site
# @param block an optional implementation for the allowance
#
# @example Defines the implementation of `foo` on `bar`, using the passed block
# x = 0
# RSpec::Mocks.allow_message(bar, :foo) { x += 1 }
def allow_message(subject, message, opts={}, &block)
orig_caller = opts.fetch(:expected_from) { caller(1)[0] }
::RSpec::Mocks.proxy_for(subject).
add_stub(orig_caller, message.to_sym, opts, &block)
end
# Sets a message expectation on `subject`.
# @param subject the subject on which the message will be expected
# @param message a symbol, representing the message that will be
# expected.
# @param opts a hash of options, :expected_from is used to set the
# original call site
# @param block an optional implementation for the expectation
#
# @example Expect the message `foo` to receive `bar`, then call it
# RSpec::Mocks.expect_message(bar, :foo)
# bar.foo
def expect_message(subject, message, opts={}, &block)
orig_caller = opts.fetch(:expected_from) { caller(1)[0] }
::RSpec::Mocks.proxy_for(subject).
add_message_expectation(orig_caller, message.to_sym, opts, &block)
end
# @api private
KERNEL_METHOD_METHOD = ::Kernel.instance_method(:method)
# @api private
# Used internally to get a method handle for a particular object
# and method name.
#
# Includes handling for a few special cases:
#
# - Objects that redefine #method (e.g. an HTTPRequest struct)
# - BasicObject subclasses that mixin a Kernel dup (e.g. SimpleDelegator)
def method_handle_for(object, method_name)
if ::Kernel === object
KERNEL_METHOD_METHOD.bind(object).call(method_name)
else
object.method(method_name)
end
end
end
# @private
IGNORED_BACKTRACE_LINE = 'this backtrace line is ignored'
end
end
|
