# File lib/mocha/any_instance_method.rb, line 36 def define_new_method definition_target.class_eval(" def #{method}(*args, &block) self.class.any_instance.mocha.method_missing(:#{method}, *args, &block) end ", __FILE__, __LINE__ + 1) if @original_visibility Module.instance_method(@original_visibility).bind(definition_target).call(method) end end
# File lib/mocha/any_instance_method.rb, line 15 def hide_original_method if method_exists?(method) begin @original_method = stubbee.instance_method(method) if @original_method && @original_method.owner == stubbee @original_visibility = :public if stubbee.protected_instance_methods.include?(method) @original_visibility = :protected elsif stubbee.private_instance_methods.include?(method) @original_visibility = :private end stubbee.send(:remove_method, method) end include_prepended_module if RUBY_VERSION >= '2.0' rescue NameError # deal with nasties like ActiveRecord::Associations::AssociationProxy end end end
# File lib/mocha/any_instance_method.rb, line 58 def method_exists?(method) return true if stubbee.public_instance_methods(false).include?(method) return true if stubbee.protected_instance_methods(false).include?(method) return true if stubbee.private_instance_methods(false).include?(method) return false end
# File lib/mocha/any_instance_method.rb, line 7 def mock stubbee.any_instance.mocha end
# File lib/mocha/any_instance_method.rb, line 47 def remove_new_method definition_target.send(:remove_method, method) end
# File lib/mocha/any_instance_method.rb, line 11 def reset_mocha stubbee.any_instance.reset_mocha end
# File lib/mocha/any_instance_method.rb, line 51 def restore_original_method if @original_method && @original_method.owner == stubbee stubbee.send(:define_method, method, @original_method) Module.instance_method(@original_visibility).bind(stubbee).call(method) end end
# File lib/mocha/any_instance_method.rb, line 78 def definition_target @definition_target ||= stubbee end
# File lib/mocha/any_instance_method.rb, line 67 def include_prepended_module possible_prepended_modules = stubbee.ancestors.take_while do |mod| !(Class === mod) end if possible_prepended_modules.any? @definition_target = PrependedModule.new stubbee.__send__ :prepend, @definition_target end end