Contents

require 'database_cleaner/active_record/base'
require 'database_cleaner/generic/transaction'

module DatabaseCleaner::ActiveRecord
  class Transaction
    include ::DatabaseCleaner::ActiveRecord::Base
    include ::DatabaseCleaner::Generic::Transaction

    def start
      # Hack to make sure that the connection is properly setup for
      # the clean code.
      connection_class.connection.transaction{ }

      if connection_maintains_transaction_count?
        if connection_class.connection.respond_to?(:increment_open_transactions)
          connection_class.connection.increment_open_transactions
        else
          connection_class.__send__(:increment_open_transactions)
        end
      end
      if connection_class.connection.respond_to?(:begin_transaction)
        connection_class.connection.begin_transaction :joinable => false
      else
        connection_class.connection.begin_db_transaction
      end
    end


    def clean
      return unless connection_class.connection.open_transactions > 0

      if connection_class.connection.respond_to?(:rollback_transaction)
        connection_class.connection.rollback_transaction
      else
        connection_class.connection.rollback_db_transaction
      end

      # The below is for handling after_commit hooks.. see https://github.com/bmabey/database_cleaner/issues/99
      if connection_class.connection.respond_to?(:rollback_transaction_records, true)
        connection_class.connection.send(:rollback_transaction_records, true)
      end

      if connection_maintains_transaction_count?
        if connection_class.connection.respond_to?(:decrement_open_transactions)
          connection_class.connection.decrement_open_transactions
        else
          connection_class.__send__(:decrement_open_transactions)
        end
      end
    end

    def connection_maintains_transaction_count?
      ActiveRecord::VERSION::MAJOR < 4
    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/active_record/transaction.rb
solidus_backend-1.0.0.pre2	vendor/bundle/gems/database_cleaner-1.4.1/lib/database_cleaner/active_record/transaction.rb
solidus_backend-1.0.0.pre	vendor/bundle/gems/database_cleaner-1.4.1/lib/database_cleaner/active_record/transaction.rb
database_cleaner-1.4.1	lib/database_cleaner/active_record/transaction.rb