--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/langtools/test/tools/javac/T6403466.java Sat Dec 01 00:00:00 2007 +0000
@@ -0,0 +1,146 @@
+/*
+ * 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 6403466
+ * @summary javac TaskListener should be informed when annotation processing occurs
+ */
+
+import com.sun.source.util.*;
+import java.io.*;
+import java.lang.annotation.*;
+import java.util.*;
+import javax.annotation.processing.*;
+import javax.lang.model.*;
+import javax.lang.model.element.*;
+import javax.lang.model.type.*;
+import javax.lang.model.util.*;
+import javax.tools.*;
+import com.sun.tools.javac.api.JavacTool;
+
+@Wrap
+@SupportedAnnotationTypes("Wrap")
+@SupportedSourceVersion(SourceVersion.RELEASE_6)
+public class T6403466 extends AbstractProcessor {
+
+ static final String testSrcDir = System.getProperty("test.src");
+ static final String testClassDir = System.getProperty("test.classes");
+ static final String self = T6403466.class.getName();
+ static PrintWriter out = new PrintWriter(System.err, true);
+
+ public static void main(String[] args) throws IOException {
+ JavacTool tool = JavacTool.create();
+
+ StandardJavaFileManager fm = tool.getStandardFileManager(null, null, null);
+ Iterable<? extends JavaFileObject> files =
+ fm.getJavaFileObjectsFromFiles(Arrays.asList(new File(testSrcDir, self + ".java")));
+
+ Iterable<String> options = Arrays.asList("-processorpath", testClassDir,
+ "-processor", self,
+ "-s", ".",
+ "-d", ".");
+ JavacTask task = tool.getTask(out, fm, null, options, null, files);
+
+ VerifyingTaskListener vtl = new VerifyingTaskListener(new File(testSrcDir, self + ".out"));
+ task.setTaskListener(vtl);
+
+ if (!task.call())
+ throw new AssertionError("compilation failed");
+
+ if (vtl.iter.hasNext() || vtl.errors)
+ throw new AssertionError("comparison against golden file failed.");
+ }
+
+ public boolean process(Set<? extends TypeElement> annos, RoundEnvironment rEnv) {
+ Filer filer = processingEnv.getFiler();
+ for (TypeElement anno: annos) {
+ Set<? extends Element> elts = rEnv.getElementsAnnotatedWith(anno);
+ System.err.println("anno: " + anno);
+ System.err.println("elts: " + elts);
+ for (TypeElement te: ElementFilter.typesIn(elts)) {
+ try {
+ Writer out = filer.createSourceFile(te.getSimpleName() + "Wrapper").openWriter();
+ out.write("class " + te.getSimpleName() + "Wrapper { }");
+ out.close();
+ } catch (IOException ex) {
+ ex.printStackTrace();
+ }
+ }
+
+ }
+ return true;
+ }
+}
+
+@Retention(RetentionPolicy.SOURCE)
+@Target(ElementType.TYPE)
+@interface Wrap {
+}
+
+
+class VerifyingTaskListener implements TaskListener {
+ VerifyingTaskListener(File ref) throws IOException {
+ BufferedReader in = new BufferedReader(new FileReader(ref));
+ String line;
+ List<String> lines = new ArrayList<String>();
+ while ((line = in.readLine()) != null)
+ lines.add(line);
+ in.close();
+ iter = lines.iterator();
+ }
+
+ public void started(TaskEvent e) {
+ check("Started " + toString(e));
+ }
+ public void finished(TaskEvent e) {
+ check("Finished " + toString(e));
+ }
+
+ // similar to TaskEvent.toString(), but just prints basename of the file
+ private String toString(TaskEvent e) {
+ JavaFileObject file = e.getSourceFile();
+ return "TaskEvent["
+ + e.getKind() + ","
+ + (file == null ? null : new File(file.toUri().getPath()).getName()) + ","
+ // the compilation unit is identified by the file
+ + e.getTypeElement() + "]";
+ }
+
+ private void check(String s) {
+ System.out.println(s); // write a reference copy of expected output to stdout
+
+ String ref = iter.hasNext() ? iter.next() : null;
+ line++;
+ if (!s.equals(ref)) {
+ if (ref != null)
+ System.err.println(line + ": expected: " + ref);
+ System.err.println(line + ": found: " + s);
+ errors = true;
+ }
+ }
+
+ Iterator<String> iter;
+ int line;
+ boolean errors;
+}