1 files changed, 68 insertions, 0 deletions
diff --git a/tests/lsp/test_diagnostics.py b/tests/lsp/test_diagnostics.py
new file mode 100644
index 0000000..c420942
--- /dev/null
+++ b/tests/lsp/test_diagnostics.py
@@ -0,0 +1,68 @@
+############################################################################
+# Copyright(c) Open Law Library. All rights reserved.                      #
+# See ThirdPartyNotices.txt in the project root for additional notices.    #
+#                                                                          #
+# Licensed under the Apache License, Version 2.0 (the "License")           #
+# you may not use this file except in compliance with the License.         #
+# You may obtain a copy of the License at                                  #
+#                                                                          #
+#     http: // www.apache.org/licenses/LICENSE-2.0                         #
+#                                                                          #
+# Unless required by applicable law or agreed to in writing, software      #
+# distributed under the License is distributed on an "AS IS" BASIS,        #
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. #
+# See the License for the specific language governing permissions and      #
+# limitations under the License.                                           #
+############################################################################
+import json
+from typing import Tuple
+
+from lsprotocol import types
+
+
+from ..client import LanguageClient
+
+
+async def test_diagnostics(
+    json_server_client: Tuple[LanguageClient, types.InitializeResult],
+    uri_for,
+):
+    """Ensure that diagnostics are working as expected."""
+    client, _ = json_server_client
+
+    test_uri = uri_for("example.json")
+    assert test_uri is not None
+
+    # Get the expected error message
+    document_content = "text"
+    try:
+        json.loads(document_content)
+    except json.JSONDecodeError as err:
+        expected_message = err.msg
+
+    client.text_document_did_open(
+        types.DidOpenTextDocumentParams(
+            text_document=types.TextDocumentItem(
+                uri=test_uri, language_id="json", version=1, text=document_content
+            )
+        )
+    )
+
+    await client.wait_for_notification(types.TEXT_DOCUMENT_PUBLISH_DIAGNOSTICS)
+
+    diagnostics = client.diagnostics[test_uri]
+    assert diagnostics[0].message == expected_message
+
+    result = await client.text_document_diagnostic_async(
+        types.DocumentDiagnosticParams(
+            text_document=types.TextDocumentIdentifier(test_uri)
+        )
+    )
+    diagnostics = result.items
+    assert diagnostics[0].message == expected_message
+
+    workspace_result = await client.workspace_diagnostic_async(
+        types.WorkspaceDiagnosticParams(previous_result_ids=[])
+    )
+    diagnostics = workspace_result.items[0].items
+    assert diagnostics[0].message == expected_message