genTrivia for SynEnumCase. Fixes fsprojects#1247.

nojaf · Nov 13, 2020 · f8358a3 · f8358a3
1 parent 0ce38c1
commit f8358a3
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/src/Fantomas.Tests/UnionTests.fs b/src/Fantomas.Tests/UnionTests.fs
@@ -559,3 +559,17 @@ type CardValue =
             yield King
         ]
 """
+
+[<Test>]
+let ``comment after enum field, 1247`` () =
+    formatSourceString false """
+type Foo =
+    | Bar = 3 // Foo
+    | Baz = 5 // Eee
+"""  config
+    |> prepend newline
+    |> should equal """
+type Foo =
+    | Bar = 3 // Foo
+    | Baz = 5 // Eee
+"""
diff --git a/src/Fantomas/CodePrinter.fs b/src/Fantomas/CodePrinter.fs
@@ -3650,7 +3650,9 @@ and genEnumCase astContext (EnumCase (ats, px, _, (_, _)) as node) =
     let genCase (ctx: Context) =
         let expr =
             match node with
-            | EnumCase (_, _, identInAST, (c, r)) -> !-identInAST +> !- " = " +> genConst c r
+            | EnumCase (_, _, identInAST, (c, r)) ->
+                !-identInAST +> !- " = " +> genConst c r
+                |> genTriviaFor EnumCase_ r
 
         expr ctx