43 lines
1.3 KiB
Ruby
43 lines
1.3 KiB
Ruby
require_relative '../../migration_helpers'
|
|
|
|
module RuboCop
|
|
module Cop
|
|
module Migration
|
|
# Cop that checks if a spec file exists for any migration using
|
|
# `update_column_in_batches`.
|
|
class UpdateColumnInBatches < RuboCop::Cop::Cop
|
|
include MigrationHelpers
|
|
|
|
MSG = 'Migration running `update_column_in_batches` must have a spec file at' \
|
|
' `%s`.'.freeze
|
|
|
|
def on_send(node)
|
|
return unless in_migration?(node)
|
|
return unless node.children[1] == :update_column_in_batches
|
|
|
|
spec_path = spec_filename(node)
|
|
|
|
unless File.exist?(File.expand_path(spec_path, rails_root))
|
|
add_offense(node, location: :expression, message: format(MSG, spec_path))
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def spec_filename(node)
|
|
source_name = node.location.expression.source_buffer.name
|
|
path = Pathname.new(source_name).relative_path_from(rails_root)
|
|
dirname = File.dirname(path)
|
|
.sub(%r{db/(migrate|post_migrate)}, 'spec/migrations')
|
|
filename = File.basename(source_name, '.rb').sub(/\A\d+_/, '')
|
|
|
|
File.join(dirname, "#{filename}_spec.rb")
|
|
end
|
|
|
|
def rails_root
|
|
Pathname.new(File.expand_path('../../..', __dir__))
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|