Retain data flow info after when-expressions

This means the subject expression and the common data flow info of all entries'
conditions

compiler/testData/diagnostics/tests/dataFlowInfoTraversal/WhenEntry.kt

@@ -0,0 +1,24 @@
+fun foo(x: Number, y: Int) {
+    when (x) {
+        x as Int -> x : Int
+        y -> {}
+        else -> {}
+    }
+    <!TYPE_MISMATCH!>x<!> : Int
+}
+
+fun bar(x: Number) {
+    when (x) {
+        x as Int -> x : Int
+        else -> {}
+    }
+    x : Int
+}
+
+fun whenWithoutSubject(x: Number) {
+    when {
+        (x as Int) == 42 -> x : Int
+        else -> {}
+    }
+    x : Int
+}

compiler/testData/diagnostics/tests/dataFlowInfoTraversal/WhenIn.kt

@@ -0,0 +1,13 @@
+fun foo(x: Int, list: List<Int>?) {
+    when (x) {
+        in list!! -> list : List<Int>
+        else -> {}
+    }
+}
+
+fun whenWithoutSubject(x: Int, list: List<Int>?) {
+    when {
+        x in list!! -> list : List<Int>
+        else -> {}
+    }
+}

compiler/testData/diagnostics/tests/dataFlowInfoTraversal/WhenSubject.kt

@@ -0,0 +1,6 @@
+fun foo(x: Number) {
+    when (x as Int) {
+        else -> x : Int
+    }
+    x : Int
+}