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
|
require 'mocha/mockery'
require 'mocha/class_method'
require 'mocha/any_instance_method'
module Mocha
# Methods added to all classes to allow mocking and stubbing on real (i.e. non-mock) objects.
module ClassMethods
# @private
def stubba_method
Mocha::ClassMethod
end
# @private
class AnyInstance
def initialize(klass)
@stubba_object = klass
end
def mocha
@mocha ||= Mocha::Mockery.instance.mock_impersonating_any_instance_of(@stubba_object)
end
def stubba_method
Mocha::AnyInstanceMethod
end
def stubba_object
@stubba_object
end
def method_exists?(method, include_public_methods = true)
if include_public_methods
return true if @stubba_object.public_instance_methods(include_superclass_methods = true).include?(method)
end
return true if @stubba_object.protected_instance_methods(include_superclass_methods = true).include?(method)
return true if @stubba_object.private_instance_methods(include_superclass_methods = true).include?(method)
return false
end
end
# @return [Mock] a mock object which will detect calls to any instance of this class.
# @raise [StubbingError] if attempting to stub method which is not allowed.
#
# @example Return false to invocation of +Product#save+ for any instance of +Product+.
# Product.any_instance.stubs(:save).returns(false)
# product_1 = Product.new
# assert_equal false, product_1.save
# product_2 = Product.new
# assert_equal false, product_2.save
def any_instance
if frozen?
raise StubbingError.new("can't stub method on frozen object: #{mocha_inspect}.any_instance", caller)
end
@any_instance ||= AnyInstance.new(self)
end
end
end
|
