Fix bug when chain is not detected if method expression is used

idea/idea-jvm/src/com/intellij/debugger/streams/kotlin/psi/impl/KotlinChainBuilderBase.kt

@@ -55,11 +55,16 @@ abstract class KotlinChainBuilderBase(private val transformer: ChainTransformer<
   private fun getLatestElementInScope(element: PsiElement?): PsiElement? {
     var current = element
     while (current != null) {
+      if (current is KtNamedFunction && current.hasInitializer()) {
+        break
+      }
+
       val parent = current.parent
       if (parent is KtBlockExpression || parent is KtLambdaExpression) {
         break
       }
 
+
       current = parent
     }
 

idea/testData/debugger/sequence/psi/streams/positive/location/AsMethodExpression.kt

@@ -0,0 +1,5 @@
+package location
+
+import java.util.stream.Stream
+
+<caret>fun method() = Stream.of(1, 2, 3).count()

idea/testData/debugger/sequence/psi/streams/positive/location/AsProperty.kt

@@ -0,0 +1,5 @@
+package location
+
+import java.util.stream.Stream
+
+<caret>val property = Stream.of(1, 2, 3).findAny()

idea/tests/com/intellij/debugger/streams/kotlin/psi/java/LocationPositiveChainTest.kt

@@ -36,4 +36,6 @@ class LocationPositiveChainTest : PositiveJavaStreamTest("location") {
   fun testInString() = doTest()
   fun testInVariableName() = doTest()
   fun testInMethodReference() = doTest()
+
+  fun testAsMethodExpression() = doTest()
 }