aboutsummaryrefslogtreecommitdiffstats
path: root/activerecord/test/support/connection_helper.rb
diff options
context:
space:
mode:
Diffstat (limited to 'activerecord/test/support/connection_helper.rb')
-rw-r--r--activerecord/test/support/connection_helper.rb14
1 files changed, 14 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..4a19e5df44
--- /dev/null
+++ b/activerecord/test/support/connection_helper.rb
@@ -0,0 +1,14 @@
+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