pytest-lsp: Add checks for window/workDoneProgress/create

swyddfa · Nov 12, 2023 · 021b9a5 · 021b9a5
1 parent 57bf078
commit 021b9a5
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/lib/pytest-lsp/pytest_lsp/checks.py b/lib/pytest-lsp/pytest_lsp/checks.py
@@ -245,6 +245,17 @@ def document_links(
             assert tooltip_support, "Client does not support tooltips."
 
 
+@check_params_of(method=types.WINDOW_WORK_DONE_PROGRESS_CREATE)
+def work_done_progress_create(
+    capabilities: types.ClientCapabilities,
+    params: types.WorkDoneProgressCreateParams,
+):
+    """Assert that the client has support for ``window/workDoneProgress/create``
+    requests."""
+    is_supported = get_capability(capabilities, "window.workDoneProgress", False)
+    assert is_supported, "Client does not support 'window/workDoneProgress/create'"
+
+
 @check_params_of(method=types.WORKSPACE_CONFIGURATION)
 def workspace_configuration(
     capabilities: types.ClientCapabilities,

diff --git a/lib/pytest-lsp/tests/test_checks.py b/lib/pytest-lsp/tests/test_checks.py
@@ -14,6 +14,14 @@
 @pytest.mark.parametrize(
     "capabilities,method,params,expected",
     [
+        (
+            types.ClientCapabilities(
+                window=types.WindowClientCapabilities(work_done_progress=False)
+            ),
+            types.WINDOW_WORK_DONE_PROGRESS_CREATE,
+            types.WorkDoneProgressCreateParams(token="id-123"),
+            "does not support 'window/workDoneProgress/create'",
+        ),
         (
             types.ClientCapabilities(
                 workspace=types.WorkspaceClientCapabilities(configuration=False)