IDEA-175018 Unwrap if produces red code

java/java-analysis-impl/src/com/intellij/codeInsight/daemon/impl/quickfix/SimplifyBooleanExpressionFix.java

@@ -189,9 +189,16 @@ public class SimplifyBooleanExpressionFix extends LocalQuickFixOnPsiElement {
     PsiElement parent = orig.getParent();
     if (parent == null) return;
 
+    PsiElement grandParent = parent.getParent();
     if (parent instanceof PsiCodeBlock && blockAlwaysReturns(statement)) {
       removeFollowingStatements(orig, (PsiCodeBlock)parent);
     }
+    else if (grandParent instanceof PsiCodeBlock && parent instanceof PsiIfStatement) {
+      PsiIfStatement ifStmt = (PsiIfStatement)parent;
+      if (ifStmt.getElseBranch() == orig && blockAlwaysReturns(ifStmt.getThenBranch()) && blockAlwaysReturns(statement)) {
+        removeFollowingStatements(ifStmt, (PsiCodeBlock)grandParent);
+      }
+    }
 
     if (parent instanceof PsiCodeBlock) {
       if (statement instanceof PsiBlockStatement &&
@@ -246,7 +253,8 @@ public class SimplifyBooleanExpressionFix extends LocalQuickFixOnPsiElement {
     orig.delete();
   }
 
-  private static boolean blockAlwaysReturns(@NotNull PsiStatement statement) {
+  private static boolean blockAlwaysReturns(@Nullable PsiStatement statement) {
+    if (statement == null) return false;
     try {
       return ControlFlowUtil.returnPresent(HighlightControlFlowUtil.getControlFlowNoConstantEvaluate(statement));
     }

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/unwrapIfStatement/afterAlwaysReturning.java

@@ -0,0 +1,14 @@
+// "Unwrap 'if' statement" "true"
+class X {
+    boolean f(int[] a) {
+        if (a.length == 0) return false;
+        if (a.length == 1) {
+            System.out.println();
+            return true;
+        }
+        else {
+            System.out.println("2");
+            return true;
+        }
+    }
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/unwrapIfStatement/beforeAlwaysReturning.java

@@ -0,0 +1,15 @@
+// "Unwrap 'if' statement" "true"
+class X {
+    boolean f(int[] a) {
+        if (a.length == 0) return false;
+        if (a.length == 1) {
+            System.out.println();
+            return true;
+        }
+        else if (a.<caret>length > 1) {
+            System.out.println("2");
+            return true;
+        }
+        return false;
+    }
+}