Welcome to mirror list, hosted at ThFree Co, Russian Federation.

gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to 'spec/support/db_cleaner.rb')
-rw-r--r--spec/support/db_cleaner.rb8
1 files changed, 7 insertions, 1 deletions
diff --git a/spec/support/db_cleaner.rb b/spec/support/db_cleaner.rb
index ff913ebf22b..155dc3c17d9 100644
--- a/spec/support/db_cleaner.rb
+++ b/spec/support/db_cleaner.rb
@@ -1,6 +1,10 @@
# frozen_string_literal: true
module DbCleaner
+ def all_connection_classes
+ ::ActiveRecord::Base.connection_handler.connection_pool_names.map(&:constantize)
+ end
+
def delete_from_all_tables!(except: [])
except << 'ar_internal_metadata'
@@ -12,7 +16,9 @@ module DbCleaner
end
def setup_database_cleaner
- DatabaseCleaner[:active_record, { connection: ActiveRecord::Base }]
+ all_connection_classes.each do |connection_class|
+ DatabaseCleaner[:active_record, { connection: connection_class }]
+ end
end
end