FIX: Allow post migrations using `#change` to carry out unsafe migration

This commit is contained in:
Guo Xiang Tan 2020-05-15 14:23:27 +08:00
parent 4601833e4e
commit 9ab5801a1b
No known key found for this signature in database
GPG Key ID: FBD110179AAC1F20
3 changed files with 30 additions and 6 deletions

View File

@ -34,13 +34,17 @@ class Migration::SafeMigrate
private
def is_post_deploy_migration?
instance_methods = self.class.instance_methods(false)
method =
if self.respond_to?(:up)
if instance_methods.include?(:up)
:up
elsif self.respond_to?(:change)
elsif instance_methods.include?(:change)
:change
end
return false if !method
self.method(method).source_location.first.include?(
Discourse::DB_POST_MIGRATE_PATH
)

View File

@ -99,18 +99,28 @@ describe Migration::SafeMigrate do
end
describe 'for a post deployment migration' do
it 'should not ban unsafe migrations' do
user = Fabricate(:user)
it 'should not ban unsafe migrations using up' do
Migration::SafeMigrate::SafeMigration.enable_safe!
path = File.expand_path "#{Rails.root}/spec/fixtures/db/post_migrate"
path = File.expand_path "#{Rails.root}/spec/fixtures/db/post_migrate/drop_table"
output = capture_stdout do
migrate_up(path)
end
expect(output).to include("drop_table(:email_logs)")
end
it 'should not ban unsafe migrations using change' do
Migration::SafeMigrate::SafeMigration.enable_safe!
path = File.expand_path "#{Rails.root}/spec/fixtures/db/post_migrate/change"
output = capture_stdout do
migrate_up(path)
end
expect(output).to include("drop_table(:email_logs)")
expect(user.reload).to eq(user)
end
end
end

View File

@ -0,0 +1,10 @@
# frozen_string_literal: true
class DropEmailLogs < ActiveRecord::Migration[5.2]
DROPPED_TABLES ||= %i{email_logs}
def change
drop_table :email_logs
raise ActiveRecord::Rollback
end
end