Implement YieldNode

lib/ast_to_prism/parser.rb

@@ -1263,7 +1263,16 @@ def convert_node(node, block: nil)
           location(node) # location
         )
       when :YIELD
-        not_supported(node)
+        nd_head, = node.children
+
+        Prism::YieldNode.new(
+          source,                     # source
+          null_location,              # keyword_loc
+          null_location,              # lparen_loc
+          convert_arguments(nd_head), # arguments
+          null_location,              # rparen_loc
+          location(node)              # location
+        )
       when :LVAR
         nd_vid, = node.children
 

spec/basic_spec.rb

@@ -1637,6 +1637,48 @@ def m
     end
   end
 
+  describe "yield" do
+    it "tests" do
+      pending "YieldNode locations are not supported"
+
+      test_code(<<~CODE)
+        def m
+          yield
+        end
+      CODE
+    end
+
+    it "tests" do
+      pending "YieldNode locations are not supported"
+
+      test_code(<<~CODE)
+        def m
+          yield()
+        end
+      CODE
+    end
+
+    it "tests" do
+      pending "YieldNode locations are not supported"
+
+      test_code(<<~CODE)
+        def m
+          yield 1, 2
+        end
+      CODE
+    end
+
+    it "tests" do
+      pending "YieldNode locations are not supported"
+
+      test_code(<<~CODE)
+        def m
+          yield(1, 2)
+        end
+      CODE
+    end    
+  end
+
   describe "redo" do
     it "tests" do
       pending "opening_loc and closing_loc of BlockParametersNode are not supported"

spec/spec_helper.rb

@@ -414,6 +414,18 @@ def ===(other)
     }
   end
 
+  class YieldNode
+    prepend Module.new {
+      def ===(other)
+        super(other) &&
+        self.location == other.location &&
+        self.keyword_loc == other.keyword_loc &&
+        self.lparen_loc == other.lparen_loc &&
+        self.rparen_loc == other.rparen_loc
+      end
+    }
+  end
+
   class ForNode
     prepend Module.new {
       def ===(other)