Printing trivia after SynConst. Fixes #598. (#670)

fsprojects · Feb 6, 2020 · eb2dc77 · eb2dc77
1 parent 67ee883
commit eb2dc77
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 2 deletions.
diff --git a/src/Fantomas.Tests/SynConstTests.fs b/src/Fantomas.Tests/SynConstTests.fs
@@ -100,4 +100,31 @@ let nullstr = "\u0000"
     |> should equal """
 let nulchar = '\u0000'
 let nullstr = "\u0000"
-"""
+"""
+
+[<Test>]
+let ``line comment after custom measure type, 598`` () =
+    formatSourceString false """namespace Krach
+
+module Runner =
+
+    let mPerSecond = 1000<m/second> // foo
+
+    [<Measure>]
+    type ProcessId =
+        class
+        end
+"""  config
+    |> prepend newline
+    |> should equal """
+namespace Krach
+
+module Runner =
+
+    let mPerSecond = 1000<m/second> // foo
+
+    [<Measure>]
+    type ProcessId =
+        class
+        end
+"""
diff --git a/src/Fantomas/CodePrinter.fs b/src/Fantomas/CodePrinter.fs
@@ -2308,7 +2308,7 @@ and genConst (c:SynConst) (r:range) =
             | Measure m -> !- m
 
         genConstNumber c r +> measure
-
+    |> genTrivia r
 and genConstNumber (c:SynConst) (r: range) =
     fun (ctx: Context) ->
         ctx.Trivia