[ruby/yarp] Remove visitor usage to not rely on overriding visit

https://github.com/ruby/yarp/commit/6903860981

1 files changed, 15 insertions, 6 deletions

diff --git a/test/yarp/newline_test.rb b/test/yarp/newline_test.rb
index ba82b5ef45..7fb6d173bb 100644
--- a/test/yarp/newline_test.rb
+++ b/test/yarp/newline_test.rb
@@ -36,12 +36,7 @@ module YARP
 
       result = YARP.parse_file(filepath)
       assert_empty result.errors
-
-      result.mark_newlines!
-      visitor = NewlineVisitor.new(result.source)
-
-      result.value.accept(visitor)
-      actual = visitor.newlines
+      actual = yarp_lines(result)
 
       source.each_line.with_index(1) do |line, line_number|
         # Lines like `while (foo = bar)` result in two line flags in the
@@ -92,5 +87,19 @@ module YARP
 
       lines.sort
     end
+
+    def yarp_lines(result)
+      result.mark_newlines!
+
+      queue = [result.value]
+      newlines = []
+
+      while node = queue.shift
+        queue.concat(node.compact_child_nodes)
+        newlines << result.source.line(node.location.start_offset) if node&.newline?
+      end
+
+      newlines
+    end
   end
 end