[LL FIR] AbstractInBlockModificationTest: add more tests

^KT-60518

analysis/low-level-api-fir/testdata/inBlockModification/entireContract.kt

@@ -0,0 +1,9 @@
+import kotlin.contracts.InvocationKind
+
+inline fun foo(block: () -> Unit) {
+    <expr>kotlin.contracts.contract {
+        callsInPlace(block, InvocationKind.EXACTLY_ONCE)
+    }</expr>
+
+    block()
+}

analysis/low-level-api-fir/testdata/inBlockModification/entireContract.txt

@@ -0,0 +1 @@
+IN-BLOCK MODIFICATION IS NOT APPLICABLE FOR THIS PLACE

analysis/low-level-api-fir/testdata/inBlockModification/entireFunctionBody.kt

@@ -0,0 +1,3 @@
+// LOOK_UP_FOR_ELEMENT_OF_TYPE: org.jetbrains.kotlin.psi.KtBlockExpression
+
+fun doo() <expr>{}</expr>

analysis/low-level-api-fir/testdata/inBlockModification/entireFunctionBody.txt

@@ -0,0 +1,6 @@
+BEFORE MODIFICATION:
+public final [ResolvedTo(BODY_RESOLVE)] fun doo(): R|kotlin/Unit| {
+}
+
+AFTER MODIFICATION:
+public final [ResolvedTo(ANNOTATIONS_ARGUMENTS_MAPPING)] fun doo(): R|kotlin/Unit| { LAZY_BLOCK }

analysis/low-level-api-fir/testdata/inBlockModification/entireGetterBody.kt

@@ -0,0 +1,4 @@
+// LOOK_UP_FOR_ELEMENT_OF_TYPE: org.jetbrains.kotlin.psi.KtBlockExpression
+
+val i: Int
+    get() <expr>{}</expr>

analysis/low-level-api-fir/testdata/inBlockModification/entireGetterBody.txt

@@ -0,0 +1,8 @@
+BEFORE MODIFICATION:
+public final [ResolvedTo(BODY_RESOLVE)] val i: R|kotlin/Int|
+    public [ResolvedTo(BODY_RESOLVE)] get(): R|kotlin/Int| {
+    }
+
+AFTER MODIFICATION:
+public final [ResolvedTo(ANNOTATIONS_ARGUMENTS_MAPPING)] val i: R|kotlin/Int|
+    public [ResolvedTo(ANNOTATIONS_ARGUMENTS_MAPPING)] get(): R|kotlin/Int| { LAZY_BLOCK }

analysis/low-level-api-fir/testdata/inBlockModification/entireInitializer.kt

@@ -0,0 +1,3 @@
+// LOOK_UP_FOR_ELEMENT_OF_TYPE: org.jetbrains.kotlin.psi.KtConstantExpression
+
+val i: Int = <expr>42</expr>

analysis/low-level-api-fir/testdata/inBlockModification/entireInitializer.txt

@@ -0,0 +1,7 @@
+BEFORE MODIFICATION:
+public final [ResolvedTo(BODY_RESOLVE)] val i: R|kotlin/Int| = Int(42)
+    public [ResolvedTo(BODY_RESOLVE)] get(): R|kotlin/Int|
+
+AFTER MODIFICATION:
+public final [ResolvedTo(ANNOTATIONS_ARGUMENTS_MAPPING)] val i: R|kotlin/Int| = LAZY_EXPRESSION
+    public [ResolvedTo(BODY_RESOLVE)] get(): R|kotlin/Int|

analysis/low-level-api-fir/testdata/inBlockModification/entireSetterBody.kt

@@ -0,0 +1,4 @@
+// LOOK_UP_FOR_ELEMENT_OF_TYPE: org.jetbrains.kotlin.psi.KtBlockExpression
+
+var i: Int = 4
+    set(value) <expr>{}</expr>

analysis/low-level-api-fir/testdata/inBlockModification/entireSetterBody.txt

@@ -0,0 +1,10 @@
+BEFORE MODIFICATION:
+public final [ResolvedTo(BODY_RESOLVE)] var i: R|kotlin/Int| = Int(4)
+    public [ResolvedTo(BODY_RESOLVE)] get(): R|kotlin/Int|
+    public [ResolvedTo(BODY_RESOLVE)] set([ResolvedTo(BODY_RESOLVE)] value: R|kotlin/Int|): R|kotlin/Unit| {
+    }
+
+AFTER MODIFICATION:
+public final [ResolvedTo(ANNOTATIONS_ARGUMENTS_MAPPING)] var i: R|kotlin/Int| = Int(4)
+    public [ResolvedTo(BODY_RESOLVE)] get(): R|kotlin/Int|
+    public [ResolvedTo(ANNOTATIONS_ARGUMENTS_MAPPING)] set([ResolvedTo(BODY_RESOLVE)] value: R|kotlin/Int|): R|kotlin/Unit| { LAZY_BLOCK }

analysis/low-level-api-fir/testdata/inBlockModification/insideContract.kt

@@ -0,0 +1,9 @@
+import kotlin.contracts.InvocationKind
+
+inline fun foo(block: () -> Unit) {
+    kotlin.contracts.contract {
+        <expr>callsInPlace</expr>(block, InvocationKind.EXACTLY_ONCE)
+    }
+
+    block()
+}

analysis/low-level-api-fir/testdata/inBlockModification/insideContract.txt

@@ -0,0 +1 @@
+IN-BLOCK MODIFICATION IS NOT APPLICABLE FOR THIS PLACE

analysis/low-level-api-fir/testdata/inBlockModification/insideDefaultParameter.kt

@@ -0,0 +1,7 @@
+class MyProducer {
+    fun produce(): Int = 4
+}
+
+fun MyProducer.testFun(param1: Int = <expr>produce</expr>()) {
+    42
+}

analysis/low-level-api-fir/testdata/inBlockModification/insideDefaultParameter.txt

@@ -0,0 +1 @@
+IN-BLOCK MODIFICATION IS NOT APPLICABLE FOR THIS PLACE