Merge pull request fsprojects#1046 from nojaf/fix-1043

Print trivia for UnionCase
nojaf · Aug 28, 2020 · faf5419 · faf5419
2 parents 31b3e6c + 8440484
commit faf5419
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 0 deletions.
diff --git a/src/Fantomas.Tests/UnionTests.fs b/src/Fantomas.Tests/UnionTests.fs
@@ -395,3 +395,29 @@ type Foo =
     | One = 0x00000001
     | Two = 0x00000002
 """
+
+[<Test>]
+let ``comment after union case, 1043`` () =
+    formatSourceString false """
+module FantomasTools.Client.FantomasOnline.Model
+
+open FantomasOnline.Shared
+
+type FantomasMode =
+    | V2
+    | V3
+    | V4
+    | Preview // master branch
+"""  config
+    |> prepend newline
+    |> should equal """
+module FantomasTools.Client.FantomasOnline.Model
+
+open FantomasOnline.Shared
+
+type FantomasMode =
+    | V2
+    | V3
+    | V4
+    | Preview // master branch
+"""
diff --git a/src/Fantomas/CodePrinter.fs b/src/Fantomas/CodePrinter.fs
@@ -2792,6 +2792,7 @@ and genUnionCase astContext (UnionCase(ats, px, _, s, UnionCaseType fs) as node)
     +> ifElse astContext.HasVerticalBar sepBar sepNone
     +> genOnelinerAttributes astContext ats -- s
     +> colPre wordOf sepStar fs (genField { astContext with IsUnionField = true } "")
+    |> genTriviaFor UnionCase_ node.Range
 
 and genEnumCase astContext (EnumCase(ats, px, _, (_,_)) as node) =
     let genCase (ctx: Context) =