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
70
71
72
73
74
75
76
77
78
79
80
|
# Licensed under the GPL: https://www.gnu.org/licenses/old-licenses/gpl-2.0.html
# For details: https://github.com/PyCQA/pylint/blob/main/LICENSE
# Copyright (c) https://github.com/PyCQA/pylint/blob/main/CONTRIBUTORS.txt
from __future__ import annotations
import warnings
import astroid
from astroid import nodes
from pylint.checkers.base_checker import BaseChecker
from pylint.checkers.utils import only_required_for_messages
from pylint.utils import ASTWalker
class TestASTWalker:
class MockLinter:
def __init__(self, msgs: dict[str, bool]) -> None:
self._msgs = msgs
def is_message_enabled(self, msgid: str) -> bool:
return self._msgs.get(msgid, True)
class Checker(BaseChecker):
# pylint: disable-next=super-init-not-called
def __init__(self) -> None:
self.called: set[str] = set()
@only_required_for_messages("first-message")
def visit_module(
self, module: nodes.Module # pylint: disable=unused-argument
) -> None:
self.called.add("module")
@only_required_for_messages("second-message")
def visit_call(self, module: nodes.Call) -> None:
raise NotImplementedError
@only_required_for_messages("second-message", "third-message")
def visit_assignname(
self, module: nodes.AssignName # pylint: disable=unused-argument
) -> None:
self.called.add("assignname")
@only_required_for_messages("second-message")
def leave_assignname(self, module: nodes.AssignName) -> None:
raise NotImplementedError
def test_only_required_for_messages(self) -> None:
linter = self.MockLinter(
{"first-message": True, "second-message": False, "third-message": True}
)
walker = ASTWalker(linter) # type: ignore[arg-type]
checker = self.Checker()
walker.add_checker(checker)
walker.walk(astroid.parse("x = func()"))
assert {"module", "assignname"} == checker.called
def test_deprecated_methods(self) -> None:
class Checker(BaseChecker):
# pylint: disable-next=super-init-not-called
def __init__(self) -> None:
self.called = False
@only_required_for_messages("first-message")
def visit_assname(
self, node: nodes.AssignName # pylint: disable=unused-argument
) -> None:
self.called = True
linter = self.MockLinter({"first-message": True})
walker = ASTWalker(linter) # type: ignore[arg-type]
checker = Checker()
walker.add_checker(checker)
with warnings.catch_warnings(record=True):
warnings.simplefilter("always")
walker.walk(astroid.parse("x = 1"))
assert not checker.called
|
