IDEA-204900 Smart Enter fails to add function body if staying at annotation

GitOrigin-RevId: 1216c56cb12209f569f88ce4afea2528166b3437

java/java-impl/src/com/intellij/codeInsight/editorActions/smartEnter/JavaSmartEnterProcessor.java

@@ -289,7 +289,10 @@ public class JavaSmartEnterProcessor extends SmartEnterProcessor {
     }
 
     for (PsiElement each : SyntaxTraverser.psiApi().parents(atCaret).skip(1)) {
-      if (each instanceof PsiMember || each instanceof PsiAnnotation || each instanceof PsiImportStatementBase || each instanceof PsiPackageStatement) {
+      if (each instanceof PsiMember ||
+          each instanceof PsiImportStatementBase ||
+          each instanceof PsiPackageStatement ||
+          each instanceof PsiAnnotation && PsiTreeUtil.hasErrorElements(each)) {
         return each;
       }
       if (each instanceof PsiCodeBlock || each instanceof PsiComment) {

java/java-tests/testData/codeInsight/completeStatement/AddMethodBodyFromInsideAnnotation.java

@@ -0,0 +1,3 @@
+class A {
+    void foo(@Not<caret>Null String p)
+}

java/java-tests/testData/codeInsight/completeStatement/AddMethodBodyFromInsideAnnotation_after.java

@@ -0,0 +1,5 @@
+class A {
+    void foo(@NotNull String p) {
+        <caret>
+    }
+}

java/java-tests/testSrc/com/intellij/java/codeInsight/CompleteStatementTest.java

@@ -178,6 +178,7 @@ public class CompleteStatementTest extends EditorActionTestCase {
   public void testDefaultMethodBody() { doTest(); }
   public void testStaticInterfaceMethodBody() { doTest(); }
   public void testPrivateInterfaceMethodBody() { doTest(); }
+  public void testAddMethodBodyFromInsideAnnotation() { doTest(); }
   public void testArrayInitializerRBracket() { doTest(); }
   public void testArrayInitializerRBrace() { doTest(); }
   public void testArrayInitializerSeveralLines() { doTest(); }