Updated Generator to account for delegate field on yield

Author: bryanjos

CHANGELOG.md

@@ -1,3 +1,8 @@
+# v2.1.1
+
+* Enhancements
+  * ESTree.Tools.Generator: Updated yield to account for delegate field
+
 # v2.1.0
 
 * Enhancements

lib/es_tree/tools/generator.ex

@@ -458,10 +458,14 @@ defmodule ESTree.Tools.Generator do
     "(#{params})#{generator} => #{generate(ast.body)}"
   end
 
-  def do_generate(%ESTree.YieldExpression{argument: argument}, level) do
+  def do_generate(%ESTree.YieldExpression{argument: argument, delegate: false}, level) do
     "yield #{generate(argument)}"
   end
 
+  def do_generate(%ESTree.YieldExpression{argument: argument, delegate: true}, level) do
+    "yield* #{generate(argument)}"
+  end
+
   def do_generate(%ESTree.ObjectPattern{properties: properties}, level) do
      "{" <> Enum.map_join(properties, ",", &generate(&1)) <> "}"
   end

mix.exs

@@ -3,7 +3,7 @@ defmodule ESTree.Mixfile do
 
   def project do
     [app: :estree,
-     version: "2.1.0",
+     version: "2.1.1",
      elixir: "~> 1.0",
      deps: deps,
      description: description,