Allow to process return statements without parent KtDeclaration (EA-70883)

idea/testData/checker/codeFragments/startingFromReturn.kt

@@ -0,0 +1,4 @@
+fun foo(): Int? {
+    val a: Int? = 1
+    <caret>return a
+}

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

@@ -0,0 +1 @@
+<error descr="[RETURN_NOT_ALLOWED] 'return' is not allowed here">return a</error>

idea/testData/checker/recovery/returnInFileAnnotation.kt

@@ -0,0 +1 @@
+<error descr="[UNRESOLVED_REFERENCE] Unresolved reference: suppress">@file:suppress</error>(<error descr="[RETURN_NOT_ALLOWED] 'return' is not allowed here"><error descr="[RETURN_NOT_ALLOWED] 'return' is not allowed here">return <error descr="[UNRESOLVED_REFERENCE] Unresolved reference: a"><error descr="[UNRESOLVED_REFERENCE] Unresolved reference: a">a</error></error></error></error>)

idea/tests/org/jetbrains/kotlin/checkers/PsiCheckerTestGenerated.java

@@ -630,6 +630,12 @@ public class PsiCheckerTestGenerated extends AbstractPsiCheckerTest {
             String fileName = KotlinTestUtils.navigationMetadata("idea/testData/checker/recovery/namelessToplevelDeclarations.kt");
             doTest(fileName);
         }
+
+        @TestMetadata("returnInFileAnnotation.kt")
+        public void testReturnInFileAnnotation() throws Exception {
+            String fileName = KotlinTestUtils.navigationMetadata("idea/testData/checker/recovery/returnInFileAnnotation.kt");
+            doTest(fileName);
+        }
     }
 
     @TestMetadata("idea/testData/checker/rendering")

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

@@ -151,6 +151,12 @@ public class CodeFragmentHighlightingTestGenerated extends AbstractCodeFragmentH
             doTest(fileName);
         }
 
+        @TestMetadata("startingFromReturn.kt")
+        public void testStartingFromReturn() throws Exception {
+            String fileName = KotlinTestUtils.navigationMetadata("idea/testData/checker/codeFragments/startingFromReturn.kt");
+            doTest(fileName);
+        }
+
         @TestMetadata("withoutBodyFunction.kt")
         public void testWithoutBodyFunction() throws Exception {
             String fileName = KotlinTestUtils.navigationMetadata("idea/testData/checker/codeFragments/withoutBodyFunction.kt");