--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/langtools/test/tools/javac/api/TestTrees.java Sat Dec 01 00:00:00 2007 +0000
@@ -0,0 +1,216 @@
+/*
+ * Copyright 2006 Sun Microsystems, Inc. All Rights Reserved.
+ * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
+ *
+ * This code is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License version 2 only, as
+ * published by the Free Software Foundation.
+ *
+ * This code is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
+ * version 2 for more details (a copy is included in the LICENSE file that
+ * accompanied this code).
+ *
+ * You should have received a copy of the GNU General Public License version
+ * 2 along with this work; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
+ *
+ * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
+ * CA 95054 USA or visit www.sun.com if you need additional information or
+ * have any questions.
+ */
+
+/*
+ * @test
+ * @bug 6346249 6392177
+ * @summary new Trees API
+ */
+
+import com.sun.source.tree.*;
+import com.sun.source.util.*;
+import java.io.*;
+import java.lang.annotation.*;
+import java.util.*;
+import javax.annotation.processing.*;
+import javax.lang.model.SourceVersion;
+import javax.lang.model.element.*;
+import javax.lang.model.type.*;
+import javax.tools.*;
+import com.sun.tools.javac.api.JavacTool;
+import com.sun.tools.javac.tree.JCTree;
+import com.sun.tools.javac.tree.TreeInfo;
+
+@Anno
+@SupportedAnnotationTypes("*")
+public class TestTrees extends AbstractProcessor {
+
+ @Anno
+ void annoMethod() { }
+
+ @Anno
+ int annoField;
+
+
+ static final String testSrcDir = System.getProperty("test.src");
+ static final String testClassDir = System.getProperty("test.classes");
+ static final String self = TestTrees.class.getName();
+ static PrintWriter out = new PrintWriter(System.err, true);
+
+ public static void main(String[] args) throws IOException {
+ new TestTrees().run();
+ }
+
+ void run() throws IOException {
+
+ JavacTool tool = JavacTool.create();
+
+ DiagnosticListener<JavaFileObject> dl = new DiagnosticListener<JavaFileObject>() {
+ public void report(Diagnostic d) {
+ error(d.toString());
+ }
+ };
+
+ StandardJavaFileManager fm = tool.getStandardFileManager(dl, null, null);
+ Iterable<? extends JavaFileObject> files =
+ fm.getJavaFileObjectsFromFiles(Arrays.asList(new File(testSrcDir, self + ".java")));
+
+ Iterable<String> opts = Arrays.asList("-d", ".");
+
+ System.err.println("simple compilation, no processing");
+ JavacTask task = tool.getTask(out, fm, dl, opts, null, files);
+ task.setTaskListener(new MyTaskListener(task));
+ if (!task.call())
+ throw new AssertionError("compilation failed");
+
+ opts = Arrays.asList("-d", ".", "-processorpath", testClassDir, "-processor", self);
+
+ System.err.println();
+ System.err.println("compilation with processing");
+ task = tool.getTask(out, fm, dl,opts, null, files);
+ if (!task.call())
+ throw new AssertionError("compilation failed");
+
+ if (errors > 0)
+ throw new AssertionError(errors + " errors occurred");
+ }
+
+ void testElement(Trees trees, Element e) {
+ trees.getClass();
+ e.getClass();
+
+ System.err.println("testElement: " + e);
+ Tree tree = trees.getTree(e);
+ //System.err.println(tree);
+
+ if (TreeInfo.symbolFor((JCTree)tree) != e)
+ error("bad result from getTree");
+
+ TreePath path = trees.getPath(e);
+ if (path == null) {
+ error("getPath returned null");
+ return;
+ }
+ if (path.getLeaf() != tree)
+ error("bad result from getPath");
+
+ Element e2 = trees.getElement(path);
+ if (e2 == null) {
+ error("getElement returned null");
+ return;
+ }
+ if (e2 != e)
+ error("bad result from getElement");
+
+ // The TypeMirror is not available yet when annotation processing;
+ // it is set up later during ANALYSE.
+ TypeMirror t = trees.getTypeMirror(path);
+ if (t != null && t.getKind() == TypeKind.DECLARED &&
+ ((DeclaredType)t).asElement() != e2)
+ error("bad result from getTypeMirror");
+
+ for (AnnotationMirror m: e.getAnnotationMirrors()) {
+ testAnnotation(trees, e, m);
+ }
+ }
+
+ void testAnnotation(Trees trees, Element e, AnnotationMirror a) {
+ System.err.println("testAnnotation: " + e + " " + a);
+ Tree tree = trees.getTree(e, a);
+
+ if (tree.getKind() != Tree.Kind.ANNOTATION)
+ error("bad result from getTree");
+
+ TreePath path = trees.getPath(e, a);
+ if (path.getLeaf() != tree)
+ error("bad result from getPath");
+ }
+
+ void error(String msg) {
+ if (messager != null)
+ // annotation processing will happen in a separate instance/classloader
+ // so pass the message back to the calling instance.
+ messager.printMessage(Diagnostic.Kind.ERROR, msg);
+ else {
+ System.err.println(msg);
+ errors++;
+ }
+
+ }
+
+ Messager messager;
+ int errors;
+
+
+ public boolean process(Set<? extends TypeElement> annos, RoundEnvironment rEnv) {
+ Trees trees = Trees.instance(processingEnv);
+ messager = processingEnv.getMessager();
+
+ for (Element e: rEnv.getRootElements()) {
+ testElement(trees, e);
+ }
+
+ for (TypeElement anno: annos) {
+ Set<? extends Element> elts = rEnv.getElementsAnnotatedWith(anno);
+ System.err.println("anno: " + anno);
+ System.err.println("elts: " + elts);
+ if (elts != null) { // 6397298, should return empty set
+ for (Element e: rEnv.getElementsAnnotatedWith(anno))
+ testElement(trees, e);
+ }
+ }
+
+ return true;
+ }
+
+ @Override
+ public SourceVersion getSupportedSourceVersion() {
+ return SourceVersion.latest();
+ }
+
+ class MyTaskListener implements TaskListener {
+ MyTaskListener(JavacTask task) {
+ this.task = task;
+ }
+
+ public void started(TaskEvent e) {
+ System.err.println("started " + e);
+ }
+
+ public void finished(TaskEvent e) {
+ //System.err.println("finished " + e);
+ switch (e.getKind()) {
+ case ANALYZE:
+ testElement(Trees.instance(task), e.getTypeElement());
+ break;
+ }
+ }
+
+ private final JavacTask task;
+ }
+
+}
+
+@Retention(RetentionPolicy.SOURCE)
+@interface Anno {
+}