Contents

require 'database_cleaner/mongo/base'
require 'database_cleaner/generic/truncation'
require 'database_cleaner/mongo/truncation_mixin'
module DatabaseCleaner
  module Mongo
    class Truncation
      include ::DatabaseCleaner::Generic::Truncation
      include TruncationMixin
      include Base
      private

      def database
        db
      end

      def collections_cache
        @@collections_cache ||= {}
      end

      def mongoid_collection_names
        @@mongoid_collection_names ||= Hash.new{|h,k| h[k]=[]}.tap do |names|
          ObjectSpace.each_object(Class) do |klass|
            next unless klass.ancestors.map(&:to_s).include?('Mongoid::Document')
            next if klass.embedded
            next if klass.collection_name.empty?
            names[klass.db.name] << klass.collection_name
          end
        end
      end

      def not_caching(db_name, list)
        @@not_caching ||= {}

        unless @@not_caching.has_key?(db_name)
          @@not_caching[db_name] = true

          puts "Not caching collection names for db #{db_name}. Missing these from models: #{list}"
        end
      end

      def collections
        return collections_cache[database.name] if collections_cache.has_key?(database.name)
        db_collections = database.collections.select { |c| c.name !~ /^system\./ }

        missing_collections = mongoid_collection_names[database.name] - db_collections.map(&:name)

        if missing_collections.empty?
          collections_cache[database.name] = db_collections
        else
          not_caching(database.name, missing_collections)
        end

        db_collections
      end
    end
  end
end

Version data entries

4 entries across 4 versions & 2 rubygems

Version	Path
solidus_backend-1.0.0.pre3	vendor/bundle/gems/database_cleaner-1.4.1/lib/database_cleaner/mongo/truncation.rb
solidus_backend-1.0.0.pre2	vendor/bundle/gems/database_cleaner-1.4.1/lib/database_cleaner/mongo/truncation.rb
solidus_backend-1.0.0.pre	vendor/bundle/gems/database_cleaner-1.4.1/lib/database_cleaner/mongo/truncation.rb
database_cleaner-1.4.1	lib/database_cleaner/mongo/truncation.rb