Add a pipe when a single case DU has attributes.

Fixes fsprojects#596
nojaf · Dec 15, 2019 · 7ad5e3f · 7ad5e3f
1 parent 4913ed6
commit 7ad5e3f
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/src/Fantomas.Tests/UnionTests.fs b/src/Fantomas.Tests/UnionTests.fs
@@ -217,4 +217,16 @@ type internal Foo = private | Bar
     |> should equal """namespace meh
 
 type internal Foo = private | Bar
+"""
+
+[<Test>]
+let ``preserve pipe when single choice contains attribute, 596`` () =
+    formatSourceString false """type [<StringEnum>] [<RequireQualifiedAccess>] PayableFilters =
+    | [<CompiledName "statusSelector">] Status
+"""  config
+    |> prepend newline
+    |> should equal """
+[<StringEnum>]
+[<RequireQualifiedAccess>]
+type PayableFilters = | [<CompiledName "statusSelector">] Status
 """
diff --git a/src/Fantomas/CodePrinter.fs b/src/Fantomas/CodePrinter.fs
@@ -1525,8 +1525,8 @@ and genTypeDefn astContext (TypeDef(ats, px, ao, tds, tcs, tdr, ms, s, preferPos
         let unionCases =  
             match xs with
             | [] -> id
-            | [x] when List.isEmpty ms ->
-                let hasVerticalBar = Option.isSome ao'
+            | [UnionCase(attrs, _,_,_,_) as x] when List.isEmpty ms ->
+                let hasVerticalBar = Option.isSome ao' || not (List.isEmpty attrs)
 
                 indent +> sepSpace +> sepNlnBasedOnTrivia
                 +> genTrivia tdr.Range