X-Git-Url: http://git.lukelau.me/?a=blobdiff_plain;f=test%2FTest.hs;h=79bb30312668857a68322a99a3ad46a34896015f;hb=25c948d3bfc9f3876805377a0b923570bd0330a0;hp=652485e0af160279340ea8d6730dc37d33b075e9;hpb=88c70a40654c7152fb50b2a4e171fbdc00324f51;p=opengl.git

diff --git a/test/Test.hs b/test/Test.hs
index 652485e..79bb303 100644
--- a/test/Test.hs
+++ b/test/Test.hs
@@ -1,33 +1,35 @@
 {-# 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           Data.Proxy
+import           Control.Concurrent
 import           Control.Monad.IO.Class
 import           Control.Lens hiding (List)
+import           GHC.Generics
+import           Language.Haskell.LSP.Messages
 import           Language.Haskell.LSP.Test
 import           Language.Haskell.LSP.Test.Replay
-import           Language.Haskell.LSP.Types
-import           Language.Haskell.LSP.Messages
+import           Language.Haskell.LSP.TH.ClientCapabilities
+import           Language.Haskell.LSP.Types hiding (capabilities)
 import           ParsingTests
 
 main = hspec $ do
-  describe "manual session validation" $ do
+  describe "manual session" $ do
     it "passes a test" $
-      runSession "test/recordings/renamePass" $ do
+      runSession "hie --lsp" "test/data/renamePass" $ do
         doc <- openDoc "Desktop/simple.hs" "haskell"
 
         skipMany loggingNotification
 
-        NotPublishDiagnostics diagsNot <- notification
-
-        liftIO $ diagsNot ^. params . diagnostics `shouldBe` List []
+        noDiagnostics
 
-        sendRequest (Proxy :: Proxy DocumentSymbolRequest)
-                    TextDocumentDocumentSymbol
-                    (DocumentSymbolParams doc)
-
-        RspDocumentSymbols rspSymbols <- response
+        rspSymbols <- getDocumentSymbols doc
 
         liftIO $ do
           let (List symbols) = fromJust (rspSymbols ^. result)
@@ -39,16 +41,122 @@ main = hspec $ do
 
     it "fails a test" $
       -- TODO: Catch the exception in haskell-lsp-test and provide nicer output
-      let session = runSession "test/recordings/renamePass" $ do
+      let session = runSession "hie --lsp" "test/data/renamePass" $ do
                       openDoc "Desktop/simple.hs" "haskell"
                       skipMany loggingNotification
-                    request
+                      anyRequest
         in session `shouldThrow` anyException
+    it "can get initialize response" $ runSession "hie --lsp" "test/data/renamePass" $ do
+      rsp <- initializeResponse
+      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 ()
+
+    describe "exceptions" $ do
+      it "throw on time out" $
+        let sesh = runSessionWithConfig (def {timeout = 10}) "hie --lsp" "test/data/renamePass" $ do
+                skipMany loggingNotification
+                _ <- request :: Session ApplyWorkspaceEditRequest
+                return ()
+        in sesh `shouldThrow` anySessionException
+
+      it "don't throw when no time out" $ runSessionWithConfig (def {timeout = 5}) "hie --lsp" "test/data/renamePass" $ do
+        loggingNotification
+        liftIO $ threadDelay 10
+        _ <- openDoc "Desktop/simple.hs" "haskell"
+        return ()
+
+      it "throw when there's an unexpected message" $
+        let selector (UnexpectedMessageException "Publish diagnostics notification" (NotLogMessage _)) = True
+            selector _ = False
+          in runSession "hie --lsp" "test/data/renamePass" publishDiagnosticsNotification `shouldThrow` selector
+
+      it "throw when there's an unexpected message 2" $
+        let selector (UnexpectedMessageException "Response" (NotPublishDiagnostics _)) = True
+            selector _ = False
+            sesh = do
+              doc <- openDoc "Desktop/simple.hs" "haskell"
+              sendRequest TextDocumentDocumentSymbol (DocumentSymbolParams doc)
+              skipMany anyNotification
+              response :: Session RenameResponse -- the wrong type
+          in runSession "hie --lsp" "test/data/renamePass" sesh
+            `shouldThrow` selector
 
   describe "replay session" $ do
     it "passes a test" $
-      replaySession "test/recordings/renamePass" `shouldReturn` True
+      replaySession "hie --lsp" "test/data/renamePass"
     it "fails a test" $
-      replaySession "test/recordings/renameFail" `shouldReturn` False
+      let selector (ReplayOutOfOrderException _ _) = True
+          selector _ = False
+        in replaySession "hie --lsp" "test/data/renameFail" `shouldThrow` selector
+
+  describe "manual javascript session" $
+    it "passes a test" $
+      runSession "javascript-typescript-stdio" "test/data/javascriptPass" $ do
+        doc <- openDoc "test.js" "javascript"
+
+        noDiagnostics
+
+        rspSymbols <- getDocumentSymbols doc
+
+        let (List symbols) = fromJust (rspSymbols ^. result)
+            fooSymbol = head symbols
+        liftIO $ do
+          fooSymbol ^. name `shouldBe` "foo"
+          fooSymbol ^. kind `shouldBe` SkFunction
+
+  describe "text document VFS" $
+    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"]
+
+        noDiagnostics
+
+        contents <- documentContents doc
+        liftIO $ contents `shouldBe` "main :: IO Int\nmain = return 42"
+    
+  describe "documentEdit" $
+    it "automatically consumes applyedit requests" $
+      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
+
+        contents <- getDocumentEdit doc
+        liftIO $ contents `shouldBe` "main :: IO Int\nmain = return 42"
+        noDiagnostics
 
   parsingSpec
+
+data ApplyOneParams = AOP
+  { file      :: Uri
+  , start_pos :: Position
+  , hintTitle :: String
+  } deriving (Generic, ToJSON)
+