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
|
require 'mocha/mockery'
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
class AnyInstance
def initialize(klass)
@stubba_object = klass
end
def mocha(instantiate = true)
if instantiate
@mocha ||= Mocha::Mockery.instance.mock_impersonating_any_instance_of(@stubba_object)
else
defined?(@mocha) ? @mocha : nil
end
end
def stubba_method
Mocha::AnyInstanceMethod
end
def stubba_class
@stubba_object
end
def respond_to?(symbol, include_all = false)
@stubba_object.allocate.respond_to?(symbol.to_sym, include_all)
end
attr_reader :stubba_object
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
# @private
# rubocop:disable Metrics/CyclomaticComplexity
def __method_visibility__(method, include_public_methods = true)
(include_public_methods && public_method_defined?(method) && :public) ||
(protected_method_defined?(method) && :protected) ||
(private_method_defined?(method) && :private)
end
# rubocop:enable Metrics/CyclomaticComplexity
alias_method :__method_exists__?, :__method_visibility__
end
end
|
