Fix embarassing error with publishDiagnosticsNotification

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

diff --git a/test/Test.hs b/test/Test.hs
index 604ea1c834978c7432b5eeb0db92ae2d3d49e301..652485e0af160279340ea8d6730dc37d33b075e9 100644 (file)
--- a/test/Test.hs
+++ b/test/Test.hs
@@ -1,34 +1,54 @@
 {-# LANGUAGE DuplicateRecordFields #-}
 {-# LANGUAGE OverloadedStrings #-}
 import           Test.Hspec
+import           Data.Maybe
 import           Data.Proxy
 import           Control.Monad.IO.Class
 import           Control.Lens hiding (List)
 import           Language.Haskell.LSP.Test
-import           Language.Haskell.LSP.TH.DataTypesJSON
+import           Language.Haskell.LSP.Test.Replay
+import           Language.Haskell.LSP.Types
+import           Language.Haskell.LSP.Messages
+import           ParsingTests
 
-main = hspec $
-  describe "manual session validation" $ 
+main = hspec $ do
+  describe "manual session validation" $ do
     it "passes a test" $
       runSession "test/recordings/renamePass" $ do
-        docItem <- getDocItem "Desktop/simple.hs" "haskell"
-        docId   <- TextDocumentIdentifier <$> getDocUri "Desktop/simple.hs"
+        doc <- openDoc "Desktop/simple.hs" "haskell"
 
-        sendNotification TextDocumentDidOpen (DidOpenTextDocumentParams docItem)
+        skipMany loggingNotification
 
-        (NotificationMessage _ TextDocumentPublishDiagnostics (PublishDiagnosticsParams _ (List diags))) <-
-          getMessage :: Session PublishDiagnosticsNotification
+        NotPublishDiagnostics diagsNot <- notification
 
-        liftIO $ diags `shouldBe` []
+        liftIO $ diagsNot ^. params . diagnostics `shouldBe` List []
         
         sendRequest (Proxy :: Proxy DocumentSymbolRequest)
                     TextDocumentDocumentSymbol
-                    (DocumentSymbolParams docId)
+                    (DocumentSymbolParams doc)
+
+        RspDocumentSymbols rspSymbols <- response
         
-        (ResponseMessage _ _ (Just (List symbols)) Nothing) <- getMessage :: Session DocumentSymbolsResponse
         liftIO $ do
-          let mainSymbol = head symbols
+          let (List symbols) = fromJust (rspSymbols ^. result)
+              mainSymbol = head symbols
           mainSymbol ^. name `shouldBe` "main"
           mainSymbol ^. kind `shouldBe` SkFunction
           mainSymbol ^. location . range `shouldBe` Range (Position 3 0) (Position 3 4)
           mainSymbol ^. containerName `shouldBe` Nothing
+    
+    it "fails a test" $
+      -- TODO: Catch the exception in haskell-lsp-test and provide nicer output
+      let session = runSession "test/recordings/renamePass" $ do
+                    openDoc "Desktop/simple.hs" "haskell"
+                    skipMany loggingNotification
+                    request
+        in session `shouldThrow` anyException
+  
+  describe "replay session" $ do
+    it "passes a test" $
+      replaySession "test/recordings/renamePass" `shouldReturn` True
+    it "fails a test" $
+      replaySession "test/recordings/renameFail" `shouldReturn` False
+  
+  parsingSpec
\ No newline at end of file