class Mocha::StubbedMethod
Constants
- PrependedModule
Attributes
method_name[R]
stubbee[R]
Public Class Methods
new(stubbee, method_name)
click to toggle source
# File lib/mocha/stubbed_method.rb, line 10 def initialize(stubbee, method_name) @stubbee = stubbee @original_method = nil @original_visibility = nil @method_name = PRE_RUBY_V19 ? method_name.to_s : method_name.to_sym end
Public Instance Methods
define_new_method()
click to toggle source
# File lib/mocha/stubbed_method.rb, line 57 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.method_missing(method_name_in_scope, *args, &block) end retain_original_visibility(stub_method_owner) end
hide_original_method()
click to toggle source
# File lib/mocha/stubbed_method.rb, line 38 def hide_original_method return unless original_method_owner.__method_exists__?(method_name) store_original_method_visibility if use_prepended_module_for_stub_method? use_prepended_module_for_stub_method else begin store_original_method # rubocop:disable Lint/HandleExceptions rescue NameError # deal with nasties like ActiveRecord::Associations::AssociationProxy end # rubocop:enable Lint/HandleExceptions if stub_method_overwrites_original_method? remove_original_method_from_stubbee end end end
matches?(other)
click to toggle source
# File lib/mocha/stubbed_method.rb, line 82 def matches?(other) return false unless other.class == self.class (stubbee.object_id == other.stubbee.object_id) && (method_name == other.method_name) end
mock()
click to toggle source
# File lib/mocha/stubbed_method.rb, line 30 def mock mock_owner.mocha end
remove_new_method()
click to toggle source
# File lib/mocha/stubbed_method.rb, line 66 def remove_new_method stub_method_owner.send(:remove_method, method_name) end
reset_mocha()
click to toggle source
# File lib/mocha/stubbed_method.rb, line 34 def reset_mocha mock_owner.reset_mocha end
restore_original_method()
click to toggle source
# File lib/mocha/stubbed_method.rb, line 74 def restore_original_method return if use_prepended_module_for_stub_method? if stub_method_overwrites_original_method? original_method_owner.send(:define_method, method_name, method_body(@original_method)) end retain_original_visibility(original_method_owner) end
store_original_method()
click to toggle source
# File lib/mocha/stubbed_method.rb, line 70 def store_original_method @original_method = stubbee_method(method_name) end
stub()
click to toggle source
# File lib/mocha/stubbed_method.rb, line 17 def stub hide_original_method define_new_method end
to_s()
click to toggle source
# File lib/mocha/stubbed_method.rb, line 89 def to_s "#{stubbee}.#{method_name}" end
unstub()
click to toggle source
# File lib/mocha/stubbed_method.rb, line 22 def unstub remove_new_method restore_original_method mock.unstub(method_name.to_sym) return if mock.any_expectations? reset_mocha end
Private Instance Methods
remove_original_method_from_stubbee()
click to toggle source
# File lib/mocha/stubbed_method.rb, line 108 def remove_original_method_from_stubbee original_method_owner.send(:remove_method, method_name) end
retain_original_visibility(method_owner)
click to toggle source
# File lib/mocha/stubbed_method.rb, line 95 def retain_original_visibility(method_owner) return unless @original_visibility Module.instance_method(@original_visibility).bind(method_owner).call(method_name) end
store_original_method_visibility()
click to toggle source
# File lib/mocha/stubbed_method.rb, line 100 def store_original_method_visibility @original_visibility = original_method_owner.__method_visibility__(method_name) end
stub_method_overwrites_original_method?()
click to toggle source
# File lib/mocha/stubbed_method.rb, line 104 def stub_method_overwrites_original_method? @original_method && @original_method.owner == original_method_owner end
stub_method_owner()
click to toggle source
# File lib/mocha/stubbed_method.rb, line 121 def stub_method_owner @stub_method_owner ||= original_method_owner end
use_prepended_module_for_stub_method()
click to toggle source
# File lib/mocha/stubbed_method.rb, line 116 def use_prepended_module_for_stub_method @stub_method_owner = PrependedModule.new original_method_owner.__send__ :prepend, @stub_method_owner end
use_prepended_module_for_stub_method?()
click to toggle source
# File lib/mocha/stubbed_method.rb, line 112 def use_prepended_module_for_stub_method? RUBY_V2_PLUS end