File: test_pyflakes_lint.py

package info (click to toggle)
python-lsp-server 1.12.0-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 796 kB
  • sloc: python: 7,791; sh: 12; makefile: 4
file content (69 lines) | stat: -rw-r--r-- 1,903 bytes parent folder | download
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
69
# Copyright 2017-2020 Palantir Technologies, Inc.
# Copyright 2021- Python Language Server Contributors.

import sys

from pylsp import lsp, uris
from pylsp.plugins import pyflakes_lint
from pylsp.workspace import Document

DOC_URI = uris.from_fs_path(__file__)
DOC = """import sys

def hello():
\tpass

import json
"""

DOC_SYNTAX_ERR = """def hello()
    pass
"""

DOC_UNDEFINED_NAME_ERR = "a = b"


DOC_ENCODING = """# encoding=utf-8
import sys
"""


def test_pyflakes(workspace) -> None:
    doc = Document(DOC_URI, workspace, DOC)
    diags = pyflakes_lint.pylsp_lint(workspace, doc)

    # One we're expecting is:
    msg = "'sys' imported but unused"
    unused_import = [d for d in diags if d["message"] == msg][0]

    assert unused_import["range"]["start"] == {"line": 0, "character": 0}
    assert unused_import["severity"] == lsp.DiagnosticSeverity.Warning


def test_syntax_error_pyflakes(workspace) -> None:
    doc = Document(DOC_URI, workspace, DOC_SYNTAX_ERR)
    diag = pyflakes_lint.pylsp_lint(workspace, doc)[0]

    if sys.version_info[:2] >= (3, 10):
        assert diag["message"] == "expected ':'"
    else:
        assert diag["message"] == "invalid syntax"
    assert diag["range"]["start"] == {"line": 0, "character": 12}
    assert diag["severity"] == lsp.DiagnosticSeverity.Error


def test_undefined_name_pyflakes(workspace) -> None:
    doc = Document(DOC_URI, workspace, DOC_UNDEFINED_NAME_ERR)
    diag = pyflakes_lint.pylsp_lint(workspace, doc)[0]

    assert diag["message"] == "undefined name 'b'"
    assert diag["range"]["start"] == {"line": 0, "character": 4}
    assert diag["severity"] == lsp.DiagnosticSeverity.Error


def test_unicode_encoding(workspace) -> None:
    doc = Document(DOC_URI, workspace, DOC_ENCODING)
    diags = pyflakes_lint.pylsp_lint(workspace, doc)

    assert len(diags) == 1
    assert diags[0]["message"] == "'sys' imported but unused"