class RSpec::Mocks::MarshalExtension

# File lib/rspec/mocks/marshal_extension.rb, line 12
def dump_with_rspec_mocks(object, *rest)
  if !::RSpec::Mocks.space.registered?(object) || NilClass === object
    dump_without_rspec_mocks(object, *rest)
  else
    dump_without_rspec_mocks(object.dup, *rest)
  end
end

# File lib/rspec/mocks/marshal_extension.rb, line 7
def self.patch!
  return if Marshal.respond_to?(:dump_with_rspec_mocks)

  Marshal.instance_eval do
    class << self
      def dump_with_rspec_mocks(object, *rest)
        if !::RSpec::Mocks.space.registered?(object) || NilClass === object
          dump_without_rspec_mocks(object, *rest)
        else
          dump_without_rspec_mocks(object.dup, *rest)
        end
      end

      alias_method :dump_without_rspec_mocks, :dump
      undef_method :dump
      alias_method :dump, :dump_with_rspec_mocks
    end
  end
end

# File lib/rspec/mocks/marshal_extension.rb, line 27
def self.unpatch!
  return unless Marshal.respond_to?(:dump_with_rspec_mocks)

  Marshal.instance_eval do
    class << self
      undef_method :dump_with_rspec_mocks
      undef_method :dump
      alias_method :dump, :dump_without_rspec_mocks
      undef_method :dump_without_rspec_mocks
    end
  end
end