Test apply-refact preserve comments

haskell · Dec 23, 2020 · 1c56e0e · 1c56e0e
1 parent 1526f90
commit 1c56e0e
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 0 deletions.
diff --git a/test/functional/FunctionalCodeAction.hs b/test/functional/FunctionalCodeAction.hs
@@ -155,6 +155,9 @@ hlintTests = testGroup "hlint suggestions" [
       testCase "hlint diagnostics ignore hints honouring HLINT annotations" $ runHlintSession "" $ do
         doc <- openDoc "ApplyRefact5.hs" "haskell"
         expectNoMoreDiagnostics 3 doc "hlint"
+
+    , testCase "apply-refact preserve regular comments" $ runHlintSession "" $ do
+        testRefactor "ApplyRefact6.hs" "Redundant bracket" expectedComments
     ]
     where
         runHlintSession :: FilePath -> Session a -> IO a
@@ -193,6 +196,14 @@ hlintTests = testGroup "hlint suggestions" [
                              , "g = 2"
                              , "#endif", ""
                              ]
+        expectedComments =   [ "-- comment before header"
+                             , "module ApplyRefact6 where", ""
+                             , "{-# standalone annotation #-}", ""
+                             , "-- standalone comment", ""
+                             , "-- | haddock comment"
+                             , "f = {- inline comment -} 1 -- ending comment", ""
+                             , "-- final comment"
+                             ]
 
 renameTests :: TestTree
 renameTests = testGroup "rename suggestions" [

diff --git a/test/testdata/hlint/ApplyRefact6.hs b/test/testdata/hlint/ApplyRefact6.hs
@@ -0,0 +1,11 @@
+-- comment before header
+module ApplyRefact6 where
+
+{-# standalone annotation #-}
+
+-- standalone comment
+
+-- | haddock comment
+f = {- inline comment -} (1) -- ending comment
+
+-- final comment
diff --git a/test/testdata/hlint/hie.yaml b/test/testdata/hlint/hie.yaml
@@ -8,3 +8,4 @@ cradle:
       - "ApplyRefact3"
       - "ApplyRefact4"
       - "ApplyRefact5"
+      - "ApplyRefact6"