def test_rename_mod(Script, dir_with_content): script = Script( 'import modx; modx\n', path=os.path.join(dir_with_content, 'some_script.py'), project=medi.Project(dir_with_content), ) refactoring = script.rename(line=1, new_name='modr') refactoring.apply() p1 = os.path.join(dir_with_content, 'modx.py') p2 = os.path.join(dir_with_content, 'modr.py') expected_code = 'import modr\nfoo\n' assert not os.path.exists(p1) with open(p2, newline='') as f: assert f.read() == expected_code assert refactoring.get_renames() == [(p1, p2)] assert refactoring.get_changed_files()[p1].get_new_code() == expected_code assert refactoring.get_diff() == dedent('''\ rename from modx.py rename to modr.py --- modx.py +++ modr.py @@ -1,3 +1,3 @@ -import modx +import modr foo --- some_script.py +++ some_script.py @@ -1,2 +1,2 @@ -import modx; modx +import modr; modr ''').format(dir=dir_with_content)
def refactor(self, environment): project = medi.Project(os.path.join(test_dir, 'refactor')) script = medi.Script(self._code, path=self._path, project=project, environment=environment) refactor_func = getattr(script, self.refactor_type) return refactor_func(self._line_nr, self._index, **self._kwargs)
def test_sub_module(Script, medi_path): """ ``full_name needs to check sys.path to actually find it's real path module path. """ sys_path = [medi_path] project = medi.Project('.', sys_path=sys_path) defs = Script('from medi.api import classes; classes', project=project).infer() assert [d.full_name for d in defs] == ['medi.api.classes'] defs = Script('import medi.api; medi.api', project=project).infer() assert [d.full_name for d in defs] == ['medi.api']
def test_follow_definition_nested_import(Script): Script = partial(Script, project=medi.Project( join(test_dir, 'completion', 'import_tree'))) types = check_follow_definition_types(Script, "import pkg.mod1; pkg") assert types == ['module'] types = check_follow_definition_types(Script, "import pkg.mod1; pkg.mod1") assert types == ['module'] types = check_follow_definition_types(Script, "import pkg.mod1; pkg.mod1.a") assert types == ['instance']
def test_completion_docstring(Script, medi_path): """ Medi should follow imports in certain conditions """ def docstr(src, result): c = Script(src, project=project).complete()[0] assert c.docstring(raw=True, fast=False) == cleandoc(result) project = medi.Project('.', sys_path=[medi_path]) c = Script('import medi\nmed', project=project).complete()[0] assert c.docstring(fast=False) == cleandoc(medi_doc) docstr('import medi\nmedi.Scr', cleandoc(medi.Script.__doc__)) docstr('abcd=3;abcd', '') docstr('"hello"\nabcd=3\nabcd', '') docstr( dedent(''' def x(): "hello" 0 x'''), 'hello') docstr(dedent(''' def x(): "hello";0 x'''), 'hello') # Shouldn't work with a tuple. docstr(dedent(''' def x(): "hello",0 x'''), '') # Should also not work if we rename something. docstr( dedent(''' def x(): "hello" y = x y'''), '')
def ScriptWithProject(Script): project = medi.Project(test_dir) return partial(medi.Script, project=project)