test/tools/javac/T6403466.java

Fri, 21 Aug 2009 14:58:21 -0700

author
jjg
date
Fri, 21 Aug 2009 14:58:21 -0700
changeset 377
d9febdd5ae21
parent 1
9a66ca7c79fa
child 494
af18e3956985
permissions
-rw-r--r--

6873845: refine access to symbol file
Reviewed-by: darcy

duke@1 1 /*
duke@1 2 * Copyright 2006 Sun Microsystems, Inc. All Rights Reserved.
duke@1 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
duke@1 4 *
duke@1 5 * This code is free software; you can redistribute it and/or modify it
duke@1 6 * under the terms of the GNU General Public License version 2 only, as
duke@1 7 * published by the Free Software Foundation.
duke@1 8 *
duke@1 9 * This code is distributed in the hope that it will be useful, but WITHOUT
duke@1 10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
duke@1 11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
duke@1 12 * version 2 for more details (a copy is included in the LICENSE file that
duke@1 13 * accompanied this code).
duke@1 14 *
duke@1 15 * You should have received a copy of the GNU General Public License version
duke@1 16 * 2 along with this work; if not, write to the Free Software Foundation,
duke@1 17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
duke@1 18 *
duke@1 19 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
duke@1 20 * CA 95054 USA or visit www.sun.com if you need additional information or
duke@1 21 * have any questions.
duke@1 22 */
duke@1 23
duke@1 24 /*
duke@1 25 * @test
duke@1 26 * @bug 6403466
duke@1 27 * @summary javac TaskListener should be informed when annotation processing occurs
duke@1 28 */
duke@1 29
duke@1 30 import com.sun.source.util.*;
duke@1 31 import java.io.*;
duke@1 32 import java.lang.annotation.*;
duke@1 33 import java.util.*;
duke@1 34 import javax.annotation.processing.*;
duke@1 35 import javax.lang.model.*;
duke@1 36 import javax.lang.model.element.*;
duke@1 37 import javax.lang.model.type.*;
duke@1 38 import javax.lang.model.util.*;
duke@1 39 import javax.tools.*;
duke@1 40 import com.sun.tools.javac.api.JavacTool;
duke@1 41
duke@1 42 @Wrap
duke@1 43 @SupportedAnnotationTypes("Wrap")
duke@1 44 @SupportedSourceVersion(SourceVersion.RELEASE_6)
duke@1 45 public class T6403466 extends AbstractProcessor {
duke@1 46
duke@1 47 static final String testSrcDir = System.getProperty("test.src");
duke@1 48 static final String testClassDir = System.getProperty("test.classes");
duke@1 49 static final String self = T6403466.class.getName();
duke@1 50 static PrintWriter out = new PrintWriter(System.err, true);
duke@1 51
duke@1 52 public static void main(String[] args) throws IOException {
duke@1 53 JavacTool tool = JavacTool.create();
duke@1 54
duke@1 55 StandardJavaFileManager fm = tool.getStandardFileManager(null, null, null);
duke@1 56 Iterable<? extends JavaFileObject> files =
duke@1 57 fm.getJavaFileObjectsFromFiles(Arrays.asList(new File(testSrcDir, self + ".java")));
duke@1 58
duke@1 59 Iterable<String> options = Arrays.asList("-processorpath", testClassDir,
duke@1 60 "-processor", self,
duke@1 61 "-s", ".",
duke@1 62 "-d", ".");
duke@1 63 JavacTask task = tool.getTask(out, fm, null, options, null, files);
duke@1 64
duke@1 65 VerifyingTaskListener vtl = new VerifyingTaskListener(new File(testSrcDir, self + ".out"));
duke@1 66 task.setTaskListener(vtl);
duke@1 67
duke@1 68 if (!task.call())
duke@1 69 throw new AssertionError("compilation failed");
duke@1 70
duke@1 71 if (vtl.iter.hasNext() || vtl.errors)
duke@1 72 throw new AssertionError("comparison against golden file failed.");
duke@1 73 }
duke@1 74
duke@1 75 public boolean process(Set<? extends TypeElement> annos, RoundEnvironment rEnv) {
duke@1 76 Filer filer = processingEnv.getFiler();
duke@1 77 for (TypeElement anno: annos) {
duke@1 78 Set<? extends Element> elts = rEnv.getElementsAnnotatedWith(anno);
duke@1 79 System.err.println("anno: " + anno);
duke@1 80 System.err.println("elts: " + elts);
duke@1 81 for (TypeElement te: ElementFilter.typesIn(elts)) {
duke@1 82 try {
duke@1 83 Writer out = filer.createSourceFile(te.getSimpleName() + "Wrapper").openWriter();
duke@1 84 out.write("class " + te.getSimpleName() + "Wrapper { }");
duke@1 85 out.close();
duke@1 86 } catch (IOException ex) {
duke@1 87 ex.printStackTrace();
duke@1 88 }
duke@1 89 }
duke@1 90
duke@1 91 }
duke@1 92 return true;
duke@1 93 }
duke@1 94 }
duke@1 95
duke@1 96 @Retention(RetentionPolicy.SOURCE)
duke@1 97 @Target(ElementType.TYPE)
duke@1 98 @interface Wrap {
duke@1 99 }
duke@1 100
duke@1 101
duke@1 102 class VerifyingTaskListener implements TaskListener {
duke@1 103 VerifyingTaskListener(File ref) throws IOException {
duke@1 104 BufferedReader in = new BufferedReader(new FileReader(ref));
duke@1 105 String line;
duke@1 106 List<String> lines = new ArrayList<String>();
duke@1 107 while ((line = in.readLine()) != null)
duke@1 108 lines.add(line);
duke@1 109 in.close();
duke@1 110 iter = lines.iterator();
duke@1 111 }
duke@1 112
duke@1 113 public void started(TaskEvent e) {
duke@1 114 check("Started " + toString(e));
duke@1 115 }
duke@1 116 public void finished(TaskEvent e) {
duke@1 117 check("Finished " + toString(e));
duke@1 118 }
duke@1 119
duke@1 120 // similar to TaskEvent.toString(), but just prints basename of the file
duke@1 121 private String toString(TaskEvent e) {
duke@1 122 JavaFileObject file = e.getSourceFile();
duke@1 123 return "TaskEvent["
duke@1 124 + e.getKind() + ","
duke@1 125 + (file == null ? null : new File(file.toUri().getPath()).getName()) + ","
duke@1 126 // the compilation unit is identified by the file
duke@1 127 + e.getTypeElement() + "]";
duke@1 128 }
duke@1 129
duke@1 130 private void check(String s) {
duke@1 131 System.out.println(s); // write a reference copy of expected output to stdout
duke@1 132
duke@1 133 String ref = iter.hasNext() ? iter.next() : null;
duke@1 134 line++;
duke@1 135 if (!s.equals(ref)) {
duke@1 136 if (ref != null)
duke@1 137 System.err.println(line + ": expected: " + ref);
duke@1 138 System.err.println(line + ": found: " + s);
duke@1 139 errors = true;
duke@1 140 }
duke@1 141 }
duke@1 142
duke@1 143 Iterator<String> iter;
duke@1 144 int line;
duke@1 145 boolean errors;
duke@1 146 }

mercurial