test/test_helper.rb in attr_encrypted-3.0.3 vs test/test_helper.rb in attr_encrypted-3.1.0

@@ -25,10 +25,11 @@
 
 require 'active_record'
 require 'data_mapper'
 require 'digest/sha2'
 require 'sequel'
+ActiveSupport::Deprecation.behavior = :raise
 
 $:.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
 $:.unshift(File.dirname(__FILE__))
 require 'attr_encrypted'
 
@@ -46,6 +47,12 @@
 
 SECRET_KEY = SecureRandom.random_bytes(32)
 
 def base64_encoding_regex
   /^(?:[A-Za-z0-9+\/]{4})*(?:[A-Za-z0-9+\/]{2}==|[A-Za-z0-9+\/]{3}=|[A-Za-z0-9+\/]{4})$/
+end
+
+def drop_all_tables
+  connection = ActiveRecord::Base.connection
+  tables = (ActiveRecord::VERSION::MAJOR >= 5 ? connection.data_sources : connection.tables)
+  tables.each { |table| ActiveRecord::Base.connection.drop_table(table) }
 end