Partial body resolve: more efficient for-loop handling

idea/ide-common/src/org/jetbrains/jet/lang/resolve/lazy/PartialBodyResolveFilter.kt

@@ -162,6 +162,11 @@ class PartialBodyResolveFilter(elementToResolve: JetElement, private val body: J
                 }
             }
 
+            override fun visitForExpression(expression: JetForExpression) {
+                // analyze only the loop-range expression, do not enter the loop body
+                expression.getLoopRange()?.accept(this)
+            }
+
             //TODO: when
         })
 

idea/testData/resolve/partialBodyResolve/For.dump

@@ -0,0 +1,3 @@
+Resolve target: value-parameter val p: kotlin.Any? smart-casted to kotlin.String
+Skipped statements:
+print(p1!!)

idea/testData/resolve/partialBodyResolve/For.kt

@@ -0,0 +1,9 @@
+fun x(s: String): Collection<String>{}
+
+fun foo(p: Any?, p1: Any?) {
+    for (e in x(p as String)) {
+        print(p1!!)
+    }
+
+    <caret>p.size
+}

idea/tests/org/jetbrains/jet/resolve/PartialBodyResolveTestGenerated.java

@@ -19,6 +19,7 @@ package org.jetbrains.jet.resolve;
 import com.intellij.testFramework.TestDataPath;
 import org.jetbrains.jet.JUnit3RunnerWithInners;
 import org.jetbrains.jet.JetTestUtils;
+import org.jetbrains.jet.test.InnerTestClasses;
 import org.jetbrains.jet.test.TestMetadata;
 import org.junit.runner.RunWith;
 
@@ -53,6 +54,12 @@ public class PartialBodyResolveTestGenerated extends AbstractPartialBodyResolveT
         doTest(fileName);
     }
 
+    @TestMetadata("For.kt")
+    public void testFor() throws Exception {
+        String fileName = JetTestUtils.navigationMetadata("idea/testData/resolve/partialBodyResolve/For.kt");
+        doTest(fileName);
+    }
+
     @TestMetadata("IfBranchesAutoCasts.kt")
     public void testIfBranchesAutoCasts() throws Exception {
         String fileName = JetTestUtils.navigationMetadata("idea/testData/resolve/partialBodyResolve/IfBranchesAutoCasts.kt");