diff --git a/java/java-impl/src/com/intellij/codeInspection/ReplaceNullCheckInspection.java b/java/java-impl/src/com/intellij/codeInspection/ReplaceNullCheckInspection.java index 389a34aa95d1..a0e11fa98556 100644 --- a/java/java-impl/src/com/intellij/codeInspection/ReplaceNullCheckInspection.java +++ b/java/java-impl/src/com/intellij/codeInspection/ReplaceNullCheckInspection.java @@ -147,7 +147,7 @@ public class ReplaceNullCheckInspection extends AbstractBaseJavaLocalInspectionT CommentTracker tracker = new CommentTracker(); PsiExpression requireCall = createRequireExpression(tracker, context.myNullExpr, project, context.myReferenceExpression, context.myNullExpr); - result = tracker.replace(context.myTernary, requireCall); + result = tracker.replaceAndRestoreComments(context.myTernary, requireCall); } else return; LambdaCanBeMethodReferenceInspection.replaceAllLambdasWithMethodReferences(result); CodeStyleManager.getInstance(project).reformat(JavaCodeStyleManager.getInstance(project).shortenClassReferences(result)); diff --git a/java/java-tests/testData/inspection/requireNonNull/afterTernaryOnly.java b/java/java-tests/testData/inspection/requireNonNull/afterTernaryOnly.java index d9371a3dff13..4e132a007b81 100644 --- a/java/java-tests/testData/inspection/requireNonNull/afterTernaryOnly.java +++ b/java/java-tests/testData/inspection/requireNonNull/afterTernaryOnly.java @@ -8,6 +8,12 @@ class Test { } public void test(Object o) { - work(Objects.requireNonNullElse(o, "")); + /*2*/ + /*3*/ + /*4*/ + /*5*/ + /*6*/ + /*7*/ + work(/*1*/Objects.requireNonNullElse(o, "")/*8*/); } } \ No newline at end of file diff --git a/java/java-tests/testData/inspection/requireNonNull/beforeTernaryOnly.java b/java/java-tests/testData/inspection/requireNonNull/beforeTernaryOnly.java index 0575b2665a25..364016a17ede 100644 --- a/java/java-tests/testData/inspection/requireNonNull/beforeTernaryOnly.java +++ b/java/java-tests/testData/inspection/requireNonNull/beforeTernaryOnly.java @@ -8,6 +8,6 @@ class Test { } public void test(Object o) { - work(o == null? "" : o); + work(/*1*/o/*2*/ == /*3*/null/*4*/?/*5*/ ""/*6*/ :/*7*/ o/*8*/); } } \ No newline at end of file