summaryrefslogtreecommitdiffstats
path: root/tests/lsp/test_diagnostics.py
blob: c420942a95dc453b9da50bcc20f0636ae63dea58 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
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