Add importQualifiedPostTests

Author: 3kyro

test/functional/FunctionalCodeAction.hs

@@ -28,6 +28,7 @@ tests = testGroup "code actions" [
 #if hls_refactor
       importTests
     , ignoreInEnv [HostOS Windows, GhcVer GHC94] "Diagnostic failure for Windows-ghc9.4.2" importQualifiedTests
+    , ignoreInEnv [HostOS Windows, GhcVer GHC94] "Diagnostic failure for Windows-ghc9.4.2" importQualifiedPostTests
     , packageTests
     , redundantImportTests
     , renameTests
@@ -105,7 +106,7 @@ importTests = testGroup "import suggestions" [
     ]
 
 importQualifiedTests :: TestTree
-importQualifiedTests = testGroup "import qualified suggestions" [
+importQualifiedTests = testGroup "import qualified prefix suggestions" [
     testCase "qualified import works with 3.8 code action kinds" $ runSession hlsCommand fullCaps "test/testdata" $ do
         doc <- openDoc "CodeActionImportQualified.hs" "haskell"
         -- No Formatting:
@@ -118,7 +119,7 @@ importQualifiedTests = testGroup "import qualified suggestions" [
         actionsOrCommands <- getAllCodeActions doc
         let actns = map fromAction actionsOrCommands
 
-        let importQualifiedSuggestion = "import qualified Control.Monad "
+        let importQualifiedSuggestion = "import qualified Control.Monad as Control"
         importControlMonadQualified <- liftIO $ inspectCodeAction actionsOrCommands [importQualifiedSuggestion]
         liftIO $ do
             dontExpectCodeAction actionsOrCommands ["import Control.Monad (when)"]
@@ -130,6 +131,32 @@ importQualifiedTests = testGroup "import qualified suggestions" [
         liftIO $ contents @?= "import qualified Control.Monad as Control\nmain :: IO ()\nmain = Control.when True $ putStrLn \"hello\"\n"
     ]
 
+importQualifiedPostTests :: TestTree
+importQualifiedPostTests = testGroup "import qualified postfix suggestions" [
+    testCase "qualified import in postfix position works with 3.8 code action kinds" $ runSession hlsCommand fullCaps "test/testdata" $ do
+        doc <- openDoc "CodeActionImportPostQualified.hs" "haskell"
+        -- No Formatting:
+        let config = def { formattingProvider = "none" }
+        sendConfigurationChanged (toJSON config)
+
+        (diag:_) <- waitForDiagnosticsFrom doc
+        liftIO $ diag ^. L.message @?= "Not in scope: ‘Control.when’\nNo module named ‘Control’ is imported."
+
+        actionsOrCommands <- getAllCodeActions doc
+        let actns = map fromAction actionsOrCommands
+
+        let importQualifiedPostSuggestion = "import Control.Monad qualified as Control"
+        importControlMonadQualified <- liftIO $ inspectCodeAction actionsOrCommands [importQualifiedPostSuggestion]
+        liftIO $ do
+            dontExpectCodeAction actionsOrCommands ["import qualified Control.Monad as Control", "import Control.Monad (when)"]
+            length actns >= 10 @? "There are some actions"
+
+        executeCodeAction importControlMonadQualified
+
+        contents <- documentContents doc
+        liftIO $ T.lines contents !! 2 @?= "import Control.Monad qualified as Control"
+    ]
+
 packageTests :: TestTree
 packageTests = testGroup "add package suggestions" [
     ignoreTestBecause "no support for adding dependent packages via code action" $ testCase "adds to .cabal files" $ do

test/testdata/CodeActionImportPostQualified.hs

@@ -0,0 +1,4 @@
+{-# LANGUAGE ImportQualifiedPost #-}
+{-# OPTIONS_GHC -Wprepositive-qualified-module #-}
+main :: IO ()
+main = Control.when True $ putStrLn "hello"