SSR: fix method body whitespace and restore test

This commit is contained in:
Bas Leijdekkers
2016-10-13 10:15:41 +02:00
parent 4efe283ebe
commit 49564f7ffd
2 changed files with 14 additions and 10 deletions
@@ -283,6 +283,10 @@ public class JavaReplaceHandler extends StructuralReplaceHandler {
private static void copyMethodBodyIfNotReplaced(PsiMethod original, PsiMethod query, PsiMethod replacement) {
final PsiCodeBlock originalBody = original.getBody();
if (originalBody != null && query.getBody() == null && replacement.getBody() == null) {
final PsiElement sibling = originalBody.getPrevSibling();
if (sibling instanceof PsiWhiteSpace) {
replacement.add(sibling);
}
replacement.add(originalBody);
}
}