Fix expectNoMoreDiagnostics

Needs also lukel97/lsp-test#74
haskell · Sep 6, 2020 · 1972d71 · 1972d71
1 parent 3999c3e
commit 1972d71
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/src/Development/IDE/Test.hs b/test/src/Development/IDE/Test.hs
@@ -70,9 +70,9 @@ expectNoMoreDiagnostics timeout = do
             "Got unexpected diagnostics for " <> show fileUri <>
             " got " <> show actual
     handleCustomMethodResponse =
-        -- the CustomClientMethod triggers a log message about ignoring it
+        -- the CustomClientMethod triggers a RspCustomServer
         -- handle that and then exit
-        void (LspTest.message :: Session LogMessageNotification)
+        void (LspTest.message :: Session CustomResponse)
     ignoreOthers = void anyMessage >> handleMessages
 
 expectDiagnostics :: [(FilePath, [(DiagnosticSeverity, Cursor, T.Text)])] -> Session ()