[test] test added for pragma addition after shebang

haskell · Feb 23, 2021 · 50bd268 · 50bd268
1 parent 7817a10
commit 50bd268
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 0 deletions.
diff --git a/test/functional/FunctionalCodeAction.hs b/test/functional/FunctionalCodeAction.hs
@@ -561,6 +561,35 @@ missingPragmaTests = testGroup "missing pragma warning code actions" [
                     , ""
                     , "f Record{a, b} = a"
                     ]
+            liftIO $ T.lines contents @?= expected
+    , testCase "After Shebang" $ do
+        runSession hlsCommand fullCaps "test/testdata/addPragmas" $ do
+            doc <- openDoc "AfterShebang.hs" "haskell"
+
+            _ <- waitForDiagnosticsFrom doc
+            cas <- map fromAction <$> getAllCodeActions doc
+
+            liftIO $ "Add \"NamedFieldPuns\"" `elem` map (^. L.title) cas @? "Contains NamedFieldPuns code action"
+
+            executeCodeAction $ head cas
+
+            contents <- documentContents doc
+
+            let expected =
+                    [ "#! /usr/bin/env nix-shell"
+                    , "#! nix-shell --pure -i runghc -p \"haskellPackages.ghcWithPackages (hp: with hp; [ turtle ])\""
+                    , ""
+                    , "{-# LANGUAGE NamedFieldPuns #-}"
+                    , "module AfterShebang where"
+                    , ""
+                    , "data Record = Record"
+                    , "  { a :: Int,"
+                    , "    b :: Double,"
+                    , "    c :: String"
+                    , "  }"
+                    , ""
+                    , "f Record{a, b} = a"
+                    ]
 
             liftIO $ T.lines contents @?= expected
     ]

diff --git a/test/testdata/addPragmas/AfterShebang.hs b/test/testdata/addPragmas/AfterShebang.hs
@@ -0,0 +1,12 @@
+#! /usr/bin/env nix-shell
+#! nix-shell --pure -i runghc -p "haskellPackages.ghcWithPackages (hp: with hp; [ turtle ])"
+
+module AfterShebang where
+
+data Record = Record
+  { a :: Int,
+    b :: Double,
+    c :: String
+  }
+
+f Record{a, b} = a