third_party.pylibs.pylint.src/test/unittest_checker_variables.py

100 lines
3.3 KiB
Python

"""Unit tests for the variables checker."""
import sys
import os
import unittest
from astroid import test_utils
from pylint.checkers import variables
from pylint.testutils import CheckerTestCase, linter, set_config, Message
class VariablesCheckerTC(CheckerTestCase):
CHECKER_CLASS = variables.VariablesChecker
def test_bitbucket_issue_78(self):
""" Issue 78 report a false positive for unused-module """
module = test_utils.build_module("""
from sys import path
path += ['stuff']
def func():
other = 1
return len(other)
""")
with self.assertNoMessages():
self.walk(module)
@set_config(ignored_modules=('argparse',))
def test_no_name_in_module_skipped(self):
"""Make sure that 'from ... import ...' does not emit a
'no-name-in-module' with a module that is configured
to be ignored.
"""
node = test_utils.extract_node("""
from argparse import THIS_does_not_EXIST
""")
with self.assertNoMessages():
self.checker.visit_from(node)
@set_config(callbacks=('callback_', '_callback'))
def test_custom_callback_string(self):
""" Test the --calbacks option works. """
def cleanup():
self.checker._to_consume = _to_consume
_to_consume = self.checker._to_consume
self.checker._to_consume = []
self.addCleanup(cleanup)
node = test_utils.extract_node("""
def callback_one(abc):
''' should not emit unused-argument. '''
""")
with self.assertNoMessages():
self.checker.visit_function(node)
self.checker.leave_function(node)
node = test_utils.extract_node("""
def two_callback(abc, defg):
''' should not emit unused-argument. '''
""")
with self.assertNoMessages():
self.checker.visit_function(node)
self.checker.leave_function(node)
node = test_utils.extract_node("""
def normal_func(abc):
''' should emit unused-argument. '''
""")
with self.assertAddsMessages(
Message('unused-argument', node=node['abc'], args='abc')):
self.checker.visit_function(node)
self.checker.leave_function(node)
node = test_utils.extract_node("""
def cb_func(abc):
''' Previous callbacks are overriden. '''
""")
with self.assertAddsMessages(
Message('unused-argument', node=node['abc'], args='abc')):
self.checker.visit_function(node)
self.checker.leave_function(node)
class MissingSubmoduleTest(CheckerTestCase):
CHECKER_CLASS = variables.VariablesChecker
def test_package_all(self):
regr_data = os.path.join(os.path.dirname(os.path.abspath(__file__)),
'regrtest_data')
sys.path.insert(0, regr_data)
try:
linter.check(os.path.join(regr_data, 'package_all'))
got = linter.reporter.finalize().strip()
self.assertEqual(got, "E: 3: Undefined variable name "
"'missing' in __all__")
finally:
sys.path.pop(0)
if __name__ == '__main__':
unittest.main()