Add option to configure capabilities

[opengl.git] / test / Test.hs

diff --git a/test/Test.hs b/test/Test.hs
index 16c8d01cdb57be9f4710a98df48ad00c121c6fa3..0bffa5e1a7c3f7a23d40205887b9216441bf5454 100644 (file)
--- a/test/Test.hs
+++ b/test/Test.hs
@@ -1,11 +1,18 @@
 {-# LANGUAGE DuplicateRecordFields #-}
 {-# LANGUAGE OverloadedStrings #-}
+{-# LANGUAGE DeriveGeneric #-}
+{-# LANGUAGE DeriveAnyClass #-}
 import           Test.Hspec
+import           Data.Aeson
+import           Data.Default
+import qualified Data.HashMap.Strict as HM
 import           Data.Maybe
 import           Control.Monad.IO.Class
 import           Control.Lens hiding (List)
+import           GHC.Generics
 import           Language.Haskell.LSP.Test
 import           Language.Haskell.LSP.Test.Replay
+import           Language.Haskell.LSP.TH.ClientCapabilities
 import           Language.Haskell.LSP.Types
 import           ParsingTests
 
@@ -36,6 +43,15 @@ main = hspec $ do
                       skipMany loggingNotification
                       anyRequest
         in session `shouldThrow` anyException
+    it "can get initialize response" $ runSession "hie --lsp" "test/data/renamePass" $ do
+      rsp <- getInitializeResponse
+      liftIO $ rsp ^. result `shouldNotBe` Nothing
+
+    it "can register specific capabilities" $ do
+      let caps = def { _workspace = Just workspaceCaps }
+          workspaceCaps = def { _didChangeConfiguration = Just configCaps }
+          configCaps = DidChangeConfigurationClientCapabilities (Just True)
+      runSessionWithCapabilities caps "hie --lsp" "test/data/renamePass" $ return ()
 
   describe "replay session" $ do
     it "passes a test" $
@@ -58,8 +74,36 @@ main = hspec $ do
           fooSymbol ^. name `shouldBe` "foo"
           fooSymbol ^. kind `shouldBe` SkFunction
 
+  describe "text document state" $
+    it "sends back didChange notifications" $
+      runSession "hie --lsp" "test/data/refactor" $ do
+        doc <- openDoc "Main.hs" "haskell"
+
+        let args = toJSON $ AOP (doc ^. uri)
+                                (Position 1 14)
+                                "Redundant bracket"
+            reqParams = ExecuteCommandParams "applyrefact:applyOne" (Just (List [args]))
+        sendRequest WorkspaceExecuteCommand reqParams
+        skipMany anyNotification
+        _ <- response :: Session ExecuteCommandResponse
+
+        editReq <- request :: Session ApplyWorkspaceEditRequest
+        liftIO $ do
+          let (Just cs) = editReq ^. params . edit . changes
+              [(u, List es)] = HM.toList cs
+          u `shouldBe` doc ^. uri
+          es `shouldBe` [TextEdit (Range (Position 1 0) (Position 1 18)) "main = return 42"]
+
+        checkNoDiagnostics
+
   parsingSpec
 
+data ApplyOneParams = AOP
+  { file      :: Uri
+  , start_pos :: Position
+  , hintTitle :: String
+  } deriving (Generic, ToJSON)
+
 checkNoDiagnostics :: Session ()
 checkNoDiagnostics = do
   diagsNot <- notification :: Session PublishDiagnosticsNotification