gitlab-org--gitlab-foss/rubocop/cop/migration/update_column_in_batches.rb
Rémy Coutable a2d39b8010 Use gitlab-styles
Signed-off-by: Rémy Coutable <remy@rymai.me>
2017-12-22 08:34:44 -06:00

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{\Adb/(migrate|post_migrate)}, 'spec/migrations')
filename = File.basename(source_name, '.rb').sub(%r{\A\d+_}, '')
File.join(dirname, "#{filename}_spec.rb")
end
def rails_root
Pathname.new(File.expand_path('../../..', __dir__))
end
end
end
end
end