[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

analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/file/structure/OutOfContentRootInBlockModificationTestGenerated.java

@@ -30,6 +30,36 @@ public class OutOfContentRootInBlockModificationTestGenerated extends AbstractOu
         runTest("analysis/low-level-api-fir/testdata/inBlockModification/contract.kt");
     }
 
+    @Test
+    @TestMetadata("entireContract.kt")
+    public void testEntireContract() throws Exception {
+        runTest("analysis/low-level-api-fir/testdata/inBlockModification/entireContract.kt");
+    }
+
+    @Test
+    @TestMetadata("entireFunctionBody.kt")
+    public void testEntireFunctionBody() throws Exception {
+        runTest("analysis/low-level-api-fir/testdata/inBlockModification/entireFunctionBody.kt");
+    }
+
+    @Test
+    @TestMetadata("entireGetterBody.kt")
+    public void testEntireGetterBody() throws Exception {
+        runTest("analysis/low-level-api-fir/testdata/inBlockModification/entireGetterBody.kt");
+    }
+
+    @Test
+    @TestMetadata("entireInitializer.kt")
+    public void testEntireInitializer() throws Exception {
+        runTest("analysis/low-level-api-fir/testdata/inBlockModification/entireInitializer.kt");
+    }
+
+    @Test
+    @TestMetadata("entireSetterBody.kt")
+    public void testEntireSetterBody() throws Exception {
+        runTest("analysis/low-level-api-fir/testdata/inBlockModification/entireSetterBody.kt");
+    }
+
     @Test
     @TestMetadata("extensionWithDefaultParameters.kt")
     public void testExtensionWithDefaultParameters() throws Exception {
@@ -42,6 +72,18 @@ public class OutOfContentRootInBlockModificationTestGenerated extends AbstractOu
         runTest("analysis/low-level-api-fir/testdata/inBlockModification/functionWithDefaultParameters.kt");
     }
 
+    @Test
+    @TestMetadata("insideContract.kt")
+    public void testInsideContract() throws Exception {
+        runTest("analysis/low-level-api-fir/testdata/inBlockModification/insideContract.kt");
+    }
+
+    @Test
+    @TestMetadata("insideDefaultParameter.kt")
+    public void testInsideDefaultParameter() throws Exception {
+        runTest("analysis/low-level-api-fir/testdata/inBlockModification/insideDefaultParameter.kt");
+    }
+
     @Test
     @TestMetadata("memberFunWithBodyWithContract.kt")
     public void testMemberFunWithBodyWithContract() throws Exception {

analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/file/structure/SourceInBlockModificationTestGenerated.java

@@ -30,6 +30,36 @@ public class SourceInBlockModificationTestGenerated extends AbstractSourceInBloc
         runTest("analysis/low-level-api-fir/testdata/inBlockModification/contract.kt");
     }
 
+    @Test
+    @TestMetadata("entireContract.kt")
+    public void testEntireContract() throws Exception {
+        runTest("analysis/low-level-api-fir/testdata/inBlockModification/entireContract.kt");
+    }
+
+    @Test
+    @TestMetadata("entireFunctionBody.kt")
+    public void testEntireFunctionBody() throws Exception {
+        runTest("analysis/low-level-api-fir/testdata/inBlockModification/entireFunctionBody.kt");
+    }
+
+    @Test
+    @TestMetadata("entireGetterBody.kt")
+    public void testEntireGetterBody() throws Exception {
+        runTest("analysis/low-level-api-fir/testdata/inBlockModification/entireGetterBody.kt");
+    }
+
+    @Test
+    @TestMetadata("entireInitializer.kt")
+    public void testEntireInitializer() throws Exception {
+        runTest("analysis/low-level-api-fir/testdata/inBlockModification/entireInitializer.kt");
+    }
+
+    @Test
+    @TestMetadata("entireSetterBody.kt")
+    public void testEntireSetterBody() throws Exception {
+        runTest("analysis/low-level-api-fir/testdata/inBlockModification/entireSetterBody.kt");
+    }
+
     @Test
     @TestMetadata("extensionWithDefaultParameters.kt")
     public void testExtensionWithDefaultParameters() throws Exception {
@@ -42,6 +72,18 @@ public class SourceInBlockModificationTestGenerated extends AbstractSourceInBloc
         runTest("analysis/low-level-api-fir/testdata/inBlockModification/functionWithDefaultParameters.kt");
     }
 
+    @Test
+    @TestMetadata("insideContract.kt")
+    public void testInsideContract() throws Exception {
+        runTest("analysis/low-level-api-fir/testdata/inBlockModification/insideContract.kt");
+    }
+
+    @Test
+    @TestMetadata("insideDefaultParameter.kt")
+    public void testInsideDefaultParameter() throws Exception {
+        runTest("analysis/low-level-api-fir/testdata/inBlockModification/insideDefaultParameter.kt");
+    }
+
     @Test
     @TestMetadata("memberFunWithBodyWithContract.kt")
     public void testMemberFunWithBodyWithContract() throws Exception {