Sha256: 662bac5030fdbad494f889f056120a7053c948b0b26e664b36087bf40ba2731b
Contents?: true
Size: 1.72 KB
Versions: 3
Compression:
Stored size: 1.72 KB
Contents
# frozen_string_literal: true
module RuboCop
module Cop
module RSpec
module Rails
# Check if using Minitest matchers.
#
# @example
# # bad
# assert_equal(a, b)
# assert_equal a, b, "must be equal"
# refute_equal(a, b)
#
# # good
# expect(b).to eq(a)
# expect(b).to(eq(a), "must be equal")
# expect(b).not_to eq(a)
#
class MinitestAssertions < Base
extend AutoCorrector
MSG = 'Use `%<prefer>s`.'
RESTRICT_ON_SEND = %i[assert_equal refute_equal].freeze
# @!method minitest_assertion(node)
def_node_matcher :minitest_assertion, <<~PATTERN
(send nil? {:assert_equal :refute_equal} $_ $_ $_?)
PATTERN
def on_send(node)
minitest_assertion(node) do |expected, actual, failure_message|
prefer = replacement(node, expected, actual,
failure_message.first)
add_offense(node, message: message(prefer)) do |corrector|
corrector.replace(node, prefer)
end
end
end
private
def replacement(node, expected, actual, failure_message)
runner = node.method?(:assert_equal) ? 'to' : 'not_to'
if failure_message.nil?
"expect(#{actual.source}).#{runner} eq(#{expected.source})"
else
"expect(#{actual.source}).#{runner}(eq(#{expected.source}), " \
"#{failure_message.source})"
end
end
def message(prefer)
format(MSG, prefer: prefer)
end
end
end
end
end
end
Version data entries
3 entries across 3 versions & 2 rubygems