Migrate db schema for context in :migration specs

2 files changed, 3 insertions, 3 deletions

diff --git a/spec/spec_helper.rb b/spec/spec_helper.rb
index 3eea39d4bf4..c606030220f 100644
--- a/spec/spec_helper.rb
+++ b/spec/spec_helper.rb
@@ -132,14 +132,14 @@ RSpec.configure do |config|
     Sidekiq.redis(&:flushall)
   end
 
-  config.before(:example, :migration) do
+  config.before(:context, :migration) do |example|
     ActiveRecord::Migrator
       .migrate(migrations_paths, previous_migration.version)
 
     reset_column_in_migration_models
   end
 
-  config.after(:example, :migration) do
+  config.after(:context, :migration) do
     ActiveRecord::Migrator.migrate(migrations_paths)
 
     reset_column_in_migration_models
diff --git a/spec/support/db_cleaner.rb b/spec/support/db_cleaner.rb
index 7f5769209bb..7492f999a18 100644
--- a/spec/support/db_cleaner.rb
+++ b/spec/support/db_cleaner.rb
@@ -19,7 +19,7 @@ RSpec.configure do |config|
     DatabaseCleaner.strategy = :truncation
   end
 
-  config.before(:each, :migration) do
+  config.before(:context, :migration) do
     DatabaseCleaner.strategy = :truncation
   end