Fix caret positioning in unfolded assignments

idea/testData/codeInsight/codeTransformations/branched/unfolding/assignmentToIf/simpleIfWithComplexAssignmentLHS.kt

@@ -0,0 +1,7 @@
+fun test(n: Int): Array<String> {
+    var x: Array<String> = Array<String>(1, {""})
+
+    x[0] = <caret>if (n > 5) "A" else "B"
+
+    return x
+}

idea/testData/codeInsight/codeTransformations/branched/unfolding/assignmentToIf/simpleIfWithComplexAssignmentLHS.kt.after

@@ -0,0 +1,7 @@
+fun test(n: Int): Array<String> {
+    var x: Array<String> = Array<String>(1, {""})
+
+    <caret>if (n > 5) x[0] = "A" else x[0] = "B"
+
+    return x
+}

idea/testData/codeInsight/codeTransformations/branched/unfolding/assignmentToWhen/simpleWhenWithComplexAssignmentLHS.kt

@@ -0,0 +1,12 @@
+fun test(n: Int): Array<String> {
+    var x: Array<String> = Array<String>(1, {""})
+
+    x[0] = <caret>when(n) {
+        in 0..10 -> "small"
+        in 10..100 -> "average"
+        in 100..1000 -> "big"
+        else -> "unknown"
+    }
+
+    return x
+}

idea/testData/codeInsight/codeTransformations/branched/unfolding/assignmentToWhen/simpleWhenWithComplexAssignmentLHS.kt.after

@@ -0,0 +1,12 @@
+fun test(n: Int): Array<String> {
+    var x: Array<String> = Array<String>(1, {""})
+
+    <caret>when(n) {
+        in 0..10 -> x[0] = "small"
+        in 10..100 -> x[0] = "average"
+        in 100..1000 -> x[0] = "big"
+        else -> x[0] = "unknown"
+    }
+
+    return x
+}