Intention to convert assert into an if with throw - bug fix for when the call is dot qualified

This commit is contained in:
Tal Man
2014-04-20 12:45:31 -04:00
parent feece3c326
commit 3713f50800
4 changed files with 22 additions and 1 deletions
@@ -28,6 +28,7 @@ import org.jetbrains.jet.lang.resolve.DescriptorUtils
import kotlin.properties.Delegates
import org.jetbrains.jet.lang.types.lang.KotlinBuiltIns
import org.jetbrains.jet.lang.psi.JetIfExpression
import org.jetbrains.jet.lang.psi.JetDotQualifiedExpression
public class ConvertAssertToIfWithThrowIntention : JetSelfTargetingIntention<JetCallExpression>(
"convert.assert.to.if.with.throw", javaClass()) {
@@ -98,6 +99,11 @@ public class ConvertAssertToIfWithThrowIntention : JetSelfTargetingIntention<Jet
simplifier.applyTo(ifCondition, editor)
}
element.replace(ifExpression)
val parent = element.getParent()
if (parent is JetDotQualifiedExpression) {
parent.replace(ifExpression)
} else {
element.replace(ifExpression)
}
}
}