PY-992 Extract Method doesn't handle conditional returns correctly

python/src/com/jetbrains/python/codeInsight/codeFragment/PyCodeFragmentUtil.java

@@ -100,9 +100,12 @@ public class PyCodeFragmentUtil {
       for (Instruction outerInstruction : outerInstructions) {
         // Here we check control flow when for statement content is beeing extracted
         if (outerInstruction instanceof ReadWriteInstruction &&
-            PyForStatementNavigator.getPyForStatementByIterable(outerInstruction.getElement())!=null){
-          errorFound = false;
-          break;
+            PyForStatementNavigator.getPyForStatementByIterable(outerInstruction.getElement())!=null) {
+          // In case when return instruction is inside
+          if (!returnInstructionInside){
+            errorFound = false;
+            break;
+          }
         }
       }
       if (errorFound){

python/testData/refactoring/extractmethod/conditionalreturn.before.py

@@ -0,0 +1,7 @@
+class Conference(object):
+    def __init__(self):
+        self.talks = []
+
+    def getTalkAt(self, hour):
+        for start, end, name in self.talks:
+            <selection>if hour >= start and hour < end: return name</selection>

python/testSrc/com/jetbrains/python/refactoring/PyExtractMethodTest.java

@@ -126,4 +126,9 @@ public class PyExtractMethodTest extends LightMarkedTestCase {
   public void testClass() throws Throwable {
     doTest("context/class.before.py", "bar", "context/class.after.py");
   }
+
+  public void testConditionalReturn() throws Throwable {
+    doTest("conditionalreturn.before.py", "bar", "Cannot perform refactoring when execution flow is interrupted");
+  }
+
 }