Sha256: aef8126300adb4523f8f1eb394bb300e9e73d65350f24d5a335dd1e6ef187fbb
Contents?: true
Size: 1.55 KB
Versions: 9
Compression:
Stored size: 1.55 KB
Contents
require 'database_cleaner/generic/base'
require 'database_cleaner/generic/truncation'
module DatabaseCleaner
module Mongoid
module MongoidTruncationMixin
include ::DatabaseCleaner::Generic::Base
include ::DatabaseCleaner::Generic::Truncation
def db=(desired_db)
@db = desired_db
end
def db
@db ||= :default
end
def host_port=(desired_host)
@host = desired_host
end
def host
@host ||= '127.0.0.1:27017'
end
def db_version
@db_version ||= session.command('buildinfo' => 1)['version']
end
def clean
if @only
collections.each { |c| session[c].find.remove_all if @only.include?(c) }
else
collections.each { |c| session[c].find.remove_all unless @tables_to_exclude.include?(c) }
end
wait_for_removals_to_finish
true
end
private
def collections
if db != :default
session.use(db)
end
if db_version.split('.').first.to_i >= 3
session.command(listCollections: 1, filter: { 'name' => { '$not' => /.?system\.|\$/ } })['cursor']['firstBatch'].map do |collection|
collection['name']
end
else
session['system.namespaces'].find(name: { '$not' => /\.system\.|\$/ }).to_a.map do |collection|
_, name = collection['name'].split('.', 2)
name
end
end
end
def wait_for_removals_to_finish
session.command(getlasterror: 1)
end
end
end
end
Version data entries
9 entries across 9 versions & 2 rubygems