class RuboCop::Cop::Lint::ToEnumArguments
Ensures that `to_enum`/`enum_for`, called for the current method, has correct arguments.
@example
# bad def foo(x, y = 1) return to_enum(__callee__, x) # `y` is missing end # good def foo(x, y = 1) return to_enum(__callee__, x, y) # alternatives to `__callee__` are `__method__` and `:foo` end
Constants
- MSG
- RESTRICT_ON_SEND
Public Instance Methods
on_send(node)
click to toggle source
# File lib/rubocop/cop/lint/to_enum_arguments.rb, line 41 def on_send(node) def_node = node.each_ancestor(:def, :defs).first return unless def_node enum_conversion_call?(node) do |method_node, arguments| add_offense(node) unless method_name?(method_node, def_node.method_name) && arguments_match?(arguments, def_node) end end
Private Instance Methods
argument_match?(send_arg, def_arg)
click to toggle source
rubocop:disable Metrics/AbcSize, Metrics/CyclomaticComplexity
# File lib/rubocop/cop/lint/to_enum_arguments.rb, line 68 def argument_match?(send_arg, def_arg) def_arg_name = def_arg.children[0] case def_arg.type when :arg, :restarg send_arg.source == def_arg.source when :optarg send_arg.source == def_arg_name.to_s when :kwoptarg, :kwarg send_arg.hash_type? && send_arg.pairs.any? { |pair| passing_keyword_arg?(pair, def_arg_name) } when :kwrestarg send_arg.each_child_node(:kwsplat).any? { |child| child.source == def_arg.source } when :forward_arg send_arg.forwarded_args_type? end end
arguments_match?(arguments, def_node)
click to toggle source
# File lib/rubocop/cop/lint/to_enum_arguments.rb, line 53 def arguments_match?(arguments, def_node) index = 0 def_node.arguments.reject(&:blockarg_type?).all? do |def_arg| send_arg = arguments[index] case def_arg.type when :arg, :restarg, :optarg index += 1 end send_arg && argument_match?(send_arg, def_arg) end end