test/tools/javac/T6403466.java

Print this page




  24 /*
  25  * @test
  26  * @bug 6403466
  27  * @summary javac TaskListener should be informed when annotation processing occurs
  28  */
  29 
  30 import com.sun.source.util.*;
  31 import java.io.*;
  32 import java.lang.annotation.*;
  33 import java.util.*;
  34 import javax.annotation.processing.*;
  35 import javax.lang.model.*;
  36 import javax.lang.model.element.*;
  37 import javax.lang.model.type.*;
  38 import javax.lang.model.util.*;
  39 import javax.tools.*;
  40 import com.sun.tools.javac.api.JavacTool;
  41 
  42 @Wrap
  43 @SupportedAnnotationTypes("Wrap")
  44 @SupportedSourceVersion(SourceVersion.RELEASE_6)
  45 public class T6403466 extends AbstractProcessor {
  46 
  47     static final String testSrcDir = System.getProperty("test.src");
  48     static final String testClassDir = System.getProperty("test.classes");
  49     static final String self = T6403466.class.getName();
  50     static PrintWriter out = new PrintWriter(System.err, true);
  51 
  52     public static void main(String[] args) throws IOException {
  53         JavacTool tool = JavacTool.create();
  54 
  55         StandardJavaFileManager fm = tool.getStandardFileManager(null, null, null);
  56         Iterable<? extends JavaFileObject> files =
  57             fm.getJavaFileObjectsFromFiles(Arrays.asList(new File(testSrcDir, self + ".java")));
  58 
  59         Iterable<String> options = Arrays.asList("-processorpath", testClassDir,
  60                                                  "-processor", self,
  61                                                  "-s", ".",
  62                                                  "-d", ".");
  63         JavacTask task = tool.getTask(out, fm, null, options, null, files);
  64 
  65         VerifyingTaskListener vtl = new VerifyingTaskListener(new File(testSrcDir, self + ".out"));
  66         task.setTaskListener(vtl);
  67 
  68         if (!task.call())
  69             throw new AssertionError("compilation failed");
  70 
  71         if (vtl.iter.hasNext() || vtl.errors)
  72             throw new AssertionError("comparison against golden file failed.");
  73     }
  74 
  75     public boolean process(Set<? extends TypeElement> annos, RoundEnvironment rEnv) {

  76         Filer filer = processingEnv.getFiler();
  77         for (TypeElement anno: annos) {
  78             Set<? extends Element> elts = rEnv.getElementsAnnotatedWith(anno);
  79             System.err.println("anno: " + anno);
  80             System.err.println("elts: " + elts);
  81             for (TypeElement te: ElementFilter.typesIn(elts)) {
  82                 try {
  83                     Writer out = filer.createSourceFile(te.getSimpleName() + "Wrapper").openWriter();
  84                     out.write("class " + te.getSimpleName() + "Wrapper { }");
  85                     out.close();
  86                 } catch (IOException ex) {
  87                     ex.printStackTrace();
  88                 }
  89             }
  90 
  91         }

  92         return true;
  93     }





  94 }
  95 
  96 @Retention(RetentionPolicy.SOURCE)
  97 @Target(ElementType.TYPE)
  98 @interface Wrap {
  99 }
 100 
 101 
 102 class VerifyingTaskListener implements TaskListener {
 103     VerifyingTaskListener(File ref) throws IOException {
 104         BufferedReader in = new BufferedReader(new FileReader(ref));
 105         String line;
 106         List<String> lines = new ArrayList<String>();
 107         while ((line = in.readLine()) != null)
 108             lines.add(line);
 109         in.close();
 110         iter = lines.iterator();
 111     }
 112 
 113     public void started(TaskEvent e) {




  24 /*
  25  * @test
  26  * @bug 6403466
  27  * @summary javac TaskListener should be informed when annotation processing occurs
  28  */
  29 
  30 import com.sun.source.util.*;
  31 import java.io.*;
  32 import java.lang.annotation.*;
  33 import java.util.*;
  34 import javax.annotation.processing.*;
  35 import javax.lang.model.*;
  36 import javax.lang.model.element.*;
  37 import javax.lang.model.type.*;
  38 import javax.lang.model.util.*;
  39 import javax.tools.*;
  40 import com.sun.tools.javac.api.JavacTool;
  41 
  42 @Wrap
  43 @SupportedAnnotationTypes("Wrap")

  44 public class T6403466 extends AbstractProcessor {
  45 
  46     static final String testSrcDir = System.getProperty("test.src");
  47     static final String testClassDir = System.getProperty("test.classes");
  48     static final String self = T6403466.class.getName();
  49     static PrintWriter out = new PrintWriter(System.err, true);
  50 
  51     public static void main(String[] args) throws IOException {
  52         JavacTool tool = JavacTool.create();
  53 
  54         StandardJavaFileManager fm = tool.getStandardFileManager(null, null, null);
  55         Iterable<? extends JavaFileObject> files =
  56             fm.getJavaFileObjectsFromFiles(Arrays.asList(new File(testSrcDir, self + ".java")));
  57 
  58         Iterable<String> options = Arrays.asList("-processorpath", testClassDir,
  59                                                  "-processor", self,
  60                                                  "-s", ".",
  61                                                  "-d", ".");
  62         JavacTask task = tool.getTask(out, fm, null, options, null, files);
  63 
  64         VerifyingTaskListener vtl = new VerifyingTaskListener(new File(testSrcDir, self + ".out"));
  65         task.setTaskListener(vtl);
  66 
  67         if (!task.call())
  68             throw new AssertionError("compilation failed");
  69 
  70         if (vtl.iter.hasNext() || vtl.errors)
  71             throw new AssertionError("comparison against golden file failed.");
  72     }
  73 
  74     public boolean process(Set<? extends TypeElement> annos, RoundEnvironment rEnv) {
  75         if (!rEnv.processingOver()) {
  76             Filer filer = processingEnv.getFiler();
  77             for (TypeElement anno: annos) {
  78                 Set<? extends Element> elts = rEnv.getElementsAnnotatedWith(anno);
  79                 System.err.println("anno: " + anno);
  80                 System.err.println("elts: " + elts);
  81                 for (TypeElement te: ElementFilter.typesIn(elts)) {
  82                     try {
  83                         Writer out = filer.createSourceFile(te.getSimpleName() + "Wrapper").openWriter();
  84                         out.write("class " + te.getSimpleName() + "Wrapper { }");
  85                         out.close();
  86                     } catch (IOException ex) {
  87                         ex.printStackTrace();
  88                     }
  89                 }
  90 
  91             }
  92         }
  93         return true;
  94     }
  95 
  96     @Override
  97     public SourceVersion getSupportedSourceVersion() {
  98         return SourceVersion.latest();
  99     }
 100 }
 101 
 102 @Retention(RetentionPolicy.SOURCE)
 103 @Target(ElementType.TYPE)
 104 @interface Wrap {
 105 }
 106 
 107 
 108 class VerifyingTaskListener implements TaskListener {
 109     VerifyingTaskListener(File ref) throws IOException {
 110         BufferedReader in = new BufferedReader(new FileReader(ref));
 111         String line;
 112         List<String> lines = new ArrayList<String>();
 113         while ((line = in.readLine()) != null)
 114             lines.add(line);
 115         in.close();
 116         iter = lines.iterator();
 117     }
 118 
 119     public void started(TaskEvent e) {