set notebook server capabilities

Author: seeM

jedi_language_server/server.py

@@ -63,6 +63,9 @@
     MarkupContent,
     MarkupKind,
     MessageType,
+    NotebookDocumentSyncOptions,
+    NotebookDocumentSyncOptionsNotebookSelectorType2,
+    NotebookDocumentSyncOptionsNotebookSelectorType2CellsType,
     ParameterInformation,
     RenameParams,
     SignatureHelp,
@@ -207,6 +210,18 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
     name="jedi-language-server",
     version=__version__,
     protocol_cls=JediLanguageServerProtocol,
+    # Advertise support for Python notebook cells.
+    notebook_document_sync=NotebookDocumentSyncOptions(
+        notebook_selector=[
+            NotebookDocumentSyncOptionsNotebookSelectorType2(
+                cells=[
+                    NotebookDocumentSyncOptionsNotebookSelectorType2CellsType(
+                        language="python"
+                    )
+                ]
+            )
+        ]
+    ),
 )
 
 

tests/lsp_tests/test_init_options.py renamed to tests/lsp_tests/test_initialize.py

@@ -1,9 +1,9 @@
-"""Tests for initialization options over language server protocol."""
+"""Tests for initialize requests."""
 
 import copy
 from threading import Event
 
-from hamcrest import assert_that
+from hamcrest import assert_that, is_
 
 from tests.lsp_test_client import defaults, session
 
@@ -51,3 +51,30 @@ def _window_log_message_handler(params):
                     "$.diagnostics']"
                 )
             )
+
+
+def test_notebook_server_capabilities() -> None:
+    """Test that the server's notebook capabilities are set correctly."""
+    actual = []
+
+    def _process_server_capabilities(capabilities):
+        actual.append(capabilities)
+
+    initialize_params = copy.deepcopy(defaults.VSCODE_DEFAULT_INITIALIZE)
+    initialize_params["capabilities"]["notebookDocument"] = {
+        "synchronization": {
+            "dynamicRegistration": True,
+            "executionSummarySupport": True,
+        },
+    }
+    with session.LspSession() as ls_session:
+        ls_session.initialize(initialize_params, _process_server_capabilities)
+
+        assert_that(len(actual) == 1)
+        for params in actual:
+            assert_that(
+                params["capabilities"]["notebookDocumentSync"],
+                is_(
+                    {"notebookSelector": [{"cells": [{"language": "python"}]}]}
+                ),
+            )