Attempt to fix mypy warnings for mock in conftest

Author: krassowski

python_packages/jupyter_lsp/jupyter_lsp/tests/conftest.py

@@ -1,10 +1,11 @@
 import json
 import pathlib
 import shutil
-from typing import Text
+from typing import Any, Dict, Union
 
 from jupyter_server.serverapp import ServerApp
 from pytest import fixture
+from tornado.concurrent import Future
 from tornado.httputil import HTTPServerRequest
 from tornado.queues import Queue
 from tornado.web import Application
@@ -131,9 +132,12 @@ def initialize(self, manager):
         self._messages_wrote = Queue()
         self._ping_sent = False
 
-    def write_message(self, message: Text) -> None:
+    def write_message(
+        self, message: Union[bytes, str, Dict[str, Any]], binary: bool = False
+    ) -> Future[None]:
         self.log.warning("write_message %s", message)
         self._messages_wrote.put_nowait(message)
+        return Future()
 
     def send_ping(self):
         self._ping_sent = True