X-Git-Url: http://git.lukelau.me/?a=blobdiff_plain;f=test%2FTest.hs;h=c8d6072f6ec386a88185937e2091dbaac12ab32d;hb=493d20ada6e48a8987e00a5ec92a1b31fe3c9b8c;hp=0bffa5e1a7c3f7a23d40205887b9216441bf5454;hpb=5d58e92cc411d9f89f55737b6a225075e77f4f6b;p=opengl.git

diff --git a/test/Test.hs b/test/Test.hs
index 0bffa5e..c8d6072 100644
--- a/test/Test.hs
+++ b/test/Test.hs
@@ -7,13 +7,15 @@ 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.Messages
 import           Language.Haskell.LSP.Test
 import           Language.Haskell.LSP.Test.Replay
 import           Language.Haskell.LSP.TH.ClientCapabilities
-import           Language.Haskell.LSP.Types
+import           Language.Haskell.LSP.Types hiding (capabilities)
 import           ParsingTests
 
 main = hspec $ do
@@ -51,7 +53,36 @@ main = hspec $ do
       let caps = def { _workspace = Just workspaceCaps }
           workspaceCaps = def { _didChangeConfiguration = Just configCaps }
           configCaps = DidChangeConfigurationClientCapabilities (Just True)
-      runSessionWithCapabilities caps "hie --lsp" "test/data/renamePass" $ return ()
+          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 = 10}) "hie --lsp" "test/data/renamePass" $ do
+        loggingNotification
+        liftIO $ threadDelay 5
+
+      it "throw when there's an unexpected message" $
+        let msgExc (UnexpectedMessageException "Publish diagnostics notification" (NotLogMessage _)) = True
+            msgExc _ = False
+          in runSession "hie --lsp" "test/data/renamePass" publishDiagnosticsNotification `shouldThrow` msgExc
+      
+      it "throw when there's an unexpected message 2" $
+        let msgExc (UnexpectedMessageException "Response" (NotPublishDiagnostics _)) = True
+            msgExc _ = 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` msgExc
 
   describe "replay session" $ do
     it "passes a test" $
@@ -96,6 +127,9 @@ main = hspec $ do
 
         checkNoDiagnostics
 
+        contents <- documentContents doc
+        liftIO $ contents `shouldBe` "main :: IO Int\nmain = return 42"
+
   parsingSpec
 
 data ApplyOneParams = AOP