import Control.Exception
import Language.Haskell.LSP.Messages
+import Language.Haskell.LSP.Types
import Data.Aeson
+import Data.Aeson.Encode.Pretty
+import Data.Algorithm.Diff
+import Data.Algorithm.DiffOutput
+import Data.List
import qualified Data.ByteString.Lazy.Char8 as B
-data SessionException = TimeoutException
- | UnexpectedMessageException String FromServerMessage
- | ReplayOutOfOrderException FromServerMessage [FromServerMessage]
- | UnexpectedDiagnosticsException
- | IncorrectApplyEditRequestException String
+-- | An exception that can be thrown during a 'Haskell.LSP.Test.Session.Session'
+data SessionException = Timeout
+ | NoContentLengthHeader
+ | UnexpectedMessage String FromServerMessage
+ | ReplayOutOfOrder FromServerMessage [FromServerMessage]
+ | UnexpectedDiagnostics
+ | IncorrectApplyEditRequest String
+ | UnexpectedResponseError LspIdRsp ResponseError
+ | UnexpectedServerTermination
+ deriving Eq
instance Exception SessionException
instance Show SessionException where
- show TimeoutException = "Timed out waiting to receive a message from the server."
- show (UnexpectedMessageException expected lastMsg) =
+ show Timeout = "Timed out waiting to receive a message from the server."
+ show NoContentLengthHeader = "Couldn't read Content-Length header from the server."
+ show (UnexpectedMessage expected lastMsg) =
"Received an unexpected message from the server:\n" ++
- "Expected: " ++ expected ++ "\n" ++
- "Last message accepted: " ++ show lastMsg
- show (ReplayOutOfOrderException received expected) =
- "Replay is out of order:\n" ++
+ "Was parsing: " ++ expected ++ "\n" ++
+ "Last message received: " ++ show lastMsg
+ show (ReplayOutOfOrder received expected) =
+ let expected' = nub expected
+ getJsonDiff = lines . B.unpack . encodePretty
+ showExp exp = B.unpack (encodePretty exp) ++ "\nDiff:\n" ++
+ ppDiff (getGroupedDiff (getJsonDiff received) (getJsonDiff exp))
+ in "Replay is out of order:\n" ++
-- Print json so its a bit easier to update the session logs
- "Received from server:\n" ++ B.unpack (encode received) ++ "\n" ++
- "Expected one of:\n" ++ unlines (map (B.unpack . encode) expected)
- show UnexpectedDiagnosticsException = "Unexpectedly received diagnostics from the server."
- show (IncorrectApplyEditRequestException msgStr) = "ApplyEditRequest didn't contain document, instead received:\n"
+ "Received from server:\n" ++ B.unpack (encodePretty received) ++ "\n" ++
+ "Raw from server:\n" ++ B.unpack (encode received) ++ "\n" ++
+ "Expected one of:\n" ++ unlines (map showExp expected')
+ show UnexpectedDiagnostics = "Unexpectedly received diagnostics from the server."
+ show (IncorrectApplyEditRequest msgStr) = "ApplyEditRequest didn't contain document, instead received:\n"
++ msgStr
+ show (UnexpectedResponseError lid e) = "Received an exepected error in a response for id " ++ show lid ++ ":\n"
+ ++ show e
+ show UnexpectedServerTermination = "Language server unexpectedly terminated"
+-- | A predicate that matches on any 'SessionException'
anySessionException :: SessionException -> Bool
anySessionException = const True