Sha256: 069c5ab84d6b9fb6cb4424c0f99538f17b0db5a129776a6cffec519ab942cc71
Contents?: true
Size: 919 Bytes
Versions: 9
Compression:
Stored size: 919 Bytes
Contents
module DatabaseCleaner
module Mongoid
module Mongo2TruncationMixin
def clean
if @only
collections.each { |c| database[c].find.delete_many if @only.include?(c) }
else
collections.each { |c| database[c].find.delete_many unless @tables_to_exclude.include?(c) }
end
true
end
private
def database
if @db.nil? || @db == :default
::Mongoid::Clients.default
else
::Mongoid::Clients.with_name(@db)
end
end
def collections
if db != :default
database.use(db)
end
database.collections.collect { |c| c.namespace.split('.',2)[1] }
# database['system.namespaces'].find(:name => { '$not' => /\.system\.|\$/ }).to_a.map do |collection|
# _, name = collection['name'].split('.', 2)
# name
# end
end
end
end
end
Version data entries
9 entries across 9 versions & 2 rubygems