Contents

# frozen_string_literal: true

module RuboCop
  module Cop
    module Minitest
      # This cop enforces the use of `assert_equal(expected, actual)`
      # over `assert(expected == actual)`.
      #
      # @example
      #   # bad
      #   assert("rubocop-minitest" == actual)
      #
      #   # good
      #   assert_equal("rubocop-minitest", actual)
      #
      class AssertEqual < Cop
        include ArgumentRangeHelper

        MSG = 'Prefer using `assert_equal(%<preferred>s)` over ' \
              '`assert(%<over>s)`.'

        def_node_matcher :assert_equal, <<~PATTERN
          (send nil? :assert $(send $_ :== $_) $...)
        PATTERN

        def on_send(node)
          assert_equal(node) do |first_receiver_arg, expected, actual, rest_receiver_arg|
            message = rest_receiver_arg.first
            preferred = [expected.source, actual.source, message&.source]
                        .compact.join(', ')
            over = [first_receiver_arg.source, message&.source].compact.join(', ')

            offense_message = format(MSG, preferred: preferred, over: over)

            add_offense(node, message: offense_message)
          end
        end

        def autocorrect(node)
          lambda do |corrector|
            assert_equal(node) do |_, expected, actual|
              corrector.replace(node.loc.selector, 'assert_equal')

              replacement = [expected, actual].map(&:source).join(', ')
              corrector.replace(first_argument_range(node), replacement)
            end
          end
        end
      end
    end
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
rubocop-minitest-0.7.0	lib/rubocop/cop/minitest/assert_equal.rb
rubocop-minitest-0.6.2	lib/rubocop/cop/minitest/assert_equal.rb
rubocop-minitest-0.6.1	lib/rubocop/cop/minitest/assert_equal.rb
rubocop-minitest-0.6.0	lib/rubocop/cop/minitest/assert_equal.rb
rubocop-minitest-0.5.1	lib/rubocop/cop/minitest/assert_equal.rb