Print trivia before SynTypeDefnSimpleRepr.Union in signature files. F…

…ixes #1560. (#1566)
fsprojects · Apr 1, 2021 · 82615ce · 82615ce
1 parent 44926de
commit 82615ce
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 7 deletions.
diff --git a/src/Fantomas.Tests/SignatureTests.fs b/src/Fantomas.Tests/SignatureTests.fs
@@ -1298,3 +1298,30 @@ exception FileNameNotResolved of string (*description of searched locations*)  *
 
 exception LoadedSourceNotFoundIgnoring of string * range (*filename*)
 """
+
+[<Test>]
+let ``comment above first DU case`` () =
+    formatSourceString
+        true
+        """
+namespace Baz
+
+type 'a Bar =
+    ///
+    | Foo
+    ///
+    | Quux
+"""
+        config
+    |> prepend newline
+    |> should
+        equal
+        """
+namespace Baz
+
+type 'a Bar =
+    ///
+    | Foo
+    ///
+    | Quux
+"""
diff --git a/src/Fantomas/CodePrinter.fs b/src/Fantomas/CodePrinter.fs
@@ -3742,13 +3742,14 @@ and genSigTypeDefn astContext (SigTypeDef (ats, px, ao, tds, tcs, tdr, ms, s, pr
             | xs ->
                 indent
                 +> sepNln
-                +> opt sepNln ao' genAccess
-                +> col
-                    sepNln
-                    xs
-                    (genUnionCase
-                        { astContext with
-                              HasVerticalBar = true })
+                +> (opt sepNln ao' genAccess
+                    +> col
+                        sepNln
+                        xs
+                        (genUnionCase
+                            { astContext with
+                                  HasVerticalBar = true })
+                    |> genTriviaFor SynTypeDefnSimpleRepr_Union tdr.Range)
 
         typeName
         +> sepEq