Fix completion for codeFragments on functions with expression body

idea/ide-common/src/org/jetbrains/kotlin/resolve/util/codeFragmentResolveUtils.kt

@@ -30,27 +30,16 @@ import org.jetbrains.kotlin.utils.addToStdlib.firstIsInstanceOrNull
 
 //TODO: this code should be moved into debugger which should set correct context for its code fragment
 private fun JetExpression.correctContextForExpression(): JetExpression {
-    when (this) {
-        is JetBlockExpression -> {
-            return this.getStatements().lastOrNull() ?: this
-        }
-        is JetFunctionLiteral -> {
-            return this.getBodyExpression()?.getStatements()?.lastOrNull() ?: this
-        }
+    return when (this) {
+        is JetProperty -> this.getDelegateExpressionOrInitializer()
+        is JetFunctionLiteral -> this.getBodyExpression()?.getStatements()?.lastOrNull()
+        is JetDeclarationWithBody -> this.getBodyExpression()
+        is JetBlockExpression -> this.getStatements().lastOrNull()
         else -> {
-            val previousExpression = this.siblings(forward = false, withItself = false).firstIsInstanceOrNull<JetExpression>()
-            if (previousExpression is JetExpression) {
-                return previousExpression
-            }
-            else {
-                val parentExpression = this.parents.firstIsInstanceOrNull<JetExpression>()
-                if (parentExpression is JetExpression) {
-                    return parentExpression
-                }
-            }
+            this.siblings(forward = false, withItself = false).firstIsInstanceOrNull<JetExpression>()
+                    ?: this.parents.firstIsInstanceOrNull<JetExpression>()
         }
-    }
-    return this
+    } ?: this
 }
 
 public fun JetCodeFragment.getScopeAndDataFlowForAnalyzeFragment(

idea/testData/checker/codeFragments/withoutBodyFunction.kt

@@ -0,0 +1 @@
+<caret>fun foo(aaaB: Int) = aaaB

idea/testData/checker/codeFragments/withoutBodyFunction.kt.fragment

@@ -0,0 +1 @@
+aaaB

idea/testData/checker/codeFragments/withoutBodyProperty.kt

@@ -0,0 +1,6 @@
+package withoutBodyProperties
+
+class A {
+    val aaaB = 1
+    <caret>val aaaC = 1
+}

idea/testData/checker/codeFragments/withoutBodyProperty.kt.fragment

@@ -0,0 +1 @@
+aaaB + aaaC

idea/tests/org/jetbrains/kotlin/idea/debugger/evaluate/CodeFragmentHighlightingTestGenerated.java

@@ -132,6 +132,18 @@ public class CodeFragmentHighlightingTestGenerated extends AbstractCodeFragmentH
             String fileName = JetTestUtils.navigationMetadata("idea/testData/checker/codeFragments/smartCasts.kt");
             doTest(fileName);
         }
+
+        @TestMetadata("withoutBodyFunction.kt")
+        public void testWithoutBodyFunction() throws Exception {
+            String fileName = JetTestUtils.navigationMetadata("idea/testData/checker/codeFragments/withoutBodyFunction.kt");
+            doTest(fileName);
+        }
+
+        @TestMetadata("withoutBodyProperty.kt")
+        public void testWithoutBodyProperty() throws Exception {
+            String fileName = JetTestUtils.navigationMetadata("idea/testData/checker/codeFragments/withoutBodyProperty.kt");
+            doTest(fileName);
+        }
     }
 
     @TestMetadata("idea/testData/checker/codeFragments/imports")