diff --git a/idea/src/org/jetbrains/kotlin/idea/intentions/branchedTransformations/intentions/IfThenToElvisIntention.kt b/idea/src/org/jetbrains/kotlin/idea/intentions/branchedTransformations/intentions/IfThenToElvisIntention.kt index 7a19a46ab76..f0af7d0ab9d 100644 --- a/idea/src/org/jetbrains/kotlin/idea/intentions/branchedTransformations/intentions/IfThenToElvisIntention.kt +++ b/idea/src/org/jetbrains/kotlin/idea/intentions/branchedTransformations/intentions/IfThenToElvisIntention.kt @@ -26,6 +26,7 @@ import org.jetbrains.kotlin.idea.inspections.IntentionBasedInspection import org.jetbrains.kotlin.idea.inspections.branchedTransformations.IfThenToSafeAccessInspection import org.jetbrains.kotlin.idea.intentions.SelfTargetingOffsetIndependentIntention import org.jetbrains.kotlin.idea.intentions.branchedTransformations.* +import org.jetbrains.kotlin.idea.util.CommentSaver import org.jetbrains.kotlin.idea.util.application.runWriteAction import org.jetbrains.kotlin.psi.* import org.jetbrains.kotlin.resolve.bindingContextUtil.isUsedAsExpression @@ -78,6 +79,9 @@ class IfThenToElvisIntention : SelfTargetingOffsetIndependentIntentionif (arg is Int) arg + // 2 + else 10 +} \ No newline at end of file diff --git a/idea/testData/intentions/branched/ifThenToElvis/ifToElvisSwallowComments.kt.after b/idea/testData/intentions/branched/ifThenToElvis/ifToElvisSwallowComments.kt.after new file mode 100644 index 00000000000..86a90f25ab7 --- /dev/null +++ b/idea/testData/intentions/branched/ifThenToElvis/ifToElvisSwallowComments.kt.after @@ -0,0 +1,6 @@ +fun foo(arg: Any): Int { + // 1 + return arg as? Int + // 2 + ?: 10 +} \ No newline at end of file diff --git a/idea/tests/org/jetbrains/kotlin/idea/intentions/IntentionTestGenerated.java b/idea/tests/org/jetbrains/kotlin/idea/intentions/IntentionTestGenerated.java index 25f1cea17b9..db289d073c7 100644 --- a/idea/tests/org/jetbrains/kotlin/idea/intentions/IntentionTestGenerated.java +++ b/idea/tests/org/jetbrains/kotlin/idea/intentions/IntentionTestGenerated.java @@ -2499,6 +2499,11 @@ public class IntentionTestGenerated extends AbstractIntentionTest { runTest("idea/testData/intentions/branched/ifThenToElvis/ifAsPartOfExpression.kt"); } + @TestMetadata("ifToElvisSwallowComments.kt") + public void testIfToElvisSwallowComments() throws Exception { + runTest("idea/testData/intentions/branched/ifThenToElvis/ifToElvisSwallowComments.kt"); + } + @TestMetadata("implicitReceiver.kt") public void testImplicitReceiver() throws Exception { runTest("idea/testData/intentions/branched/ifThenToElvis/implicitReceiver.kt");