third_party.pylibs.pylint.src/pylint/testutils/decorator.py
Pierre Sassoulas 9e0baf370a Simplify hard to maintain copyright notice
git is the source of truth for the copyright, copyrite (the tool)
was taking exponentially longer with each release, and it's
polluting the code with sometime as much as 50 lines of names.
2022-03-24 13:06:15 +01:00

90 lines
3.5 KiB
Python

# 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/graphs/contributors
import functools
import optparse # pylint: disable=deprecated-module
import warnings
from pylint.lint import PyLinter
from pylint.testutils.checker_test_case import CheckerTestCase
def set_config(**kwargs):
"""Decorator for setting config values on a checker.
Passing the args and kwargs back to the test function itself
allows this decorator to be used on parametrized test cases.
"""
def _wrapper(fun):
@functools.wraps(fun)
def _forward(self, *args, **test_function_kwargs):
try:
for key, value in kwargs.items():
self.checker.set_option(key.replace("_", "-"), value)
except optparse.OptionError:
# Check if option is one of the base options of the PyLinter class
for key, value in kwargs.items():
try:
self.checker.set_option(
key.replace("_", "-"),
value,
optdict=dict(PyLinter.make_options())[
key.replace("_", "-")
],
)
except KeyError:
# pylint: disable-next=fixme
# TODO: Find good way to double load checkers in unittests
# When options are used by multiple checkers we need to load both of them
# to be able to get an optdict
self.checker.set_option(
key.replace("_", "-"),
value,
optdict={},
)
if isinstance(self, CheckerTestCase):
# reopen checker in case, it may be interested in configuration change
self.checker.open()
fun(self, *args, **test_function_kwargs)
return _forward
return _wrapper
def set_config_directly(**kwargs):
"""Decorator for setting config values on a checker without validation.
Some options should be declared in two different checkers. This is
impossible without duplicating the option key. For example:
"no-docstring-rgx" in DocstringParameterChecker & DocStringChecker
This decorator allows to directly set such options.
Passing the args and kwargs back to the test function itself
allows this decorator to be used on parametrized test cases.
"""
# pylint: disable=fixme
# TODO: Remove this function in 2.14
warnings.warn(
"The set_config_directly decorator will be removed in 2.14. To decorate "
"unittests you can use set_config. If this causes a duplicate KeyError "
"you can consider writing the tests using the functional test framework.",
DeprecationWarning,
)
def _wrapper(fun):
@functools.wraps(fun)
def _forward(self, *args, **test_function_kwargs):
for key, value in kwargs.items():
setattr(self.checker.config, key, value)
if isinstance(self, CheckerTestCase):
# reopen checker in case, it may be interested in configuration change
self.checker.open()
fun(self, *args, **test_function_kwargs)
return _forward
return _wrapper