database_cleaner 1.3.0 → 1.4.0

data/lib/database_cleaner/mongoid/truncation.rb

@@ -2,6 +2,7 @@ require 'database_cleaner/mongoid/base'
 require 'database_cleaner/generic/truncation'
 require 'database_cleaner/mongo/truncation_mixin'
 require 'database_cleaner/moped/truncation_base'
+require 'mongoid/tree'
 require 'mongoid/version'
 
 module DatabaseCleaner
@@ -30,8 +31,14 @@ module DatabaseCleaner
           ::Mongoid.default_session
         end
 
+        def database
+          if not(@db.nil? or @db == :default)
+            ::Mongoid.databases[@db]
+          else
+            ::Mongoid.database
+          end
+        end
       end
-
     end
   end
 end

data/lib/database_cleaner/moped/truncation_base.rb

@@ -23,7 +23,7 @@ module DatabaseCleaner
           session.use(db)
         end
 
-        session['system.namespaces'].find(:name => { '$not' => /system|\$/ }).to_a.map do |collection|
+        session['system.namespaces'].find(:name => { '$not' => /\.system\.|\$/ }).to_a.map do |collection|
           _, name = collection['name'].split('.', 2)
           name
         end

data/lib/database_cleaner/neo4j/base.rb

@@ -0,0 +1,58 @@
+require 'database_cleaner/generic/base'
+module DatabaseCleaner
+  module Neo4j
+    def self.available_strategies
+      %w[transaction truncation deletion]
+    end
+
+    module Base
+      include ::DatabaseCleaner::Generic::Base
+
+      def db=(desired_db)
+        @db = desired_db == :default ? nil : desired_db
+      end
+
+      def db
+        @db ||= nil
+      end
+
+      def start
+        if db_type == :embedded_db and not session.running?
+          session.start
+        else
+          session
+        end
+      end
+
+      def database
+        db && default_db.merge(db) || default_db
+      end
+
+      private
+
+      def default_db
+        {:type => default_db_type, :path => default_db_path}
+      end
+
+      def default_db_type
+        :server_db
+      end
+
+      def default_db_path(type = default_db_type)
+        type == :server_db ? 'http://localhost:7475/' : './db/test'
+      end
+
+      def db_type
+        database[:type]
+      end
+
+      def db_path
+        database[:path]
+      end
+
+      def session
+        @session ||= ::Neo4j::Session.open(db_type, db_path)
+      end
+    end
+  end
+end

data/lib/database_cleaner/neo4j/deletion.rb

@@ -0,0 +1,16 @@
+require 'database_cleaner/neo4j/base'
+require 'neo4j-core'
+
+module DatabaseCleaner
+  module Neo4j
+    class Deletion
+      include ::DatabaseCleaner::Neo4j::Base
+
+      def clean
+        ::Neo4j::Transaction.run do
+          session._query('MATCH (n) OPTIONAL MATCH (n)-[r]-() DELETE n,r')
+        end
+      end
+    end
+  end
+end

data/lib/database_cleaner/neo4j/transaction.rb

@@ -0,0 +1,35 @@
+require 'database_cleaner/neo4j/base'
+require 'database_cleaner/generic/transaction'
+require 'neo4j-core'
+
+module DatabaseCleaner
+  module Neo4j
+    class Transaction
+      include ::DatabaseCleaner::Generic::Transaction
+      include ::DatabaseCleaner::Neo4j::Base
+
+      attr_accessor :tx
+
+      def start
+        super
+        rollback
+        self.tx = ::Neo4j::Transaction.new
+      end
+
+      def clean
+        rollback
+      end
+
+      private
+
+      def rollback
+        if tx
+          tx.failure
+          tx.close
+        end
+      ensure
+        self.tx = nil
+      end
+    end
+  end
+end

data/lib/database_cleaner/neo4j/truncation.rb

@@ -0,0 +1,9 @@
+require 'database_cleaner/neo4j/base'
+require 'database_cleaner/neo4j/deletion'
+
+module DatabaseCleaner
+  module Neo4j
+    class Truncation < DatabaseCleaner::Neo4j::Deletion
+    end
+  end
+end

data/lib/database_cleaner/sequel/base.rb

@@ -2,7 +2,7 @@ require 'database_cleaner/generic/base'
 module DatabaseCleaner
   module Sequel
     def self.available_strategies
-      %w[truncation transaction]
+      %w(truncation transaction deletion)
     end
 
     module Base
@@ -14,7 +14,7 @@ module DatabaseCleaner
 
       def db
         return @db if @db && @db != :default
-        raise "As you have more than one active sequel database you have to specify the one to use manually!" if ::Sequel::DATABASES.count > 1 
+        raise "As you have more than one active sequel database you have to specify the one to use manually!" if ::Sequel::DATABASES.count > 1
         ::Sequel::DATABASES.first || :default
       end
     end

data/lib/database_cleaner/sequel/deletion.rb

@@ -0,0 +1,47 @@
+require 'database_cleaner/sequel/base'
+require 'database_cleaner/generic/truncation'
+require 'database_cleaner/sequel/truncation'
+
+module DatabaseCleaner::Sequel
+  class Deletion < Truncation
+    def disable_referential_integrity(tables)
+      case db.database_type
+      when :postgres
+        db.run('SET CONSTRAINTS ALL DEFERRED')
+        tables_to_truncate(db).each do |table|
+          db.run("ALTER TABLE \"#{table}\" DISABLE TRIGGER ALL")
+        end
+      when :mysql
+        old = db.fetch('SELECT @@FOREIGN_KEY_CHECKS').first[:@@FOREIGN_KEY_CHECKS]
+        db.run('SET FOREIGN_KEY_CHECKS = 0')
+      end
+      yield
+    ensure
+      case db.database_type
+      when :postgres
+        tables.each do |table|
+          db.run("ALTER TABLE \"#{table}\" ENABLE TRIGGER ALL")
+        end
+      when :mysql
+        db.run("SET FOREIGN_KEY_CHECKS = #{old}")
+      end
+    end
+
+    def delete_tables(db, tables)
+      tables.each do |table|
+        db[table.to_sym].delete
+      end
+    end
+
+    def clean
+      return unless dirty?
+
+      tables = tables_to_truncate(db)
+      db.transaction do
+        disable_referential_integrity(tables) do
+          delete_tables(db, tables)
+        end
+      end
+    end
+  end
+end

data/lib/database_cleaner/sequel/transaction.rb

@@ -16,19 +16,19 @@ module DatabaseCleaner
       def start
         self.class.check_fiber_brokenness
 
-        @fibers||= []
-        db= self.db
-        f= Fiber.new do
+        @fibers ||= []
+        db = self.db
+        f = Fiber.new do
           db.transaction(:rollback => :always, :savepoint => true) do
             Fiber.yield
           end
         end
         f.resume
-        @fibers<< f
+        @fibers << f
       end
 
       def clean
-        f= @fibers.pop
+        f = @fibers.pop
         f.resume
       end
 

data/lib/database_cleaner/sequel/truncation.rb

@@ -1,4 +1,4 @@
-require "database_cleaner/generic/truncation"
+require 'database_cleaner/generic/truncation'
 require 'database_cleaner/sequel/base'
 
 module DatabaseCleaner
@@ -7,32 +7,59 @@ module DatabaseCleaner
       include ::DatabaseCleaner::Sequel::Base
       include ::DatabaseCleaner::Generic::Truncation
 
+      def start
+        @last_txid = txid
+      end
+
       def clean
+        return unless dirty?
+
         case db.database_type
         when :postgres
           # PostgreSQL requires all tables with FKs to be truncates in the same command, or have the CASCADE keyword
           # appended. Bulk truncation without CASCADE is:
           # * Safer. Tables outside of tables_to_truncate won't be affected.
           # * Faster. Less roundtrips to the db.
-          unless (tables= tables_to_truncate(db)).empty?
-            all_tables= tables.map{|t| %["#{t}"]}.join ','
+          unless (tables = tables_to_truncate(db)).empty?
+            all_tables = tables.map { |t| %("#{t}") }.join(',')
             db.run "TRUNCATE TABLE #{all_tables};"
           end
         else
-          # Truncate each table normally
-          each_table do |db, table|
-            db[table].truncate
+          tables = tables_to_truncate(db)
+
+          if pre_count?
+            # Count rows before truncating
+            pre_count_truncate_tables(db, tables)
+          else
+            # Truncate each table normally
+            truncate_tables(db, tables)
           end
         end
       end
 
-      def each_table
-        tables_to_truncate(db).each do |table|
-          yield db, table.to_sym
+      private
+
+      def pre_count_truncate_tables(db, tables)
+        tables = tables.reject { |table| db[table.to_sym].count == 0 }
+        truncate_tables(db, tables)
+      end
+
+      def truncate_tables(db, tables)
+        tables.each do |table|
+          db[table.to_sym].truncate
         end
       end
 
-      private
+      def dirty?
+        @last_txid != txid || @last_txid.nil?
+      end
+
+      def txid
+        case db.database_type
+        when :postgres
+          db.fetch('SELECT txid_snapshot_xmax(txid_current_snapshot()) AS txid').first[:txid]
+        end
+      end
 
       def tables_to_truncate(db)
         (@only || db.tables.map(&:to_s)) - @tables_to_exclude
@@ -40,11 +67,12 @@ module DatabaseCleaner
 
       # overwritten
       def migration_storage_names
-        %w[schema_info schema_migrations]
+        %w(schema_info schema_migrations)
       end
 
+      def pre_count?
+        @pre_count == true
+      end
     end
   end
 end
-
-

data/spec/database_cleaner/active_record/truncation/mysql2_spec.rb

@@ -6,10 +6,9 @@ require 'database_cleaner/active_record/truncation/shared_fast_truncation'
 
 module ActiveRecord
   module ConnectionAdapters
-    describe do 
+    describe do
       before(:all) { active_record_mysql2_setup }
 
-      let(:adapter) { Mysql2Adapter }
       let(:connection) { active_record_mysql2_connection }
 
       describe "#truncate_table" do
@@ -31,7 +30,6 @@ module ActiveRecord
       end
 
       it_behaves_like "an adapter with pre-count truncation" do
-        let(:adapter) { Mysql2Adapter }
         let(:connection) { active_record_mysql2_connection }
       end
     end

data/spec/database_cleaner/active_record/truncation/mysql_spec.rb

@@ -6,10 +6,9 @@ require 'database_cleaner/active_record/truncation/shared_fast_truncation'
 
 module ActiveRecord
   module ConnectionAdapters
-    describe do 
+    describe do
       before(:all) { active_record_mysql_setup }
 
-      let(:adapter) { MysqlAdapter }
       let(:connection) { active_record_mysql_connection }
 
       describe "#truncate_table" do
@@ -29,9 +28,8 @@ module ActiveRecord
           User.create.id.should eq 1
         end
       end
- 
+
       it_behaves_like "an adapter with pre-count truncation" do
-        let(:adapter) { MysqlAdapter }
         let(:connection) { active_record_mysql_connection }
       end
     end

data/spec/database_cleaner/active_record/truncation/postgresql_spec.rb

@@ -9,8 +9,6 @@ module ActiveRecord
     describe do
       before(:all) { active_record_pg_setup }
 
-      let(:adapter) { PostgreSQLAdapter }
-
       let(:connection) do
         active_record_pg_connection
       end
@@ -27,6 +25,13 @@ module ActiveRecord
           User.count.should eq 0
         end
 
+        it "truncates the table without id sequence" do
+          2.times { Agent.create }
+
+          connection.truncate_table('agents')
+          Agent.count.should eq 0
+        end
+
         it "resets AUTO_INCREMENT index of table" do
           2.times { User.create }
           User.delete_all
@@ -37,12 +42,16 @@ module ActiveRecord
         end
       end
 
+      describe '#database_cleaner_table_cache' do
+        it 'should default to the list of tables with their schema' do
+          connection.database_cleaner_table_cache.first.should match(/^public\./)
+        end
+      end
+
       it_behaves_like "an adapter with pre-count truncation" do
-        let(:adapter) { PostgreSQLAdapter }
         let(:connection) { active_record_pg_connection }
       end
 
     end
   end
 end
-

data/spec/database_cleaner/active_record/truncation/sqlite3_spec.rb

@@ -8,8 +8,6 @@ module ActiveRecord
     describe do
       before(:all) { active_record_sqlite3_setup }
 
-      let(:adapter) { SQLite3Adapter }
-
       let(:connection) do
         active_record_sqlite3_connection
       end

data/spec/database_cleaner/base_spec.rb

@@ -4,6 +4,7 @@ require 'database_cleaner/data_mapper/transaction'
 require 'database_cleaner/mongo_mapper/truncation'
 require 'database_cleaner/mongoid/truncation'
 require 'database_cleaner/couch_potato/truncation'
+require 'database_cleaner/neo4j/transaction'
 
 module DatabaseCleaner
   describe Base do
@@ -21,6 +22,7 @@ module DatabaseCleaner
          Temp_MP = ::Moped        if defined?(::Moped)        and not defined?(Temp_MP)
          Temp_RS = ::Redis        if defined?(::Redis)        and not defined?(Temp_RS)
          Temp_OH = ::Ohm          if defined?(::Ohm)          and not defined?(Temp_OH)
+         Temp_NJ = ::Neo4j        if defined?(::Neo4j)        and not defined?(Temp_NJ)
        end
 
        #Remove all ORM mocks and restore from cache
@@ -34,6 +36,7 @@ module DatabaseCleaner
          Object.send(:remove_const, 'Moped')        if defined?(::Moped)
          Object.send(:remove_const, 'Ohm')          if defined?(::Ohm)
          Object.send(:remove_const, 'Redis')        if defined?(::Redis)
+         Object.send(:remove_const, 'Neo4j')        if defined?(::Neo4j)
 
 
          # Restore ORMs
@@ -46,6 +49,7 @@ module DatabaseCleaner
          ::Moped        = Temp_MP if defined? Temp_MP
          ::Ohm          = Temp_OH if defined? Temp_OH
          ::Redis        = Temp_RS if defined? Temp_RS
+         ::Neo4j        = Temp_NJ if defined? Temp_NJ
        end
 
        #reset the orm mocks
@@ -59,6 +63,7 @@ module DatabaseCleaner
          Object.send(:remove_const, 'Moped')        if defined?(::Moped)
          Object.send(:remove_const, 'Ohm')          if defined?(::Ohm)
          Object.send(:remove_const, 'Redis')        if defined?(::Redis)
+         Object.send(:remove_const, 'Neo4j')        if defined?(::Neo4j)
        end
 
        let(:cleaner) { DatabaseCleaner::Base.new :autodetect }
@@ -77,6 +82,7 @@ module DatabaseCleaner
          Object.const_set('Moped',       'Moped mock')
          Object.const_set('Ohm',         'Ohm mock')
          Object.const_set('Redis',       'Redis mock')
+         Object.const_set('Neo4j',       'Neo4j mock')
 
          cleaner.orm.should eq :active_record
          cleaner.should be_auto_detected
@@ -91,6 +97,7 @@ module DatabaseCleaner
          Object.const_set('Moped',       'Moped mock')
          Object.const_set('Ohm',         'Ohm mock')
          Object.const_set('Redis',       'Redis mock')
+         Object.const_set('Neo4j',       'Neo4j mock')
 
          cleaner.orm.should eq :data_mapper
          cleaner.should be_auto_detected
@@ -104,6 +111,7 @@ module DatabaseCleaner
          Object.const_set('Moped',       'Moped mock')
          Object.const_set('Ohm',         'Ohm mock')
          Object.const_set('Redis',       'Redis mock')
+         Object.const_set('Neo4j',       'Neo4j mock')
 
          cleaner.orm.should eq :mongo_mapper
          cleaner.should be_auto_detected
@@ -116,6 +124,7 @@ module DatabaseCleaner
          Object.const_set('Moped',       'Moped mock')
          Object.const_set('Ohm',         'Ohm mock')
          Object.const_set('Redis',       'Redis mock')
+         Object.const_set('Neo4j',       'Neo4j mock')
 
          cleaner.orm.should eq :mongoid
          cleaner.should be_auto_detected
@@ -127,6 +136,7 @@ module DatabaseCleaner
          Object.const_set('Moped',       'Moped mock')
          Object.const_set('Ohm',         'Ohm mock')
          Object.const_set('Redis',       'Redis mock')
+         Object.const_set('Neo4j',       'Neo4j mock')
 
          cleaner.orm.should eq :couch_potato
          cleaner.should be_auto_detected
@@ -137,30 +147,40 @@ module DatabaseCleaner
          Object.const_set('Moped',  'Moped mock')
          Object.const_set('Ohm',    'Ohm mock')
          Object.const_set('Redis',  'Redis mock')
+         Object.const_set('Neo4j',  'Neo4j mock')
 
          cleaner.orm.should eq :sequel
          cleaner.should be_auto_detected
        end
 
-       it 'detects Ohm seventh' do
-         Object.const_set('Ohm', 'Ohm mock')
-         Object.const_set('Redis',       'Redis mock')
+       it 'detects Moped seventh' do
+         Object.const_set('Moped', 'Moped mock')
+
+         cleaner.orm.should eq :moped
+         cleaner.should be_auto_detected
+       end
+
+       it 'detects Ohm eighth' do
+         Object.const_set('Ohm',    'Ohm mock')
+         Object.const_set('Redis',  'Redis mock')
+         Object.const_set('Neo4j',  'Neo4j mock')
 
          cleaner.orm.should eq :ohm
          cleaner.should be_auto_detected
        end
 
-       it 'detects Redis last' do
-         Object.const_set('Redis',       'Redis mock')
+       it 'detects Redis ninth' do
+         Object.const_set('Redis', 'Redis mock')
+         Object.const_set('Neo4j', 'Neo4j mock')
 
          cleaner.orm.should eq :redis
          cleaner.should be_auto_detected
        end
 
-       it 'detects Moped seventh' do
-         Object.const_set('Moped', 'Moped mock')
+       it 'detects Neo4j tenth' do
+         Object.const_set('Neo4j', 'Neo4j mock')
 
-         cleaner.orm.should eq :moped
+         cleaner.orm.should eq :neo4j
          cleaner.should be_auto_detected
        end
     end
@@ -181,7 +201,8 @@ module DatabaseCleaner
 
     describe "comparison" do
       it "should be equal if orm, connection and strategy are the same" do
-        strategy = double("strategy")
+        strategy = mock("strategy")
+        strategy.stub!(:to_ary => [strategy])
 
         one = DatabaseCleaner::Base.new(:active_record,:connection => :default)
         one.strategy = strategy
@@ -248,7 +269,11 @@ module DatabaseCleaner
     end
 
     describe "strategy_db=" do
-      let(:strategy) { double("strategy") }
+      let(:strategy) {
+        mock("strategy").tap{|strategy|
+          strategy.stub!(:to_ary => [strategy])
+        }
+      }
 
       before(:each) do
         subject.strategy = strategy
@@ -348,7 +373,11 @@ module DatabaseCleaner
     end
 
     describe "strategy=" do
-      let(:mock_strategy) { double("strategy") }
+      let(:mock_strategy) {
+        mock("strategy").tap{|strategy|
+        strategy.stub!(:to_ary => [strategy])
+      }
+      }
 
       it "should proxy symbolised strategies to create_strategy" do
         subject.should_receive(:create_strategy).with(:symbol)
@@ -388,7 +417,9 @@ module DatabaseCleaner
       end
 
       it "returns the set strategy" do
-        strategum = double("strategy")
+        strategum = mock("strategy").tap{|strategy|
+          strategy.stub!(:to_ary => [strategy])
+        }
         subject.strategy = strategum
         subject.strategy.should eq strategum
       end
@@ -562,6 +593,11 @@ module DatabaseCleaner
         cleaner = DatabaseCleaner::Base.new(:redis)
         cleaner.strategy.should be_instance_of DatabaseCleaner::Redis::Truncation
       end
+
+      it 'sets strategy to :transaction for Neo4j' do
+        cleaner = DatabaseCleaner::Base.new(:neo4j)
+        cleaner.strategy.should be_instance_of DatabaseCleaner::Neo4j::Transaction
+      end
     end
 
   end