blob: f4946ff271f6f07baa2b820a11944eff5fe5de96 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
# frozen_string_literal: true
module TestProfBeforeAllAdapter
module MultipleDatabaseAdapter
def self.all_connection_classes
@all_connection_classes ||= [ActiveRecord::Base] + ActiveRecord::Base.descendants.select(&:connection_class?) # rubocop: disable Database/MultipleDatabases
end
def self.begin_transaction
self.all_connection_classes.each do |connection_class|
connection_class.connection.begin_transaction(joinable: false)
end
end
def self.rollback_transaction
self.all_connection_classes.each do |connection_class|
if connection_class.connection.open_transactions.zero?
warn "!!! before_all transaction has been already rollbacked and " \
"could work incorrectly"
next
end
connection_class.connection.rollback_transaction
end
end
end
# This class is required so we can disable transactions on migration specs
module NoTransactionAdapter
def self.begin_transaction; end
def self.rollback_transaction; end
end
def self.default_adapter
MultipleDatabaseAdapter
end
def self.no_transaction_adapter
NoTransactionAdapter
end
end
TestProf::BeforeAll.adapter = ::TestProfBeforeAllAdapter.default_adapter
|