Sha256: 1552e1007be447c968893f7150ad0767e48922e3f207f7e4eaae616cedcc749c
Contents?: true
Size: 1.86 KB
Versions: 10
Compression:
Stored size: 1.86 KB
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
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)
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
10 entries across 5 versions & 2 rubygems