Fix exhaustiveness check for when with subject variable

compiler/testData/diagnostics/tests/when/withSubjectVariable/smartcastToEnum.kt

@@ -3,21 +3,16 @@
 
 enum class E { FIRST, SECOND }
 
-fun testSmartcastToEnumInSubjectInitializer(e: E?) {
-    val x = <!NO_ELSE_IN_WHEN!>when<!> (val ne = e!!) { // TODO fix
+fun testSmartcastToEnumInSubjectInitializer1(e: E?) {
+    val x1 = when (val ne = e!!) {
         E.FIRST -> "f"
         E.SECOND -> "s"
     }
 }
 
-
-sealed class Either
-class Left : Either()
-class Right : Either()
-
-fun testSmartcastToSealedInSubjectInitializer(x: Any?) {
-    val y = <!NO_ELSE_IN_WHEN!>when<!> (val either = x as Either) { // TODO fix
-        is Left -> "L"
-        is Right -> "R"
+fun testSmartcastToEnumInSubjectInitializer2(e: E?) {
+    val x2 = <!NO_ELSE_IN_WHEN!>when<!> (val ne: Any = e!!) { // NB explicit type annotation
+        E.FIRST -> "f"
+        E.SECOND -> "s"
     }
-}
+}

compiler/testData/diagnostics/tests/when/withSubjectVariable/smartcastToEnum.txt

@@ -1,7 +1,7 @@
 package
 
-public fun testSmartcastToEnumInSubjectInitializer(/*0*/ e: E?): kotlin.Unit
-public fun testSmartcastToSealedInSubjectInitializer(/*0*/ x: kotlin.Any?): kotlin.Unit
+public fun testSmartcastToEnumInSubjectInitializer1(/*0*/ e: E?): kotlin.Unit
+public fun testSmartcastToEnumInSubjectInitializer2(/*0*/ e: E?): kotlin.Unit
 
 public final enum class E : kotlin.Enum<E> {
     enum entry FIRST
@@ -23,24 +23,3 @@ public final enum class E : kotlin.Enum<E> {
     public final /*synthesized*/ fun valueOf(/*0*/ value: kotlin.String): E
     public final /*synthesized*/ fun values(): kotlin.Array<E>
 }
-
-public sealed class Either {
-    private constructor Either()
-    public open override /*1*/ /*fake_override*/ fun equals(/*0*/ other: kotlin.Any?): kotlin.Boolean
-    public open override /*1*/ /*fake_override*/ fun hashCode(): kotlin.Int
-    public open override /*1*/ /*fake_override*/ fun toString(): kotlin.String
-}
-
-public final class Left : Either {
-    public constructor Left()
-    public open override /*1*/ /*fake_override*/ fun equals(/*0*/ other: kotlin.Any?): kotlin.Boolean
-    public open override /*1*/ /*fake_override*/ fun hashCode(): kotlin.Int
-    public open override /*1*/ /*fake_override*/ fun toString(): kotlin.String
-}
-
-public final class Right : Either {
-    public constructor Right()
-    public open override /*1*/ /*fake_override*/ fun equals(/*0*/ other: kotlin.Any?): kotlin.Boolean
-    public open override /*1*/ /*fake_override*/ fun hashCode(): kotlin.Int
-    public open override /*1*/ /*fake_override*/ fun toString(): kotlin.String
-}

compiler/testData/diagnostics/tests/when/withSubjectVariable/smartcastToSealed.kt

@@ -0,0 +1,20 @@
+// !LANGUAGE: +VariableDeclarationInWhenSubject
+// !DIAGNOSTICS: -UNUSED_VARIABLE
+
+sealed class Either
+class Left : Either()
+class Right : Either()
+
+fun testSmartcastToSealedInSubjectInitializer1(x: Any?) {
+    val y1 = when (val either = x as Either) {
+        is Left -> "L"
+        is Right -> "R"
+    }
+}
+
+fun testSmartcastToSealedInSubjectInitializer2(x: Any?) {
+    val y2 = <!NO_ELSE_IN_WHEN!>when<!> (val either: Any = x as Either) { // NB explicit type annotation
+        is Left -> "L"
+        is Right -> "R"
+    }
+}

compiler/testData/diagnostics/tests/when/withSubjectVariable/smartcastToSealed.txt

@@ -0,0 +1,25 @@
+package
+
+public fun testSmartcastToSealedInSubjectInitializer1(/*0*/ x: kotlin.Any?): kotlin.Unit
+public fun testSmartcastToSealedInSubjectInitializer2(/*0*/ x: kotlin.Any?): kotlin.Unit
+
+public sealed class Either {
+    private constructor Either()
+    public open override /*1*/ /*fake_override*/ fun equals(/*0*/ other: kotlin.Any?): kotlin.Boolean
+    public open override /*1*/ /*fake_override*/ fun hashCode(): kotlin.Int
+    public open override /*1*/ /*fake_override*/ fun toString(): kotlin.String
+}
+
+public final class Left : Either {
+    public constructor Left()
+    public open override /*1*/ /*fake_override*/ fun equals(/*0*/ other: kotlin.Any?): kotlin.Boolean
+    public open override /*1*/ /*fake_override*/ fun hashCode(): kotlin.Int
+    public open override /*1*/ /*fake_override*/ fun toString(): kotlin.String
+}
+
+public final class Right : Either {
+    public constructor Right()
+    public open override /*1*/ /*fake_override*/ fun equals(/*0*/ other: kotlin.Any?): kotlin.Boolean
+    public open override /*1*/ /*fake_override*/ fun hashCode(): kotlin.Int
+    public open override /*1*/ /*fake_override*/ fun toString(): kotlin.String
+}