diff options
Diffstat (limited to 'javaparser-core/src/main/java/com/github/javaparser/ast/stmt/DoStmt.java')
-rw-r--r-- | javaparser-core/src/main/java/com/github/javaparser/ast/stmt/DoStmt.java | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/javaparser-core/src/main/java/com/github/javaparser/ast/stmt/DoStmt.java b/javaparser-core/src/main/java/com/github/javaparser/ast/stmt/DoStmt.java index c416aac79..072672bec 100644 --- a/javaparser-core/src/main/java/com/github/javaparser/ast/stmt/DoStmt.java +++ b/javaparser-core/src/main/java/com/github/javaparser/ast/stmt/DoStmt.java @@ -139,6 +139,14 @@ public final class DoStmt extends Statement implements NodeWithBody<DoStmt>, Nod public boolean replace(Node node, Node replacementNode) { if (node == null) return false; + if (node == body) { + setBody((Statement) replacementNode); + return true; + } + if (node == condition) { + setCondition((Expression) replacementNode); + return true; + } return super.replace(node, replacementNode); } } |