X-Git-Url: http://git.lukelau.me/?a=blobdiff_plain;f=test%2FTest.hs;h=13e7ccc5d3066b687c7ca4391a3e6e8af62cbe77;hb=dee96dc60f35d20adc03294acd14ba6bc22bb6c8;hp=401fd9815fd09e81db4f3af8dafeff13ebc2f2ad;hpb=d8ce5332e9ac2dc24bf1490f03bd30abb17196e8;p=opengl.git

diff --git a/test/Test.hs b/test/Test.hs
index 401fd98..13e7ccc 100644
--- a/test/Test.hs
+++ b/test/Test.hs
@@ -1,6 +1,7 @@
 {-# LANGUAGE DuplicateRecordFields #-}
 {-# LANGUAGE OverloadedStrings #-}
 import           Test.Hspec
+import           Data.Maybe
 import           Data.Proxy
 import           Control.Monad.IO.Class
 import           Control.Lens hiding (List)
@@ -20,18 +21,19 @@ main = hspec $ do
 
         skipMany loggingNotification
 
-        (NotPublishDiagnostics (NotificationMessage _ TextDocumentPublishDiagnostics (PublishDiagnosticsParams _ (List diags)))) <- notification
+        NotPublishDiagnostics diagsNot <- notification
 
-        liftIO $ diags `shouldBe` []
+        liftIO $ diagsNot ^. params . diagnostics `shouldBe` List []
         
         sendRequest (Proxy :: Proxy DocumentSymbolRequest)
                     TextDocumentDocumentSymbol
                     (DocumentSymbolParams docId)
 
-        (RspDocumentSymbols (ResponseMessage _ _ (Just (List symbols)) Nothing)) <- response
+        RspDocumentSymbols rspSymbols <- response
         
         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)