mirror of
https://fuchsia.googlesource.com/third_party/github.com/pylint-dev/pylint
synced 2024-09-21 16:19:21 +00:00
7c06fe8641
This makes unittests with options mimick the normal option setting behaviour. It also adds ``set_config_directly()`` for those cases where this is not possible.
67 lines
2.4 KiB
Python
67 lines
2.4 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
|
|
|
|
import functools
|
|
import optparse # pylint: disable=deprecated-module
|
|
|
|
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():
|
|
self.checker.set_option(
|
|
key.replace("_", "-"),
|
|
value,
|
|
optdict=dict(PyLinter.make_options()),
|
|
)
|
|
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.
|
|
"""
|
|
|
|
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
|