diff options
Diffstat (limited to 'activerecord/test/support/connection_helper.rb')
-rw-r--r-- | activerecord/test/support/connection_helper.rb | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/activerecord/test/support/connection_helper.rb b/activerecord/test/support/connection_helper.rb new file mode 100644 index 0000000000..3bb1b370c1 --- /dev/null +++ b/activerecord/test/support/connection_helper.rb @@ -0,0 +1,16 @@ +# frozen_string_literal: true + +module ConnectionHelper + def run_without_connection + original_connection = ActiveRecord::Base.remove_connection + yield original_connection + ensure + ActiveRecord::Base.establish_connection(original_connection) + end + + # Used to drop all cache query plans in tests. + def reset_connection + original_connection = ActiveRecord::Base.remove_connection + ActiveRecord::Base.establish_connection(original_connection) + end +end |