1 files changed, 99 insertions, 0 deletions
diff --git a/pylint/test/unittest_checker_variables.py b/pylint/test/unittest_checker_variables.py
new file mode 100644
index 0000000..e3dd939
--- /dev/null
+++ b/pylint/test/unittest_checker_variables.py
@@ -0,0 +1,99 @@
+"""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()