Return correct parent for elements under when expression body

2017-09-14 17:42:09 +02:00
parent c98c3a7033
commit adf1df3eb0
1 changed files with 5 additions and 0 deletions
@@ -72,6 +72,11 @@ fun doConvertParent(element: UElement, parent: PsiElement?): UElement? {
    if (result is UEnumConstant && element is UDeclaration) {
        return result.initializingClass
    }
+
+    if (result is USwitchClauseExpressionWithBody && element.psi !is KtWhenCondition) {
+        return result.body
+    }
+
    return result
 }