lib/rubocop/cop/minitest/refute_false.rb in rubocop-minitest-0.5.0 vs lib/rubocop/cop/minitest/refute_false.rb in rubocop-minitest-0.5.1

@@ -14,10 +14,12 @@
       #   # good
       #   refute(actual)
       #   refute(actual, 'the message')
       #
       class RefuteFalse < Cop
+        include ArgumentRangeHelper
+
         MSG = 'Prefer using `refute(%<arguments>s)` over ' \
               '`assert_equal(false, %<arguments>s)`.'
 
         def_node_matcher :assert_equal_with_false, <<~PATTERN
           (send nil? :assert_equal false $_ $...)
@@ -33,12 +35,15 @@
           end
         end
 
         def autocorrect(node)
           lambda do |corrector|
-            arguments = node.arguments.reject(&:false_type?)
-            replacement = arguments.map(&:source).join(', ')
-            corrector.replace(node.loc.expression, "refute(#{replacement})")
+            assert_equal_with_false(node) do |actual|
+              corrector.replace(node.loc.selector, 'refute')
+              corrector.replace(
+                first_and_second_arguments_range(node), actual.source
+              )
+            end
           end
         end
       end
     end
   end