Simplify heredoc detection

rubocop · Jan 10, 2019 · 9c0fd1b · 9c0fd1b
1 parent 2f5f125
commit 9c0fd1b
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/lib/rubocop/cop/rspec/aggregate_examples.rb b/lib/rubocop/cop/rspec/aggregate_examples.rb
@@ -258,10 +258,9 @@ def example_method?(method_name)
         # Searches for HEREDOC in examples. It can be tricky to aggregate,
         # especially when interleaved with parenthesis or curly braces.
         def contains_heredoc?(node)
-          return unless node.is_a? AST::Node
-          return true if (node.dstr_type? || node.str_type?) && node.heredoc?
+          return true if node.respond_to?(:heredoc) && node.heredoc?
 
-          node.children.compact.any? { |child| contains_heredoc?(child) }
+          node.each_child_node.any? { |child| contains_heredoc?(child) }
         end
 
         def_node_matcher :subject_with_no_args?, <<-PATTERN