PY-17265 Add missing test on handling imports for moved local function

diff --git a/python/testData/refactoring/makeFunctionTopLevel/localFunctionMoveToOtherFile/after/main.py b/python/testData/refactoring/makeFunctionTopLevel/localFunctionMoveToOtherFile/after/main.py
new file mode 100644 (file)
index 0000000..d602112
--- /dev/null
+++ b/python/testData/refactoring/makeFunctionTopLevel/localFunctionMoveToOtherFile/after/main.py
@@ -0,0 +1,9 @@
+CONST = 42
+
+
+class MyClass:
+    pass
+
+
+def f():
+    pass

diff --git a/python/testData/refactoring/makeFunctionTopLevel/localFunctionMoveToOtherFile/after/util.py b/python/testData/refactoring/makeFunctionTopLevel/localFunctionMoveToOtherFile/after/util.py
new file mode 100644 (file)
index 0000000..342d6d7
--- /dev/null
+++ b/python/testData/refactoring/makeFunctionTopLevel/localFunctionMoveToOtherFile/after/util.py
@@ -0,0 +1,10 @@
+from sys import path
+
+from main import f, MyClass, CONST
+
+
+def g():
+    f()
+    print(MyClass())
+    print(path)
+    print(CONST)
\ No newline at end of file

diff --git a/python/testData/refactoring/makeFunctionTopLevel/localFunctionMoveToOtherFile/before/main.py b/python/testData/refactoring/makeFunctionTopLevel/localFunctionMoveToOtherFile/before/main.py
new file mode 100644 (file)
index 0000000..508e4a5
--- /dev/null
+++ b/python/testData/refactoring/makeFunctionTopLevel/localFunctionMoveToOtherFile/before/main.py
@@ -0,0 +1,15 @@
+from sys import path
+
+CONST = 42
+
+
+class MyClass:
+    pass
+
+
+def f():
+    def <caret>g():
+        f()
+        print(MyClass())
+        print(path)
+        print(CONST)

diff --git a/python/testSrc/com/jetbrains/python/refactoring/PyMakeFunctionTopLevelTest.java b/python/testSrc/com/jetbrains/python/refactoring/PyMakeFunctionTopLevelTest.java
index c0da5618c7be7340d666c9e062967ef6c124fcfb..9101c80b77b660ca41aac93af943274d12c02013 100644 (file)
--- a/python/testSrc/com/jetbrains/python/refactoring/PyMakeFunctionTopLevelTest.java
+++ b/python/testSrc/com/jetbrains/python/refactoring/PyMakeFunctionTopLevelTest.java
@@ -210,6 +210,10 @@ public class PyMakeFunctionTopLevelTest extends PyTestCase {
     doMultiFileTest("util.py", null);
   }
 
+  public void testLocalFunctionMoveToOtherFile() throws IOException {
+    doMultiFileTest("util.py", null);
+  }
+
   public void testMethodCalledViaClass() {
     doTestSuccess();
   }