Make folding if to return/assignment work with if-else if-else #KT-13452 Fixed

idea/testData/intentions/branched/folding/ifToAssignment/ifElseIf.kt

@@ -0,0 +1,9 @@
+// IS_APPLICABLE: false
+
+fun test(n: Int) {
+    var a: String = ""
+    <caret>if (n == 1)
+        a = "one"
+    else if (n == 2)
+        a = "two"
+}

idea/testData/intentions/branched/folding/ifToAssignment/ifElseIfElse.kt

@@ -0,0 +1,9 @@
+fun test(n: Int) {
+    val a: String
+    <caret>if (n == 1)
+        a = "one"
+    else if (n == 2)
+        a = "two"
+    else
+        a = "three"
+}

idea/testData/intentions/branched/folding/ifToAssignment/ifElseIfElse.kt.after

@@ -0,0 +1,9 @@
+fun test(n: Int) {
+    val a: String
+    <caret>a = if (n == 1)
+        "one"
+    else if (n == 2)
+        "two"
+    else
+        "three"
+}

idea/testData/intentions/branched/folding/ifToAssignment/ifElseifElseInconsistent.kt

@@ -0,0 +1,12 @@
+// IS_APPLICABLE: false
+
+fun test(n: Int) {
+    val a: String
+    val b: String
+    <caret>if (n == 1)
+        a = "one"
+    else if (n == 2)
+        a = "two"
+    else
+        b = "three"
+}

idea/testData/intentions/branched/folding/ifToReturn/ifElseIf.kt

@@ -0,0 +1,10 @@
+// IS_APPLICABLE: false
+
+fun test(n: Int): String {
+    <caret>if (n == 1)
+        return "one"
+    else if (n == 2)
+        return "two"
+
+    return "three"
+}

idea/testData/intentions/branched/folding/ifToReturn/ifElseIfElse.kt

@@ -0,0 +1,8 @@
+fun test(n: Int): String {
+    <caret>if (n == 1)
+        return "one"
+    else if (n == 2)
+        return "two"
+    else
+        return "three"
+}

idea/testData/intentions/branched/folding/ifToReturn/ifElseIfElse.kt.after

@@ -0,0 +1,8 @@
+fun test(n: Int): String {
+    <caret>return if (n == 1)
+        "one"
+    else if (n == 2)
+        "two"
+    else
+        "three"
+}

idea/testData/intentions/branched/folding/ifToReturn/ifElseIfElseInconsistent.kt

@@ -0,0 +1,12 @@
+// IS_APPLICABLE: false
+
+fun test(n: Int): String {
+    val a: String
+    <caret>if (n == 1)
+        return "one"
+    else if (n == 2)
+        return "two"
+    else
+        a = "three"
+    return a
+}