add more tests

plugins/hls-haddock-comments-plugin/src/Ide/Plugin/HaddockComments/Data.hs

@@ -25,7 +25,7 @@ import           Language.Haskell.GHC.ExactPrint.Utils (mkComment)
 genForDataDecl :: GenComments
 genForDataDecl =
     GenComments {
-        title = "Generate fields comments",
+        title = "Generate haddock comments",
         updateAnns = updateDataAnns
     }
 

plugins/hls-haddock-comments-plugin/test/Main.hs

@@ -30,9 +30,11 @@ tests =
       goldenWithHaddockComments "MultivariateFunction" Signature 4 8,
       goldenWithHaddockComments "QualFunction" Signature 2 10,
       goldenWithHaddockComments "Record" Record 7 2,
+      goldenWithHaddockComments "Record2" Record 3 6,
+      goldenWithHaddockComments "InlineRecord" Record 3 20,
       expectedNothing "ConstFunction" Signature 2 2,
       expectedNothing "StaleFunction" Signature 3 3,
-      expectedNothing "StaleRecord" Record 3 12
+      expectedNothing "StaleRecord" Record 4 9
     ]
 
 goldenWithHaddockComments :: FilePath -> GenCommentsType -> UInt -> UInt -> TestTree
@@ -54,7 +56,7 @@ data GenCommentsType = Signature | Record
 
 toTitle :: GenCommentsType -> Text
 toTitle Signature = "Generate signature comments"
-toTitle Record    = "Generate fields comments"
+toTitle Record    = "Generate haddock comments"
 
 caTitle :: (Command |? CodeAction) -> Maybe Text
 caTitle (InR CodeAction {_title}) = Just _title

plugins/hls-haddock-comments-plugin/test/testdata/InlineRecord.expected.hs

@@ -0,0 +1,10 @@
+module Record2 where
+
+-- | A record
+data Record = -- |
+              A { -- |
+                  a :: Int , -- |
+              b :: String }
+              | -- |
+              B { -- |
+                  bb :: Double }

plugins/hls-haddock-comments-plugin/test/testdata/InlineRecord.hs

@@ -0,0 +1,4 @@
+module Record2 where
+
+-- | A record
+data Record = A { a :: Int , b :: String } | B { bb :: Double }

plugins/hls-haddock-comments-plugin/test/testdata/Record2.expected.hs

@@ -0,0 +1,10 @@
+module Record2 where
+
+-- | A record
+data Record = -- |
+              RecordA
+    { -- |
+      a :: Int
+    , -- |
+      b :: String
+    }

plugins/hls-haddock-comments-plugin/test/testdata/Record2.hs

@@ -0,0 +1,7 @@
+module Record2 where
+
+-- | A record
+data Record = RecordA
+    { a :: Int
+    , b :: String
+    }