[FIR] Don't miss RETURN_NOT_ALLOWED

^KT-59914 Fixed

compiler/testData/diagnostics/tests/controlStructures/kt799.fir.kt

@@ -19,7 +19,7 @@ val <!IMPLICIT_NOTHING_PROPERTY_TYPE!>b<!> = <!RETURN_NOT_ALLOWED!>return<!> 1
 val c = doSmth(if (true) 3 else <!RETURN_NOT_ALLOWED!>return<!>)
 
 
-fun f(mi: Int = if (true) 0 else return) {}
+fun f(mi: Int = if (true) 0 else <!RETURN_NOT_ALLOWED!>return<!>) {}
 
 fun doSmth(i: Int) {
 }

compiler/testData/diagnostics/tests/inline/nonLocalReturns/lambdaAsGeneric.fir.kt

@@ -1,7 +1,7 @@
 
 fun box() : String {
     test {
-        return@box "123"
+        <!RETURN_NOT_ALLOWED!>return@box<!> "123"
     }
 
     return "OK"

compiler/testData/diagnostics/tests/inline/nonLocalReturns/lambdaAsNonFunction.fir.kt

@@ -1,11 +0,0 @@
-fun box() : String {
-    test {
-        return@box "123"
-    }
-
-    return "OK"
-}
-
-<!NOTHING_TO_INLINE!>inline<!> fun test(p: Any) {
-    p.toString()
-}

compiler/testData/diagnostics/tests/inline/nonLocalReturns/lambdaAsNonFunction.kt

@@ -1,3 +1,4 @@
+// FIR_IDENTICAL
 fun box() : String {
     test {
         <!RETURN_NOT_ALLOWED!>return@box<!> "123"