""" Tests for :attr:`.BaseDefinition.full_name`. There are three kinds of test: #. Test classes derived from :class:`MixinTestFullName`. Child class defines :attr:`.operation` to alter how the api definition instance is created. #. :class:`TestFullDefinedName` is to test combination of ``obj.full_name`` and ``jedi.defined_names``. #. Misc single-function tests. """ import textwrap import pytest import jedi from ..helpers import TestCase class MixinTestFullName(object): operation = None def check(self, source, desired): script = jedi.Script(textwrap.dedent(source)) definitions = getattr(script, type(self).operation)() for d in definitions: self.assertEqual(d.full_name, desired) def test_os_path_join(self): self.check('import os; os.path.join', 'os.path.join') def test_builtin(self): self.check('TypeError', 'TypeError') class TestFullNameWithGotoDefinitions(MixinTestFullName, TestCase): operation = 'goto_definitions' @pytest.mark.skipif('sys.version_info[0] < 3', reason='Python 2 also yields None.') def test_tuple_mapping(self): self.check(""" import re any_re = re.compile('.*') any_re""", '_sre.SRE_Pattern') def test_from_import(self): self.check('from os import path', 'os.path') class TestFullNameWithCompletions(MixinTestFullName, TestCase): operation = 'completions' class TestFullDefinedName(TestCase): """ Test combination of ``obj.full_name`` and ``jedi.defined_names``. """ def check(self, source, desired): definitions = jedi.names(textwrap.dedent(source)) full_names = [d.full_name for d in definitions] self.assertEqual(full_names, desired) def test_local_names(self): self.check(""" def f(): pass class C: pass """, ['__main__.f', '__main__.C']) def test_imports(self): self.check(""" import os from os import path from os.path import join from os import path as opath """, ['os', 'os.path', 'os.path.join', 'os.path']) def test_sub_module(): """ ``full_name needs to check sys.path to actually find it's real path module path. """ defs = jedi.Script('from jedi.api import classes; classes').goto_definitions() assert [d.full_name for d in defs] == ['jedi.api.classes'] defs = jedi.Script('import jedi.api; jedi.api').goto_definitions() assert [d.full_name for d in defs] == ['jedi.api'] def test_os_path(): d, = jedi.Script('from os.path import join').completions() assert d.full_name == 'os.path.join' d, = jedi.Script('import os.p').completions() assert d.full_name == 'os.path' def test_os_issues(): """Issue #873""" c, = jedi.Script('import os\nos.nt''').completions() assert c.full_name == 'nt'