Mark all generated getNodeLists methods generated

1 files changed, 6 insertions, 18 deletions

diff --git a/javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/node/GetNodeListsGenerator.java b/javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/node/GetNodeListsGenerator.java
index 676d14cd9..85550ce99 100644
--- a/javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/node/GetNodeListsGenerator.java
+++ b/javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/node/GetNodeListsGenerator.java
@@ -1,18 +1,13 @@
 package com.github.javaparser.generator.core.node;
 
-import com.github.javaparser.JavaParser;
 import com.github.javaparser.ast.CompilationUnit;
-import com.github.javaparser.ast.Node;
 import com.github.javaparser.ast.body.ClassOrInterfaceDeclaration;
 import com.github.javaparser.ast.body.MethodDeclaration;
-import com.github.javaparser.ast.stmt.BlockStmt;
 import com.github.javaparser.generator.NodeGenerator;
-import com.github.javaparser.utils.SeparatedItemStringBuilder;
-import com.github.javaparser.utils.SourceRoot;
 import com.github.javaparser.metamodel.BaseNodeMetaModel;
 import com.github.javaparser.metamodel.PropertyMetaModel;
-
-import java.util.List;
+import com.github.javaparser.utils.SeparatedItemStringBuilder;
+import com.github.javaparser.utils.SourceRoot;
 
 import static com.github.javaparser.JavaParser.parseClassBodyDeclaration;
 import static com.github.javaparser.utils.CodeGenerationUtils.f;
@@ -38,20 +33,13 @@ public class GetNodeListsGenerator extends NodeGenerator {
                 }
             }
         }
-
-        List<MethodDeclaration> getNodeListsMethods = nodeCoid.getMethodsByName("getNodeLists");
+        
         if (!statement.hasItems()) {
-            getNodeListsMethods.forEach(Node::remove);
-            return;
-        }
-
-        if (getNodeListsMethods.isEmpty()) {
-            nodeCoid.addMember(parseClassBodyDeclaration(f("@Override public List<NodeList<?>> getNodeLists() {%s}", statement)));
             return;
         }
 
-        BlockStmt block = getNodeListsMethods.get(0).getBody().get();
-        block.getStatements().clear();
-        block.addStatement(statement.toString());
+        final MethodDeclaration getNodeListsMethod = (MethodDeclaration) parseClassBodyDeclaration(f("@Override public List<NodeList<?>> getNodeLists() {%s}", statement));
+        addOrReplaceWhenSameSignature(nodeCoid, getNodeListsMethod);
+        markGenerated(getNodeListsMethod);
     }
 }