codewriter: Prepend "yield" if is_yield_expression is set on MethodCall/ObjectCreationExpression

1 files changed, 8 insertions, 0 deletions

diff --git a/vala/valacodewriter.vala b/vala/valacodewriter.vala
index 43b4312f6..620e45896 100644
--- a/vala/valacodewriter.vala
+++ b/vala/valacodewriter.vala
@@ -1220,6 +1220,10 @@ public class Vala.CodeWriter : CodeVisitor {
 	}
 
 	public override void visit_method_call (MethodCall expr) {
+		if (expr.is_yield_expression) {
+			write_string ("yield ");
+		}
+
 		expr.call.accept (this);
 		write_string (" (");
 
@@ -1276,6 +1280,10 @@ public class Vala.CodeWriter : CodeVisitor {
 	}
 
 	public override void visit_object_creation_expression (ObjectCreationExpression expr) {
+		if (expr.is_yield_expression) {
+			write_string ("yield ");
+		}
+
 		if (!expr.struct_creation) {
 			write_string ("new ");
 		}