test_evaluate -> test_inference

test/test_inference/test_buildout_detection.py

@@ -0,0 +1,91 @@
+import os
+from textwrap import dedent
+
+from jedi._compatibility import force_unicode
+from jedi.evaluate.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()
+    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)._evaluator.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