avoid_setup_hook.rb

# frozen_string_literal: true

module RuboCop
  module Cop
    module RSpecRails
      # Checks that tests use RSpec `before` hook over Rails `setup` method.
      #
      # @example
      #   # bad
      #   setup do
      #     allow(foo).to receive(:bar)
      #   end
      #
      #   # good
      #   before do
      #     allow(foo).to receive(:bar)
      #   end
      #
      class AvoidSetupHook < ::RuboCop::Cop::RSpec::Base
        extend AutoCorrector

        MSG = 'Use `before` instead of `setup`.'

        # @!method setup_call(node)
        def_node_matcher :setup_call, <<~PATTERN
          (block
            $(send nil? :setup)
            (args) _)
        PATTERN

        def on_block(node) # rubocop:disable InternalAffairs/NumblockHandler
          setup_call(node) do |setup|
            add_offense(node) do |corrector|
              corrector.replace setup, 'before'
            end
          end
        end
      end
    end
  end
end