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
|
# frozen_string_literal: true
require 'ruby2_keywords'
require 'mocha/ruby_version'
module Mocha
class StubbedMethod
class PrependedModule < Module
end
attr_reader :stubba_object, :method_name
def initialize(stubba_object, method_name)
@stubba_object = stubba_object
@original_method = nil
@original_visibility = nil
@method_name = method_name.to_sym
end
def stub
hide_original_method
define_new_method
end
def unstub
remove_new_method
mock.unstub(method_name.to_sym)
return if mock.any_expectations?
reset_mocha
end
def mock
stubbee.mocha
end
def reset_mocha
stubbee.reset_mocha
end
def hide_original_method
return unless original_method_owner.__method_exists__?(method_name)
store_original_method_visibility
use_prepended_module_for_stub_method
end
def define_new_method
self_in_scope = self
method_name_in_scope = method_name
stub_method_owner.send(:define_method, method_name) do |*args, &block|
self_in_scope.mock.handle_method_call(method_name_in_scope, args, block)
end
stub_method_owner.send(:ruby2_keywords, method_name)
retain_original_visibility(stub_method_owner)
end
def remove_new_method
stub_method_owner.send(:remove_method, method_name)
end
def matches?(other)
return false unless other.instance_of?(self.class)
(stubba_object.object_id == other.stubba_object.object_id) && # rubocop:disable Lint/IdentityComparison
(method_name == other.method_name)
end
alias_method :==, :eql?
def to_s
"#{stubba_object}.#{method_name}"
end
private
def retain_original_visibility(method_owner)
return unless @original_visibility
Module.instance_method(@original_visibility).bind(method_owner).call(method_name)
end
def store_original_method_visibility
@original_visibility = original_method_owner.__method_visibility__(method_name)
end
def use_prepended_module_for_stub_method
@stub_method_owner = PrependedModule.new
original_method_owner.__send__ :prepend, @stub_method_owner
end
def stub_method_owner
@stub_method_owner ||= original_method_owner
end
end
end
|
