database_cleaner 1.7.0 → 2.0.1

data/lib/database_cleaner/generic/base.rb

@@ -1,29 +0,0 @@
-module ::DatabaseCleaner
-  module Generic
-    module Base
-
-      def self.included(base)
-        base.extend(ClassMethods)
-      end
-
-      def db
-        :default
-      end
-
-      def cleaning(&block)
-        begin
-          start
-          yield
-        ensure
-          clean
-        end
-      end
-
-      module ClassMethods
-        def available_strategies
-          %W[]
-        end
-      end
-    end
-  end
-end

data/lib/database_cleaner/generic/transaction.rb

@@ -1,11 +0,0 @@
-module DatabaseCleaner
-  module Generic
-    module Transaction
-      def initialize(opts = {})
-        if !opts.empty?
-          raise ArgumentError, "Options are not available for transaction strategies."
-        end
-      end
-    end
-  end
-end

data/lib/database_cleaner/generic/truncation.rb

@@ -1,40 +0,0 @@
-module DatabaseCleaner
-  module Generic
-    module Truncation
-      def initialize(opts={})
-        if !opts.empty? && !(opts.keys - [:only, :except, :pre_count, :reset_ids, :cache_tables]).empty?
-          raise ArgumentError, "The only valid options are :only, :except, :pre_count, :reset_ids or :cache_tables. You specified #{opts.keys.join(',')}."
-        end
-        if opts.has_key?(:only) && opts.has_key?(:except)
-          raise ArgumentError, "You may only specify either :only or :except.  Doing both doesn't really make sense does it?"
-        end
-
-        @only = opts[:only]
-        @tables_to_exclude = Array( (opts[:except] || []).dup ).flatten
-        @tables_to_exclude += migration_storage_names
-        @pre_count = opts[:pre_count]
-        @reset_ids = opts[:reset_ids]
-        @cache_tables = opts.has_key?(:cache_tables) ? !!opts[:cache_tables] : true
-      end
-
-      def start
-        #included for compatability reasons, do nothing if you don't need to
-      end
-
-      def clean
-        raise NotImplementedError
-      end
-
-      private
-      def tables_to_truncate
-        raise NotImplementedError
-      end
-
-      # overwrite in subclasses
-      # default implementation given because migration storage need not be present
-      def migration_storage_names
-        %w[]
-      end
-    end
-  end
-end

data/lib/database_cleaner/mongo/base.rb

@@ -1,16 +0,0 @@
-module DatabaseCleaner
-  module Mongo
-    def self.available_strategies
-      %w[truncation]
-    end
-    module Base
-      def db=(desired_db)
-        @db = desired_db
-      end
-
-      def db
-        @db || raise("You have not specified a database.  (see Mongo::Database)")
-      end
-    end
-  end
-end

data/lib/database_cleaner/mongo/truncation.rb

@@ -1,62 +0,0 @@
-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|
-            (names[klass.db.name] << klass.collection_name) if valid_collection_name?(klass)
-          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
-
-      private
-
-      def valid_collection_name?(klass)
-        klass.ancestors.map(&:to_s).include?('Mongoid::Document') &&
-        !klass.embedded &&
-        !klass.collection_name.empty?
-      end
-    end
-  end
-end

data/lib/database_cleaner/mongo/truncation_mixin.rb

@@ -1,26 +0,0 @@
-module DatabaseCleaner
-  module Mongo
-    module TruncationMixin
-
-      def clean
-        if @only
-          collections.each { |c| c.send(truncate_method_name) if @only.include?(c.name) }
-        else
-          collections.each { |c| c.send(truncate_method_name) unless @tables_to_exclude.include?(c.name) }
-        end
-        true
-      end
-
-      private
-
-      def collections
-        database.collections.select { |c| c.name !~ /^system\./ }
-      end
-
-      def truncate_method_name
-        # This constant only exists in the 2.x series.
-        defined?(::Mongo::VERSION) ? :delete_many : :remove
-      end
-    end
-  end
-end

data/lib/database_cleaner/mongo2/base.rb

@@ -1,16 +0,0 @@
-module DatabaseCleaner
-  module Mongo2
-    def self.available_strategies
-      %w[truncation]
-    end
-    module Base
-      def db=(desired_db)
-        @db = desired_db
-      end
-
-      def db
-        @db || raise("You have not specified a database.  (see Mongo2::Database)")
-      end
-    end
-  end
-end

data/lib/database_cleaner/mongo2/truncation_mixin.rb

@@ -1,39 +0,0 @@
-module DatabaseCleaner
-  module Mongo2
-    module TruncationMixin
-
-      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

data/lib/database_cleaner/mongo_mapper/base.rb

@@ -1,20 +0,0 @@
-require 'database_cleaner/generic/base'
-module DatabaseCleaner
-  module MongoMapper
-    def self.available_strategies
-      %w[truncation]
-    end
-
-    module Base
-      include ::DatabaseCleaner::Generic::Base
-
-      def db=(desired_db)
-        @db = desired_db
-      end
-
-      def db
-        @db ||= :default
-      end
-    end
-  end
-end

data/lib/database_cleaner/mongo_mapper/truncation.rb

@@ -1,19 +0,0 @@
-require 'database_cleaner/mongo_mapper/base'
-require 'database_cleaner/generic/truncation'
-require 'database_cleaner/mongo/truncation_mixin'
-
-module DatabaseCleaner
-  module MongoMapper
-    class Truncation
-      include ::DatabaseCleaner::MongoMapper::Base
-      include ::DatabaseCleaner::Generic::Truncation
-      include ::DatabaseCleaner::Mongo::TruncationMixin
-      
-      private
-
-      def database
-        ::MongoMapper.database
-      end
-    end
-  end
-end

data/lib/database_cleaner/mongoid/base.rb

@@ -1,20 +0,0 @@
-require 'database_cleaner/generic/base'
-module DatabaseCleaner
-  module Mongoid
-    def self.available_strategies
-      %w[truncation]
-    end
-
-    module Base
-      include ::DatabaseCleaner::Generic::Base
-
-      def db=(desired_db)
-        @db = desired_db
-      end
-
-      def db
-        @db || :default
-      end
-    end
-  end
-end

data/lib/database_cleaner/mongoid/truncation.rb

@@ -1,49 +0,0 @@
-require 'database_cleaner/mongoid/base'
-require 'database_cleaner/generic/truncation'
-require 'database_cleaner/mongo/truncation_mixin'
-require 'database_cleaner/mongo2/truncation_mixin'
-require 'database_cleaner/moped/truncation_base'
-require 'mongoid/version'
-
-module DatabaseCleaner
-  module Mongoid
-    class Truncation
-      include ::DatabaseCleaner::Mongoid::Base
-      include ::DatabaseCleaner::Generic::Truncation
-
-      if ::Mongoid::VERSION < '3'
-
-        include ::DatabaseCleaner::Mongo::TruncationMixin
-
-        private
-
-        def database
-          ::Mongoid.database
-        end
-
-      elsif ::Mongoid::VERSION < '5'
-
-        include ::DatabaseCleaner::Moped::TruncationBase
-
-        private
-
-        def session
-          ::Mongoid::VERSION > "5.0.0" ? ::Mongoid.default_client : ::Mongoid.default_session
-        end
-
-        def database
-          if not(@db.nil? or @db == :default)
-            ::Mongoid.databases[@db]
-          else
-            ::Mongoid.database
-          end
-        end
-
-      else
-
-        include ::DatabaseCleaner::Mongo2::TruncationMixin
-
-      end
-    end
-  end
-end

data/lib/database_cleaner/moped/base.rb

@@ -1,39 +0,0 @@
-require 'database_cleaner/generic/base'
-
-module DatabaseCleaner
-  module Moped
-    def self.available_strategies
-      %w[truncation]
-    end
-
-    module Base
-      include ::DatabaseCleaner::Generic::Base
-
-      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
-
-      private
-
-      def session
-        ::Moped::Session.new([host], database: db)
-      end
-    end
-  end
-end

data/lib/database_cleaner/moped/truncation.rb

@@ -1,9 +0,0 @@
-require 'database_cleaner/moped/truncation_base'
-
-module DatabaseCleaner
-  module Moped
-    class Truncation
-      include ::DatabaseCleaner::Moped::TruncationBase
-    end
-  end
-end

data/lib/database_cleaner/moped/truncation_base.rb

@@ -1,40 +0,0 @@
-require 'database_cleaner/moped/base'
-require 'database_cleaner/generic/truncation'
-
-module DatabaseCleaner
-  module Moped
-    module TruncationBase
-      include ::DatabaseCleaner::Moped::Base
-      include ::DatabaseCleaner::Generic::Truncation
-
-      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
-        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
-
-    end
-  end
-end