module Spec module Mocks class Mock include Methods # Creates a new mock with a +name+ (that will be used in error messages only) # == Options: # * :null_object - if true, the mock object acts as a forgiving null object allowing any message to be sent to it. def initialize(name, stubs_and_options={}) @name = name @options = parse_options(stubs_and_options) assign_stubs(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 # This is an unfortunate side effect from ActiveRecord, but this should be safe unless the RHS redefines == in a nonsensical manner def ==(other) other == __mock_proxy end def method_missing(sym, *args, &block) __mock_proxy.instance_eval {@messages_received << [sym, args, block]} begin return self if __mock_proxy.null_object? super(sym, *args, &block) rescue NameError __mock_proxy.raise_unexpected_message_error sym, *args end end def inspect "#<#{self.class}:#{sprintf '0x%x', self.object_id} @name=#{@name.inspect}>" end private def parse_options(options) options.has_key?(:null_object) ? {:null_object => options.delete(:null_object)} : {} end def assign_stubs(stubs) stubs.each_pair do |message, response| stub!(message).and_return(response) end end end end end