switch expressions 2.0: expected type by yield

GitOrigin-RevId: f4e396ddc90d66790db044525611c536c3f620fa

java/java-impl/src/com/intellij/codeInsight/ExpectedTypesProvider.java

@@ -371,6 +371,14 @@ public class ExpectedTypesProvider {
       }
     }
 
+    @Override
+    public void visitYieldStatement(PsiYieldStatement statement) {
+      PsiSwitchExpression exitedElement = statement.findEnclosingExpression();
+      if (exitedElement != null) {
+        Collections.addAll(myResult, getExpectedTypes(exitedElement, myForCompletion));
+      }
+    }
+
     @Override
     public void visitMethodCallExpression(@NotNull PsiMethodCallExpression expression) {
       myExpr = (PsiExpression)myExpr.getParent();

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createLocalFromUsage/afterSwitchExprBreakJava13Preview.java

@@ -4,7 +4,7 @@ class Foo {
         return switch (i) {
             default -> {
                 String foo;
-                break foo;
+                yield foo;
             }
         };
     }

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createLocalFromUsage/afterSwitchExprSelectorJava13Preview.java

@@ -4,7 +4,7 @@ class Foo {
         int foo;
         return switch (foo) {
             default -> {
-                break i;
+                yield i;
             }
         };
     }

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createLocalFromUsage/afterSwitchInFieldInitializer2Java13Preview.java

@@ -3,7 +3,7 @@ class Foo {
     int x = foo ? 0 : switch(1) {
         default -> {
             boolean foo;
-            break foo;
+            yield foo;
         }
     };
 }

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createLocalFromUsage/beforeSwitchExprBreakJava13Preview.java

@@ -3,7 +3,7 @@ class Foo {
     String test(int i) {
         return switch (i) {
             default -> {
-                break fo<caret>o;
+                yield fo<caret>o;
             }
         };
     }

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createLocalFromUsage/beforeSwitchExprSelectorJava13Preview.java

@@ -3,7 +3,7 @@ class Foo {
     String test(int i) {
         return switch (f<caret>oo) {
             default -> {
-                break i;
+                yield i;
             }
         };
     }

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/createLocalFromUsage/beforeSwitchInFieldInitializer2Java13Preview.java

@@ -2,7 +2,7 @@
 class Foo {
     int x = foo ? 0 : switch(1) {
         default -> {
-            break f<caret>oo;
+            yield f<caret>oo;
         }
     };
 }