gitlab-org--gitlab-foss/rubocop/cop/rspec/web_mock_enable.rb

39 lines
1 KiB
Ruby

# frozen_string_literal: true
module RuboCop
module Cop
module RSpec
class WebMockEnable < RuboCop::Cop::Cop
# 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.'
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, location: :expression, message: MESSAGE)
end
end
def autocorrect(node)
lambda do |corrector|
corrector.replace(node, 'webmock_enable!')
end
end
end
end
end
end