class RSpec::Mocks::MarshalExtension

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

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

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