diff options
Diffstat (limited to 'rubocop/cop/include_sidekiq_worker.rb')
-rw-r--r-- | rubocop/cop/include_sidekiq_worker.rb | 12 |
1 files changed, 5 insertions, 7 deletions
diff --git a/rubocop/cop/include_sidekiq_worker.rb b/rubocop/cop/include_sidekiq_worker.rb index e39b8bf92c2..1a42de64aa9 100644 --- a/rubocop/cop/include_sidekiq_worker.rb +++ b/rubocop/cop/include_sidekiq_worker.rb @@ -3,7 +3,9 @@ module RuboCop module Cop # Cop that makes sure workers include `ApplicationWorker`, not `Sidekiq::Worker`. - class IncludeSidekiqWorker < RuboCop::Cop::Cop + class IncludeSidekiqWorker < RuboCop::Cop::Base + extend RuboCop::Cop::AutoCorrector + MSG = 'Include `ApplicationWorker`, not `Sidekiq::Worker`.' def_node_matcher :includes_sidekiq_worker?, <<~PATTERN @@ -13,12 +15,8 @@ module RuboCop def on_send(node) return unless includes_sidekiq_worker?(node) - add_offense(node.arguments.first, location: :expression) - end - - def autocorrect(node) - lambda do |corrector| - corrector.replace(node.source_range, 'ApplicationWorker') + add_offense(node.arguments.first) do |corrector| + corrector.replace(node.arguments.first, 'ApplicationWorker') end end end |