module RSpec module Mocks # Implements the methods needed for a pure test double. RSpec::Mocks::Mock # includes this module, and it is provided for cases where you want a # pure test double without subclassing RSpec::Mocks::Mock. module TestDouble include Methods # Extends the TestDouble module onto the given object and # initializes it as a test double. # # @example # # module = Module.new # RSpec::Mocks::TestDouble.extend_onto(module, "MyMixin", :foo => "bar") # module.foo #=> "bar" def self.extend_onto(object, name=nil, stubs_and_options={}) object.extend self object.send(:__initialize_as_test_double, name, stubs_and_options) end # Creates a new test double with a `name` (that will be used in error # messages only) def initialize(name=nil, stubs_and_options={}) __initialize_as_test_double(name, stubs_and_options) end # This allows for comparing the mock to other objects that proxy such as # ActiveRecords belongs_to proxy objects. By making the other object run # the comparison, we're sure the call gets delegated to the proxy # target. def ==(other) other == __mock_proxy end # @private def inspect "#<#{self.class}:#{sprintf '0x%x', self.object_id} @name=#{@name.inspect}>" end # @private def to_s inspect.gsub('<','[').gsub('>',']') end alias_method :to_str, :to_s # @private def respond_to?(message, incl_private=false) __mock_proxy.null_object? && message != :to_ary ? true : super end private def __initialize_as_test_double(name=nil, stubs_and_options={}) if name.is_a?(Hash) && stubs_and_options.empty? stubs_and_options = name @name = nil else @name = name end @options = extract_options(stubs_and_options) assign_stubs(stubs_and_options) end def method_missing(message, *args, &block) raise NoMethodError if message == :to_ary return 0 if message == :to_int && __mock_proxy.null_object? __mock_proxy.record_message_received(message, *args, &block) begin __mock_proxy.null_object? ? self : super rescue NameError __mock_proxy.raise_unexpected_message_error(message, *args) end end def extract_options(stubs_and_options) if stubs_and_options[:null_object] @null_object = stubs_and_options.delete(:null_object) RSpec.deprecate(%Q["double('name', :null_object => true)"], %Q["double('name').as_null_object"]) end options = {} extract_option(stubs_and_options, options, :__declared_as, 'Mock') options end def extract_option(source, target, key, default=nil) if source[key] target[key] = source.delete(key) elsif default target[key] = default end end def assign_stubs(stubs) stubs.each_pair do |message, response| stub(message).and_return(response) end end end end end