Add tests for folding/unfolding operations

idea/testData/codeInsight/codeTransformations/branched/unfolding/assignment/innerIfTransformed.kt

@@ -0,0 +1,21 @@
+fun test(n: Int): String {
+    var res: String
+
+    if (n == 1) {
+        <caret>res = if (3 > 2) {
+            println("***")
+            "one"
+        } else {
+            println("***")
+            "???"
+        }
+    } else if (n == 2) {
+        println("***")
+        res = "two"
+    } else {
+        println("***")
+        res = "too many"
+    }
+
+    return res
+}

idea/testData/codeInsight/codeTransformations/branched/unfolding/assignment/innerIfTransformed.kt.after

@@ -0,0 +1,21 @@
+fun test(n: Int): String {
+    var res: String
+
+    if (n == 1) {
+        <caret>if (3 > 2) {
+            println("***")
+            res = "one"
+        } else {
+            println("***")
+            res = "???"
+        }
+    } else if (n == 2) {
+        println("***")
+        res = "two"
+    } else {
+        println("***")
+        res = "too many"
+    }
+
+    return res
+}

idea/testData/codeInsight/codeTransformations/branched/unfolding/assignment/nestedIfs.kt

@@ -0,0 +1,21 @@
+fun test(n: Int): String {
+    var res: String
+
+    <caret>res = if (n == 1) {
+        if (3 > 2) {
+            println("***")
+            "one"
+        } else {
+            println("***")
+            "???"
+        }
+    } else if (n == 2) {
+        println("***")
+        "two"
+    } else {
+        println("***")
+        "too many"
+    }
+
+    return res
+}

idea/testData/codeInsight/codeTransformations/branched/unfolding/assignment/nestedIfs.kt.after

@@ -0,0 +1,21 @@
+fun test(n: Int): String {
+    var res: String
+
+    <caret>if (n == 1) {
+        res = if (3 > 2) {
+            println("***")
+            "one"
+        } else {
+            println("***")
+            "???"
+        }
+    } else res = if (n == 2) {
+        println("***")
+        "two"
+    } else {
+        println("***")
+        "too many"
+    }
+
+    return res
+}

idea/testData/codeInsight/codeTransformations/branched/unfolding/assignment/simpleIf.kt

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

idea/testData/codeInsight/codeTransformations/branched/unfolding/assignment/simpleIf.kt.after

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

idea/testData/codeInsight/codeTransformations/branched/unfolding/assignment/simpleIfWithAugmentedAssignment.kt

@@ -0,0 +1,7 @@
+fun test(n: Int): String {
+    var res: String = "!"
+
+    <caret>res += if (n == 1) "one" else "two"
+
+    return res
+}

idea/testData/codeInsight/codeTransformations/branched/unfolding/assignment/simpleIfWithAugmentedAssignment.kt.after

@@ -0,0 +1,7 @@
+fun test(n: Int): String {
+    var res: String = "!"
+
+    <caret>if (n == 1) res += "one" else res += "two"
+
+    return res
+}

idea/testData/codeInsight/codeTransformations/branched/unfolding/assignment/simpleIfWithBlocks.kt

@@ -0,0 +1,13 @@
+fun test(n: Int): String {
+    var res: String
+
+    <caret>res = if (n == 1) {
+        println("***")
+        "one"
+    } else {
+        println("***")
+        "two"
+    }
+
+    return res
+}

idea/testData/codeInsight/codeTransformations/branched/unfolding/assignment/simpleIfWithBlocks.kt.after

@@ -0,0 +1,13 @@
+fun test(n: Int): String {
+    var res: String
+
+    <caret>if (n == 1) {
+        println("***")
+        res = "one"
+    } else {
+        println("***")
+        res = "two"
+    }
+
+    return res
+}

idea/testData/codeInsight/codeTransformations/branched/unfolding/assignment/simpleIfWithoutAssignment.kt

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

idea/testData/codeInsight/codeTransformations/branched/unfolding/asymmetricReturn/simpleIf.kt

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

idea/testData/codeInsight/codeTransformations/branched/unfolding/asymmetricReturn/simpleIf.kt.after

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

idea/testData/codeInsight/codeTransformations/branched/unfolding/asymmetricReturn/simpleIfWithBlocks.kt

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

idea/testData/codeInsight/codeTransformations/branched/unfolding/asymmetricReturn/simpleIfWithBlocks.kt.after

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

idea/testData/codeInsight/codeTransformations/branched/unfolding/return/innerIfTransformed.kt

@@ -0,0 +1,17 @@
+fun test(n: Int): String {
+    if (n == 1) {
+        <caret>return if (3 > 2) {
+            println("***")
+            "one"
+        } else {
+            println("***")
+            "???"
+        }
+    } else if (n == 2) {
+        println("***")
+        return "two"
+    } else {
+        println("***")
+        return "too many"
+    }
+}

idea/testData/codeInsight/codeTransformations/branched/unfolding/return/innerIfTransformed.kt.after

@@ -0,0 +1,17 @@
+fun test(n: Int): String {
+    if (n == 1) {
+        <caret>if (3 > 2) {
+            println("***")
+            return "one"
+        } else {
+            println("***")
+            return "???"
+        }
+    } else if (n == 2) {
+        println("***")
+        return "two"
+    } else {
+        println("***")
+        return "too many"
+    }
+}

idea/testData/codeInsight/codeTransformations/branched/unfolding/return/simpleIf.kt

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

idea/testData/codeInsight/codeTransformations/branched/unfolding/return/simpleIf.kt.after

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

idea/testData/codeInsight/codeTransformations/branched/unfolding/return/simpleIfWithBlocks.kt

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

idea/testData/codeInsight/codeTransformations/branched/unfolding/return/simpleIfWithBlocks.kt.after

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