Get ASTNode from ASTProvider (#1052)

Author: jdneo

java-extension/com.microsoft.java.test.plugin/src/main/java/com/microsoft/java/test/plugin/util/TestSearchUtils.java

@@ -41,6 +41,7 @@
 import org.eclipse.jdt.core.dom.CompilationUnit;
 import org.eclipse.jdt.core.dom.ITypeBinding;
 import org.eclipse.jdt.core.dom.TypeDeclaration;
+import org.eclipse.jdt.core.manipulation.CoreASTProvider;
 import org.eclipse.jdt.core.search.IJavaSearchConstants;
 import org.eclipse.jdt.core.search.IJavaSearchScope;
 import org.eclipse.jdt.core.search.SearchEngine;
@@ -111,12 +112,16 @@ public static List<TestItem> searchCodeLens(List<Object> arguments, IProgressMon
     }
 
     public static ASTNode parseToAst(final ICompilationUnit unit, IProgressMonitor monitor) {
-        final ASTParser parser = ASTParser.newParser(AST.JLS14);
-        parser.setSource(unit);
-        parser.setFocalPosition(0);
-        parser.setResolveBindings(true);
-        parser.setIgnoreMethodBodies(true);
-        return parser.createAST(monitor);
+        final CompilationUnit astRoot = CoreASTProvider.getInstance().getAST(unit, CoreASTProvider.WAIT_YES, monitor);
+        if (astRoot == null) {
+            final ASTParser parser = ASTParser.newParser(AST.JLS14);
+            parser.setSource(unit);
+            parser.setFocalPosition(0);
+            parser.setResolveBindings(true);
+            parser.setIgnoreMethodBodies(true);
+            return parser.createAST(monitor);
+        }
+        return astRoot;
     }
 
     /**

test/test-projects/modular-gradle/build.gradle

@@ -7,7 +7,7 @@ repositories {
 }
 
 dependencies {
-    testImplementation('org.junit.jupiter:junit-jupiter:5.5.2')
+    testImplementation('org.junit.jupiter:junit-jupiter:5.6.0')
 }
 
 test {