database_cleaner 1.6.2 → 2.0.0

data/spec/database_cleaner/active_record/transaction_spec.rb

@@ -1,176 +0,0 @@
-require File.dirname(__FILE__) + '/../../spec_helper'
-require 'database_cleaner/active_record/transaction'
-require 'active_record'
-
-module DatabaseCleaner
-  module ActiveRecord
-
-    describe Transaction do
-      let (:connection) { double("connection") }
-      let (:connection_2) { double("connection") }
-      let (:connection_pool) { double("connection_pool")}
-      before(:each) do
-        ::ActiveRecord::Base.stub(:connection_pool).and_return(connection_pool)
-        connection_pool.stub(:connections).and_return([connection])
-        ::ActiveRecord::Base.stub(:connection).and_return(connection)
-      end
-
-      describe "#start" do
-        [:begin_transaction, :begin_db_transaction].each do |begin_transaction_method|
-          context "using #{begin_transaction_method}" do
-            before do
-              connection.stub(:transaction)
-              connection.stub(begin_transaction_method)
-              connection.stub(:respond_to?).with(:begin_transaction).and_return(:begin_transaction == begin_transaction_method)
-            end
-
-            it "should increment open transactions if possible" do
-              connection.stub(:respond_to?).with(:increment_open_transactions).and_return(true)
-              connection.should_receive(:increment_open_transactions)
-              Transaction.new.start
-            end
-
-            it "should tell ActiveRecord to increment connection if its not possible to increment current connection" do
-              connection.stub(:respond_to?).with(:increment_open_transactions).and_return(false)
-              ::ActiveRecord::Base.should_receive(:increment_open_transactions)
-              Transaction.new.start
-            end
-
-            it "should start a transaction" do
-              connection.stub(:respond_to?).with(:increment_open_transactions).and_return(true)
-              connection.stub(:increment_open_transactions)
-              connection.should_receive(begin_transaction_method)
-              connection.should_receive(:transaction)
-              Transaction.new.start
-            end
-          end
-        end
-      end
-
-      describe "#clean" do
-        context "manual accounting of transaction count" do
-          it "should start a transaction" do
-            connection.should_receive(:open_transactions).and_return(1)
-
-            connection.stub(:decrement_open_transactions)
-
-            connection.should_receive(:rollback_db_transaction)
-            Transaction.new.clean
-          end
-
-          it "should decrement open transactions if possible" do
-            connection.should_receive(:open_transactions).and_return(1)
-
-            connection.stub(:respond_to?).with(:decrement_open_transactions).and_return(true)
-            connection.stub(:respond_to?).with(:rollback_transaction_records, true).and_return(false)
-            connection.stub(:respond_to?).with(:rollback_transaction).and_return(false)
-            connection.stub(:rollback_db_transaction)
-
-            connection.should_receive(:decrement_open_transactions)
-            Transaction.new.clean
-          end
-
-          it "should not try to decrement or rollback if open_transactions is 0 for whatever reason" do
-            connection.should_receive(:open_transactions).and_return(0)
-
-            Transaction.new.clean
-          end
-
-          it "should decrement connection via ActiveRecord::Base if connection won't" do
-            connection.should_receive(:open_transactions).and_return(1)
-            connection.stub(:respond_to?).with(:decrement_open_transactions).and_return(false)
-            connection.stub(:respond_to?).with(:rollback_transaction_records, true).and_return(false)
-            connection.stub(:respond_to?).with(:rollback_transaction).and_return(false)
-            connection.stub(:rollback_db_transaction)
-
-            ::ActiveRecord::Base.should_receive(:decrement_open_transactions)
-            Transaction.new.clean
-          end
-
-          it "should rollback open transactions in all connections" do
-            connection_pool.stub(:connections).and_return([connection, connection_2])
-
-            connection.should_receive(:open_transactions).and_return(1)
-            connection.stub(:respond_to?).with(:decrement_open_transactions).and_return(false)
-            connection.stub(:respond_to?).with(:rollback_transaction_records, true).and_return(false)
-            connection.stub(:respond_to?).with(:rollback_transaction).and_return(false)
-            connection.stub(:rollback_db_transaction)
-
-            connection_2.should_receive(:open_transactions).and_return(1)
-            connection_2.stub(:respond_to?).with(:decrement_open_transactions).and_return(false)
-            connection_2.stub(:respond_to?).with(:rollback_transaction_records, true).and_return(false)
-            connection_2.stub(:respond_to?).with(:rollback_transaction).and_return(false)
-            connection_2.stub(:rollback_db_transaction)
-
-            ::ActiveRecord::Base.should_receive(:decrement_open_transactions).twice
-            Transaction.new.clean
-          end
-
-          it "should rollback open transactions in all connections with an open transaction" do
-            connection_pool.stub(:connections).and_return([connection, connection_2])
-
-            connection.should_receive(:open_transactions).and_return(1)
-            connection.stub(:respond_to?).with(:decrement_open_transactions).and_return(false)
-            connection.stub(:respond_to?).with(:rollback_transaction_records, true).and_return(false)
-            connection.stub(:respond_to?).with(:rollback_transaction).and_return(false)
-            connection.stub(:rollback_db_transaction)
-
-            connection_2.should_receive(:open_transactions).and_return(0)
-
-            ::ActiveRecord::Base.should_receive(:decrement_open_transactions).exactly(1).times
-            Transaction.new.clean
-          end
-        end
-
-        context "automatic accounting of transaction count (AR 4)" do
-          before {stub_const("ActiveRecord::VERSION::MAJOR", 4) }
-
-          it "should start a transaction" do
-            connection.stub(:rollback_db_transaction)
-            connection.should_receive(:open_transactions).and_return(1)
-
-            connection.should_not_receive(:decrement_open_transactions)
-            connection.should_receive(:rollback_transaction)
-            Transaction.new.clean
-          end
-
-          it "should decrement open transactions if possible" do
-            connection.stub(:rollback_transaction)
-            connection.should_receive(:open_transactions).and_return(1)
-
-            connection.should_not_receive(:decrement_open_transactions)
-            Transaction.new.clean
-          end
-
-          it "should not try to decrement or rollback if open_transactions is 0 for whatever reason" do
-            connection.should_receive(:open_transactions).and_return(0)
-
-            Transaction.new.clean
-          end
-
-          it "should decrement connection via ActiveRecord::Base if connection won't" do
-            connection.should_receive(:open_transactions).and_return(1)
-            connection.stub(:respond_to?).with(:rollback_transaction_records, true).and_return(false)
-            connection.stub(:respond_to?).with(:rollback_transaction).and_return(true)
-            connection.stub(:rollback_transaction)
-
-            ::ActiveRecord::Base.should_not_receive(:decrement_open_transactions)
-            Transaction.new.clean
-          end
-        end
-      end
-
-      describe "#connection_maintains_transaction_count?" do
-        it "should return true if the major active record version is < 4" do
-          stub_const("ActiveRecord::VERSION::MAJOR", 3)
-          Transaction.new.connection_maintains_transaction_count?.should be_true
-        end
-        it "should return false if the major active record version is > 3" do
-          stub_const("ActiveRecord::VERSION::MAJOR", 4)
-          Transaction.new.connection_maintains_transaction_count?.should be_false
-        end
-      end
-
-    end
-  end
-end

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

@@ -1,38 +0,0 @@
-require 'spec_helper'
-require 'active_record'
-require 'support/active_record/mysql2_setup'
-require 'database_cleaner/active_record/truncation'
-require 'database_cleaner/active_record/truncation/shared_fast_truncation'
-
-module ActiveRecord
-  module ConnectionAdapters
-    describe do
-      before(:all) { active_record_mysql2_setup }
-
-      let(:connection) { active_record_mysql2_connection }
-
-      describe "#truncate_table" do
-        it "should truncate the table" do
-          2.times { User.create }
-
-          connection.truncate_table('users')
-          User.count.should eq 0
-        end
-
-        it "should reset AUTO_INCREMENT index of table" do
-          2.times { User.create }
-          User.delete_all
-
-          connection.truncate_table('users')
-
-          User.create.id.should eq 1
-        end
-      end
-
-      it_behaves_like "an adapter with pre-count truncation" do
-        let(:connection) { active_record_mysql2_connection }
-      end
-    end
-  end
-end
-

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

@@ -1,38 +0,0 @@
-require 'spec_helper'
-require 'active_record'
-require 'support/active_record/mysql_setup'
-require 'database_cleaner/active_record/truncation'
-require 'database_cleaner/active_record/truncation/shared_fast_truncation'
-
-module ActiveRecord
-  module ConnectionAdapters
-    describe do
-      before(:all) { active_record_mysql_setup }
-
-      let(:connection) { active_record_mysql_connection }
-
-      describe "#truncate_table" do
-        it "should truncate the table" do
-          2.times { User.create }
-
-          connection.truncate_table('users')
-          User.count.should eq 0
-        end
-
-        it "should reset AUTO_INCREMENT index of table" do
-          2.times { User.create }
-          User.delete_all
-
-          connection.truncate_table('users')
-
-          User.create.id.should eq 1
-        end
-      end
-
-      it_behaves_like "an adapter with pre-count truncation" do
-        let(:connection) { active_record_mysql_connection }
-      end
-    end
-  end
-end
-

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

@@ -1,75 +0,0 @@
-require 'spec_helper'
-require 'active_record'
-require 'support/active_record/postgresql_setup'
-require 'database_cleaner/active_record/truncation'
-require 'database_cleaner/active_record/truncation/shared_fast_truncation'
-
-module ActiveRecord
-  module ConnectionAdapters
-    describe "schema_migrations table" do
-      it "is not truncated" do
-        active_record_pg_migrate
-        DatabaseCleaner::ActiveRecord::Truncation.new.clean
-        result = active_record_pg_connection.execute("select count(*) from schema_migrations;")
-        result.values.first.should eq ["2"]
-      end
-    end
-
-    describe do
-      before(:all) { active_record_pg_setup }
-
-      let(:connection) do
-        active_record_pg_connection
-      end
-
-      before(:each) do
-        connection.truncate_tables connection.tables
-      end
-
-      describe "#truncate_table" do
-        it "truncates the table" do
-          2.times { User.create }
-
-          connection.truncate_table('users')
-          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
-
-          connection.truncate_table('users')
-
-          User.create.id.should eq 1
-        end
-      end
-
-      describe ":except option cleanup" do
-        it "should not truncate the tables specified in the :except option" do
-          2.times { User.create }
-
-          ::DatabaseCleaner::ActiveRecord::Truncation.new(:except => ['users']).clean
-          expect( User.count ).to eq 2
-        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(:connection) { active_record_pg_connection }
-      end
-
-    end
-  end
-end

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

@@ -1,40 +0,0 @@
-shared_examples_for "an adapter with pre-count truncation" do
-  describe "#pre_count_truncate_tables" do
-    
-    context "with :reset_ids set true" do
-      it "truncates the table" do
-        2.times { User.create }
-
-        connection.pre_count_truncate_tables(%w[users], :reset_ids => true)
-        User.count.should be_zero
-      end
-
-      it "resets AUTO_INCREMENT index of table" do
-        2.times { User.create }
-        User.delete_all
-
-        connection.pre_count_truncate_tables(%w[users]) # true is also the default
-        User.create.id.should eq 1
-      end
-    end
-
-    
-    context "with :reset_ids set false" do
-      it "truncates the table" do
-        2.times { User.create }
-
-        connection.pre_count_truncate_tables(%w[users], :reset_ids => false)
-        User.count.should be_zero
-      end
-
-      it "does not reset AUTO_INCREMENT index of table" do
-        2.times { User.create }
-        User.delete_all
-
-        connection.pre_count_truncate_tables(%w[users], :reset_ids => false)
-
-        User.create.id.should eq 3
-      end
-    end
-  end
-end

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

@@ -1,40 +0,0 @@
-require 'spec_helper'
-require 'active_record'
-require 'support/active_record/sqlite3_setup'
-require 'database_cleaner/active_record/truncation'
-
-module ActiveRecord
-  module ConnectionAdapters
-    describe do
-      before(:all) { active_record_sqlite3_setup }
-
-      let(:connection) do
-        active_record_sqlite3_connection
-      end
-
-      before(:each) do
-        connection.truncate_tables connection.tables
-      end
-
-      describe "#truncate_table" do
-        it "truncates the table" do
-          2.times { User.create }
-
-          connection.truncate_table('users')
-          User.count.should eq 0
-        end
-
-        it "resets AUTO_INCREMENT index of table" do
-          2.times { User.create }
-          User.delete_all
-
-          connection.truncate_table('users')
-
-          User.create.id.should eq 1
-        end
-      end
-
-    end
-  end
-end
-

data/spec/database_cleaner/active_record/truncation_spec.rb

@@ -1,180 +0,0 @@
-require File.dirname(__FILE__) + '/../../spec_helper'
-require 'active_record'
-require 'active_record/connection_adapters/mysql_adapter'
-require 'active_record/connection_adapters/mysql2_adapter'
-require 'active_record/connection_adapters/sqlite3_adapter'
-require 'active_record/connection_adapters/postgresql_adapter'
-
-require 'database_cleaner/active_record/truncation'
-
-module ActiveRecord
-  module ConnectionAdapters
-    #JdbcAdapter IBM_DBAdapter
-    [ MysqlAdapter, Mysql2Adapter, SQLite3Adapter, PostgreSQLAdapter ].each do |adapter|
-      describe adapter, "#truncate_table" do
-        it "responds" do
-          adapter.instance_methods.should include(:truncate_table)
-        end
-      end
-    end
-  end
-end
-
-module DatabaseCleaner
-  module ActiveRecord
-
-    describe Truncation do
-      let(:connection) { double('connection') }
-
-      before(:each) do
-        connection.stub(:disable_referential_integrity).and_yield
-        connection.stub(:database_cleaner_view_cache).and_return([])
-        ::ActiveRecord::Base.stub(:connection).and_return(connection)
-      end
-
-      describe '#clean' do
-        it "should truncate all tables except for schema_migrations" do
-          connection.stub(:database_cleaner_table_cache).and_return(%w[schema_migrations widgets dogs])
-
-          connection.should_receive(:truncate_tables).with(['widgets', 'dogs'])
-          Truncation.new.clean
-        end
-
-        it "should use ActiveRecord's SchemaMigration.table_name" do
-          connection.stub(:database_cleaner_table_cache).and_return(%w[pre_schema_migrations_suf widgets dogs])
-          ::ActiveRecord::Base.stub(:table_name_prefix).and_return('pre_')
-          ::ActiveRecord::Base.stub(:table_name_suffix).and_return('_suf')
-
-          connection.should_receive(:truncate_tables).with(['widgets', 'dogs'])
-
-          Truncation.new.clean
-        end
-
-        it "should only truncate the tables specified in the :only option when provided" do
-          connection.stub(:database_cleaner_table_cache).and_return(%w[schema_migrations widgets dogs])
-
-          connection.should_receive(:truncate_tables).with(['widgets'])
-
-          Truncation.new(:only => ['widgets']).clean
-        end
-
-        it "should not truncate the tables specified in the :except option" do
-          connection.stub(:database_cleaner_table_cache).and_return(%w[schema_migrations widgets dogs])
-
-          connection.should_receive(:truncate_tables).with(['dogs'])
-
-          Truncation.new(:except => ['widgets']).clean
-        end
-
-        it "should raise an error when :only and :except options are used" do
-          running {
-            Truncation.new(:except => ['widgets'], :only => ['widgets'])
-          }.should raise_error(ArgumentError)
-        end
-
-        it "should raise an error when invalid options are provided" do
-          running { Truncation.new(:foo => 'bar') }.should raise_error(ArgumentError)
-        end
-
-        it "should not truncate views" do
-          connection.stub(:database_cleaner_table_cache).and_return(%w[widgets dogs])
-          connection.stub(:database_cleaner_view_cache).and_return(["widgets"])
-
-          connection.should_receive(:truncate_tables).with(['dogs'])
-
-          Truncation.new.clean
-        end
-
-        describe "relying on #pre_count_truncate_tables if connection allows it" do
-          subject { Truncation.new }
-
-          it "should rely on #pre_count_truncate_tables if #pre_count? returns true" do
-            connection.stub(:database_cleaner_table_cache).and_return(%w[widgets dogs])
-            connection.stub(:database_cleaner_view_cache).and_return(["widgets"])
-
-            subject.instance_variable_set(:"@pre_count", true)
-
-            connection.should_not_receive(:truncate_tables).with(['dogs'])
-            connection.should_receive(:pre_count_truncate_tables).with(['dogs'], :reset_ids => true)
-
-            subject.clean
-          end
-
-          it "should not rely on #pre_count_truncate_tables if #pre_count? return false" do
-            connection.stub(:database_cleaner_table_cache).and_return(%w[widgets dogs])
-            connection.stub(:database_cleaner_view_cache).and_return(["widgets"])
-
-            subject.instance_variable_set(:"@pre_count", false)
-
-            connection.should_not_receive(:pre_count_truncate_tables).with(['dogs'], :reset_ids => true)
-            connection.should_receive(:truncate_tables).with(['dogs'])
-
-            subject.clean
-          end
-        end
-
-        context 'when :cache_tables is set to true' do
-          it 'caches the list of tables to be truncated' do
-            connection.should_receive(:database_cleaner_table_cache).and_return([])
-            connection.should_not_receive(:tables)
-
-            connection.stub(:truncate_tables)
-            Truncation.new({ :cache_tables => true }).clean
-          end
-        end
-
-        context 'when :cache_tables is set to false' do
-          it 'does not cache the list of tables to be truncated' do
-            connection.should_not_receive(:database_cleaner_table_cache)
-            connection.should_receive(:tables).and_return([])
-
-            connection.stub(:truncate_tables)
-            Truncation.new({ :cache_tables => false }).clean
-          end
-        end
-      end
-
-      describe '#pre_count?' do
-        before(:each) do
-          connection.stub(:disable_referential_integrity).and_yield
-          connection.stub(:database_cleaner_view_cache).and_return([])
-          ::ActiveRecord::Base.stub(:connection).and_return(connection)
-        end
-
-        subject { Truncation.new }
-        its(:pre_count?) { should eq false }
-
-        it 'should return true if @reset_id is set and non false or nil' do
-          subject.instance_variable_set(:"@pre_count", true)
-          subject.send(:pre_count?).should eq true
-        end
-
-        it 'should return false if @reset_id is set to false' do
-          subject.instance_variable_set(:"@pre_count", false)
-          subject.send(:pre_count?).should eq false
-        end
-      end
-
-      describe '#reset_ids?' do
-        before(:each) do
-          connection.stub(:disable_referential_integrity).and_yield
-          connection.stub(:database_cleaner_view_cache).and_return([])
-          ::ActiveRecord::Base.stub(:connection).and_return(connection)
-        end
-
-        subject { Truncation.new }
-        its(:reset_ids?) { should eq true }
-
-        it 'should return true if @reset_id is set and non false or nil' do
-          subject.instance_variable_set(:"@reset_ids", 'Something')
-          subject.send(:reset_ids?).should eq true
-        end
-
-        it 'should return false if @reset_id is set to false' do
-          subject.instance_variable_set(:"@reset_ids", false)
-          subject.send(:reset_ids?).should eq false
-        end
-      end
-    end
-  end
-end