Add SessionConfig

[lsp-test.git] / test / Test.hs

diff --git a/test/Test.hs b/test/Test.hs
index 8f715333092c46b503316b77b0163d8596b9fa18..d9ecf219e7779794e39bb2fa053f264e186b704a 100644 (file)
--- a/test/Test.hs
+++ b/test/Test.hs
@@ -4,14 +4,17 @@
 {-# LANGUAGE DeriveAnyClass #-}
 import           Test.Hspec
 import           Data.Aeson
+import           Data.Default
 import qualified Data.HashMap.Strict as HM
 import           Data.Maybe
+import           Control.Concurrent
 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.Types
+import           Language.Haskell.LSP.TH.ClientCapabilities
+import           Language.Haskell.LSP.Types hiding (capabilities)
 import           ParsingTests
 
 main = hspec $ do
@@ -45,6 +48,25 @@ main = hspec $ 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)
+          conf = def { capabilities = caps }
+      runSessionWithConfig conf "hie --lsp" "test/data/renamePass" $ return ()
+    
+    it "times out" $
+      let sesh = runSessionWithConfig (def {timeout = 10}) "hie --lsp" "test/data/renamePass" $ do
+              skipMany loggingNotification
+              _ <- request :: Session ApplyWorkspaceEditRequest
+              return ()
+      in sesh `shouldThrow` anySessionException
+    
+    it "doesn't time out" $ runSessionWithConfig (def {timeout = 10}) "hie --lsp" "test/data/renamePass" $ do
+      loggingNotification
+      liftIO $ threadDelay 5
+
+
   describe "replay session" $ do
     it "passes a test" $
       replaySession "hie --lsp" "test/data/renamePass" `shouldReturn` True
@@ -88,6 +110,9 @@ main = hspec $ do
 
         checkNoDiagnostics
 
+        contents <- documentContents doc
+        liftIO $ contents `shouldBe` "main :: IO Int\nmain = return 42"
+
   parsingSpec
 
 data ApplyOneParams = AOP