Add line when await precedes query expression (#741)

* Add line when await precedes query expression. * Formatting.
belav · Oct 24, 2022 · 509862c · 509862c
1 parent 4aabbf0
commit 509862c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/Src/CSharpier.Tests/FormattingTests/TestFiles/QueryExpressions.cst b/Src/CSharpier.Tests/FormattingTests/TestFiles/QueryExpressions.cst
@@ -63,5 +63,11 @@ class ClassName
                 Country________________________ = g.Key,
                 CustCount____________________________ = g.Count()
             };
+
+        var asyncQuery = await
+            from c in customers
+            from u in users
+            from res in HandleAsync(c, u)
+            select res;
     }
 }
diff --git a/Src/CSharpier/SyntaxPrinter/SyntaxNodePrinters/AwaitExpression.cs b/Src/CSharpier/SyntaxPrinter/SyntaxNodePrinters/AwaitExpression.cs
@@ -4,9 +4,13 @@ internal static class AwaitExpression
 {
     public static Doc Print(AwaitExpressionSyntax node, FormattingContext context)
     {
+        var precedesQueryExpression = node.Expression is QueryExpressionSyntax;
+
         return Doc.Concat(
             Token.PrintWithSuffix(node.AwaitKeyword, " ", context),
-            Node.Print(node.Expression, context)
+            precedesQueryExpression
+                ? Doc.Indent(Doc.Line, Node.Print(node.Expression, context))
+                : Node.Print(node.Expression, context)
         );
     }
 }