# frozen_string_literal: true

require 'rubocop-rspec'

module RuboCop
  module Cop
    module RSpec
      class WebMockEnable < RuboCop::Cop::Base
        extend RuboCop::Cop::AutoCorrector

        # This cop checks for `WebMock.disable_net_connect!` usage in specs and
        # replaces it with `webmock_enable!`
        #
        # @example
        #
        #   # bad
        #   WebMock.disable_net_connect!
        #   WebMock.disable_net_connect!(allow_localhost: true)
        #
        #   # good
        #   webmock_enable!

        MESSAGE = 'Use webmock_enable! instead of calling WebMock.disable_net_connect! directly.'

        # @!method webmock_disable_net_connect?(node)
        def_node_matcher :webmock_disable_net_connect?, <<~PATTERN
          (send (const nil? :WebMock) :disable_net_connect! ...)
        PATTERN

        def on_send(node)
          if webmock_disable_net_connect?(node)
            add_offense(node, message: MESSAGE) do |corrector|
              corrector.replace(node, 'webmock_enable!')
            end
          end
        end
      end
    end
  end
end
