Print trivia after closing multi line record brace. Fixes #1124. (#1126)

fsprojects · Sep 9, 2020 · 7cb0382 · 7cb0382
1 parent 8b958bf
commit 7cb0382
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/src/Fantomas.Tests/RecordTests.fs b/src/Fantomas.Tests/RecordTests.fs
@@ -1027,3 +1027,19 @@ let person =
       Address = { Street = "Bakerstreet"; Number = 42 } // end address
     } // end person
 """
+
+[<Test>]
+let ``comment after closing brace of multi-line record type, 1124`` () =
+    formatSourceString false """
+module Test =
+    type t =
+      { long_enough : string
+        also_long_enough : string } // Hi I am a comment
+"""  config
+    |> prepend newline
+    |> should equal """
+module Test =
+    type t =
+        { long_enough: string
+          also_long_enough: string } // Hi I am a comment
+"""
diff --git a/src/Fantomas/CodePrinter.fs b/src/Fantomas/CodePrinter.fs
@@ -3224,6 +3224,7 @@ and genMultilineSimpleRecordTypeDefn tdr ms ao' fs astContext =
                 (leaveLeftBrace tdr.Range
                  +> col sepSemiNln fs (genField astContext ""))
             +> sepCloseS
+            +> leaveNodeTokenByName tdr.Range RBRACE
             +> onlyIf (List.isNotEmpty ms) sepNln
             +> sepNlnBetweenTypeAndMembers ms
             +> genMemberDefnList