Application: gvSIG desktop » gvSIG scripting

"""Unit tests for the variables checker."""

import sys

import os

import unittest

import astroid

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 = astroid.parse("""

        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_importfrom(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_functiondef(node)

            self.checker.leave_functiondef(node)

        node = test_utils.extract_node("""

        def two_callback(abc, defg):

             ''' should not emit unused-argument. '''

        """)

        with self.assertNoMessages():

            self.checker.visit_functiondef(node)

            self.checker.leave_functiondef(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_functiondef(node)

            self.checker.leave_functiondef(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_functiondef(node)

            self.checker.leave_functiondef(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()