make sure to print new lines before lock statements (#403)

closes #401 Co-authored-by: Lasath Fernando <[email protected]>
belav · Aug 19, 2021 · cae1e69 · cae1e69
1 parent edca48f
commit cae1e69
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 15 deletions.
diff --git a/Src/CSharpier.Tests/TestFiles/LockStatement/LockStatements.cst b/Src/CSharpier.Tests/TestFiles/LockStatement/LockStatements.cst
@@ -4,6 +4,7 @@ class ClassName
     {
         lock (sync)
             process();
+
         lock (sync)
         {
             process();

diff --git a/Src/CSharpier/SyntaxPrinter/SyntaxNodePrinters/LockStatement.cs b/Src/CSharpier/SyntaxPrinter/SyntaxNodePrinters/LockStatement.cs
@@ -9,24 +9,16 @@ public static class LockStatement
     {
         public static Doc Print(LockStatementSyntax node)
         {
-            var docs = new List<Doc>
-            {
+            var statement = Node.Print(node.Statement);
+
+            return Doc.Concat(
+                ExtraNewLines.Print(node),
                 Token.PrintWithSuffix(node.LockKeyword, " "),
                 Token.Print(node.OpenParenToken),
                 Node.Print(node.Expression),
-                Token.Print(node.CloseParenToken)
-            };
-            var statement = Node.Print(node.Statement);
-            if (node.Statement is BlockSyntax)
-            {
-                docs.Add(statement);
-            }
-            else
-            {
-                docs.Add(Doc.Indent(Doc.HardLine, statement));
-            }
-
-            return Doc.Concat(docs);
+                Token.Print(node.CloseParenToken),
+                node.Statement is BlockSyntax ? statement : Doc.Indent(Doc.HardLine, statement)
+            );
         }
     }
 }