Add methods to navigate to the root node of an AST

1 files changed, 13 insertions, 0 deletions

diff --git a/javaparser-testing/src/test/java/com/github/javaparser/ast/NodeTest.java b/javaparser-testing/src/test/java/com/github/javaparser/ast/NodeTest.java
index b682edce2..aef89a87b 100644
--- a/javaparser-testing/src/test/java/com/github/javaparser/ast/NodeTest.java
+++ b/javaparser-testing/src/test/java/com/github/javaparser/ast/NodeTest.java
@@ -327,4 +327,17 @@ public class NodeTest {
         // This will throw an exception if the parents are bad.
         System.out.println(unit.toString());
     }
+    
+    @Test
+    public void findCompilationUnit() {
+        CompilationUnit cu = JavaParser.parse("class X{int x;}");
+        VariableDeclarator x = cu.getClassByName("X").get().getMember(0).asFieldDeclaration().getVariables().get(0);
+        assertEquals(cu, x.findCompilationUnit().get());
+    }
+    
+    @Test
+    public void cantFindCompilationUnit() {
+        VariableDeclarator x = new VariableDeclarator();
+        assertFalse(x.findCompilationUnit().isPresent());
+    }
 }