diff options
Diffstat (limited to 'spec/support/before_all_adapter.rb')
-rw-r--r-- | spec/support/before_all_adapter.rb | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/spec/support/before_all_adapter.rb b/spec/support/before_all_adapter.rb new file mode 100644 index 00000000000..f48e0f46e80 --- /dev/null +++ b/spec/support/before_all_adapter.rb @@ -0,0 +1,27 @@ +# frozen_string_literal: true + +class BeforeAllAdapter # rubocop:disable Gitlab/NamespacedClass + def self.all_connection_pools + ::ActiveRecord::Base.connection_handler.all_connection_pools + end + + def self.begin_transaction + self.all_connection_pools.each do |connection_pool| + connection_pool.connection.begin_transaction(joinable: false) + end + end + + def self.rollback_transaction + self.all_connection_pools.each do |connection_pool| + if connection_pool.connection.open_transactions.zero? + warn "!!! before_all transaction has been already rollbacked and " \ + "could work incorrectly" + next + end + + connection_pool.connection.rollback_transaction + end + end +end + +TestProf::BeforeAll.adapter = ::BeforeAllAdapter |