lib/rubocop/cop/rspec/match_array.rb in rubocop-rspec-2.19.0 vs lib/rubocop/cop/rspec/match_array.rb in rubocop-rspec-2.20.0
- old
+ new
@@ -1,12 +1,16 @@
# frozen_string_literal: true
module RuboCop
module Cop
module RSpec
- # Prefer `contain_exactly` when matching an array literal.
+ # Checks where `match_array` is used.
#
+ # This cop checks for the following:
+ # - Prefer `contain_exactly` when matching an array with values.
+ # - Prefer `eq` when using `match_array` with an empty array literal.
+ #
# @example
# # bad
# it { is_expected.to match_array([content1, content2]) }
#
# # good
@@ -15,23 +19,37 @@
# # good
# it { is_expected.to match_array([content] + array) }
#
# # good
# it { is_expected.to match_array(%w(tremble in fear foolish mortals)) }
+ #
class MatchArray < Base
extend AutoCorrector
MSG = 'Prefer `contain_exactly` when matching an array literal.'
RESTRICT_ON_SEND = %i[match_array].freeze
+ # @!method match_array_with_empty_array?(node)
+ def_node_matcher :match_array_with_empty_array?, <<~PATTERN
+ (send nil? :match_array (array))
+ PATTERN
+
def on_send(node)
return unless node.first_argument.array_type?
+ return if match_array_with_empty_array?(node)
+
+ check_populated_array(node)
+ end
+
+ private
+
+ def check_populated_array(node)
return if node.first_argument.percent_literal?
add_offense(node) do |corrector|
array_contents = node.arguments.flat_map(&:to_a)
corrector.replace(
- node.source_range,
+ node,
"contain_exactly(#{array_contents.map(&:source).join(', ')})"
)
end
end
end