Files
jedi/test/test_inference/test_buildout_detection.py
2020-01-22 23:14:07 +01:00

88 lines
2.4 KiB
Python

import os
from textwrap import dedent
from jedi._compatibility import force_unicode
from jedi.inference.sys_path import _get_parent_dir_with_file, \
_get_buildout_script_paths, check_sys_path_modifications
from ..helpers import cwd_at
def check_module_test(Script, code):
module_context = Script(code)._get_module_context()
return check_sys_path_modifications(module_context)
@cwd_at('test/examples/buildout_project/src/proj_name')
def test_parent_dir_with_file(Script):
parent = _get_parent_dir_with_file(
os.path.abspath(os.curdir), 'buildout.cfg')
assert parent is not None
assert parent.endswith(os.path.join('test', 'examples', 'buildout_project'))
@cwd_at('test/examples/buildout_project/src/proj_name')
def test_buildout_detection(Script):
scripts = list(_get_buildout_script_paths(os.path.abspath('./module_name.py')))
assert len(scripts) == 1
curdir = os.path.abspath(os.curdir)
appdir_path = os.path.normpath(os.path.join(curdir, '../../bin/app'))
assert scripts[0] == appdir_path
def test_append_on_non_sys_path(Script):
code = dedent("""
class Dummy(object):
path = []
d = Dummy()
d.path.append('foo')""")
paths = check_module_test(Script, code)
assert not paths
assert 'foo' not in paths
def test_path_from_invalid_sys_path_assignment(Script):
code = dedent("""
import sys
sys.path = 'invalid'""")
paths = check_module_test(Script, code)
assert not paths
assert 'invalid' not in paths
@cwd_at('test/examples/buildout_project/src/proj_name/')
def test_sys_path_with_modifications(Script):
code = dedent("""
import os
""")
path = os.path.abspath(os.path.join(os.curdir, 'module_name.py'))
paths = Script(code, path=path)._inference_state.get_sys_path()
assert '/tmp/.buildout/eggs/important_package.egg' in paths
def test_path_from_sys_path_assignment(Script):
code = dedent("""
#!/usr/bin/python
import sys
sys.path[0:0] = [
'/usr/lib/python3.4/site-packages',
'/home/test/.buildout/eggs/important_package.egg'
]
path[0:0] = [1]
import important_package
if __name__ == '__main__':
sys.exit(important_package.main())""")
paths = check_module_test(Script, code)
paths = list(map(force_unicode, paths))
assert 1 not in paths
assert '/home/test/.buildout/eggs/important_package.egg' in paths