@private
We subclass `Module` in order to be able to easily detect our prepended module.
@private
@private
@private
@private
@private
# File lib/rspec/mocks/method_double.rb, line 9 def initialize(object, method_name, proxy) @method_name = method_name @object = object @proxy = proxy @original_visibility = nil @method_stasher = InstanceMethodStasher.new(object, method_name) @method_is_proxied = false @expectations = [] @stubs = [] end
@private
# File lib/rspec/mocks/method_double.rb, line 184 def add_default_stub(*args, &implementation) return if stubs.any? add_stub(*args, &implementation) end
@private
# File lib/rspec/mocks/method_double.rb, line 134 def add_expectation(error_generator, expectation_ordering, expected_from, opts, &implementation) configure_method expectation = message_expectation_class.new(error_generator, expectation_ordering, expected_from, self, :expectation, opts, &implementation) expectations << expectation expectation end
@private
# File lib/rspec/mocks/method_double.rb, line 170 def add_simple_expectation(method_name, response, error_generator, backtrace_line) setup_simple_method_double method_name, response, expectations, error_generator, backtrace_line end
A simple stub can only return a concrete value for a message, and cannot match on arguments. It is used as an optimization over `add_stub` / `add_expectation` where it is known in advance that this is all that will be required of a stub, such as when passing attributes to the `double` example method. They do not stash or restore existing method definitions.
@private
# File lib/rspec/mocks/method_double.rb, line 165 def add_simple_stub(method_name, response) setup_simple_method_double method_name, response, stubs end
@private
# File lib/rspec/mocks/method_double.rb, line 149 def add_stub(error_generator, expectation_ordering, expected_from, opts={}, &implementation) configure_method stub = message_expectation_class.new(error_generator, expectation_ordering, expected_from, self, :stub, opts, &implementation) stubs.unshift stub stub end
@private
# File lib/rspec/mocks/method_double.rb, line 143 def build_expectation(error_generator, expectation_ordering) expected_from = IGNORED_BACKTRACE_LINE message_expectation_class.new(error_generator, expectation_ordering, expected_from, self) end
@private
# File lib/rspec/mocks/method_double.rb, line 120 def clear expectations.clear stubs.clear end
@private
# File lib/rspec/mocks/method_double.rb, line 47 def configure_method @original_visibility = [visibility, method_name] @method_stasher.stash unless @method_is_proxied define_proxy_method end
@private
# File lib/rspec/mocks/method_double.rb, line 54 def define_proxy_method return if @method_is_proxied save_original_method! definition_target.class_exec(self, method_name, visibility) do |method_double, method_name, visibility| define_method(method_name) do |*args, &block| method_double.proxy_method_invoked(self, *args, &block) end self.__send__ visibility, method_name end @method_is_proxied = true end
The type of message expectation to create has been extracted to its own method so that subclasses can override it.
@private
# File lib/rspec/mocks/method_double.rb, line 129 def message_expectation_class MessageExpectation end
@private
# File lib/rspec/mocks/method_double.rb, line 42 def object_singleton_class class << @object; self; end end
# File lib/rspec/mocks/method_double.rb, line 21 def original_method # If original method is not present, uses the `method_missing` # handler of the object. This accounts for cases where the user has not # correctly defined `respond_to?`, and also 1.8 which does not provide # method handles for missing methods even if `respond_to?` is correct. @original_method ||= @method_stasher.original_method || @proxy.original_method_handle_for(method_name) || Proc.new do |*args, &block| @object.__send__(:method_missing, @method_name, *args, &block) end end
The implementation of the proxied method. Subclasses may override this method to perform additional operations.
@private
# File lib/rspec/mocks/method_double.rb, line 72 def proxy_method_invoked(obj, *args, &block) @proxy.message_received method_name, *args, &block end
@private
# File lib/rspec/mocks/method_double.rb, line 201 def raise_method_not_stubbed_error raise MockExpectationError, "The method `#{method_name}` was not stubbed or was already unstubbed" end
@private
# File lib/rspec/mocks/method_double.rb, line 190 def remove_stub raise_method_not_stubbed_error if stubs.empty? remove_stub_if_present end
@private
# File lib/rspec/mocks/method_double.rb, line 196 def remove_stub_if_present expectations.empty? ? reset : stubs.clear end
@private
# File lib/rspec/mocks/method_double.rb, line 114 def reset restore_original_method clear end
@private
# File lib/rspec/mocks/method_double.rb, line 77 def restore_original_method return show_frozen_warning if object_singleton_class.frozen? return unless @method_is_proxied definition_target.__send__(:remove_method, @method_name) if @method_stasher.method_is_stashed? @method_stasher.restore end restore_original_visibility @method_is_proxied = false end
@private
# File lib/rspec/mocks/method_double.rb, line 101 def restore_original_visibility return unless @original_visibility && MethodReference.method_defined_at_any_visibility?(object_singleton_class, @method_name) object_singleton_class.__send__(*@original_visibility) end
@private
# File lib/rspec/mocks/method_double.rb, line 175 def setup_simple_method_double(method_name, response, collection, error_generator = nil, backtrace_line = nil) define_proxy_method me = SimpleMessageExpectation.new(method_name, response, error_generator, backtrace_line) collection.unshift me me end
@private
# File lib/rspec/mocks/method_double.rb, line 92 def show_frozen_warning RSpec.warn_with( "WARNING: rspec-mocks was unable to restore the original `#{@method_name}` method on #{@object.inspect} because it has been frozen. If you reuse this object, `#{@method_name}` will continue to respond with its stub implementation.", :call_site => nil, :use_spec_location_as_call_site => true ) end
@private
# File lib/rspec/mocks/method_double.rb, line 109 def verify expectations.each {|e| e.verify_messages_received} end
@private
# File lib/rspec/mocks/method_double.rb, line 37 def visibility @proxy.visibility_for(@method_name) end
# File lib/rspec/mocks/method_double.rb, line 222 def definition_target @definition_target ||= usable_rspec_prepended_module || object_singleton_class end
# File lib/rspec/mocks/method_double.rb, line 242 def new_rspec_prepended_module RSpecPrependedModule.new.tap do |mod| object_singleton_class.__send__ :prepend, mod end end
# File lib/rspec/mocks/method_double.rb, line 226 def usable_rspec_prepended_module @proxy.prepended_modules_of_singleton_class.each do |mod| # If we have one of our modules prepended before one of the user's # modules that defines the method, use that, since our module's # definition will take precedence. return mod if RSpecPrependedModule === mod # If we hit a user module with the method defined first, # we must create a new prepend module, even if one exists later, # because ours will only take precedence if it comes first. return new_rspec_prepended_module if mod.method_defined?(method_name) end nil end