1 /*
   2  * Copyright (c) 2006, 2016, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.
   8  *
   9  * This code is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12  * version 2 for more details (a copy is included in the LICENSE file that
  13  * accompanied this code).
  14  *
  15  * You should have received a copy of the GNU General Public License version
  16  * 2 along with this work; if not, write to the Free Software Foundation,
  17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  18  *
  19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  20  * or visit www.oracle.com if you need additional information or have any
  21  * questions.
  22  */
  23 
  24 /*
  25  * @test
  26  * @bug 6358024
  27  * @summary TaskListener should be propogated between processing rounds
  28  * @modules jdk.compiler/com.sun.tools.javac.api
  29  *          jdk.compiler/com.sun.tools.javac.file
  30  *          jdk.compiler/com.sun.tools.javac.util
  31  */
  32 
  33 import java.io.*;
  34 import java.util.*;
  35 import java.util.List;
  36 import javax.annotation.processing.*;
  37 import javax.lang.model.element.*;
  38 import javax.tools.*;
  39 import com.sun.source.util.*;
  40 import com.sun.tools.javac.api.*;
  41 import com.sun.tools.javac.file.JavacFileManager;
  42 import com.sun.tools.javac.util.*;
  43 
  44 
  45 @SupportedAnnotationTypes("*")
  46 public class T6358024 extends AbstractProcessor {
  47     static JavacFileManager fm;
  48     public static void main(String... args) throws Throwable {
  49         String self = T6358024.class.getName();
  50 
  51         String testSrc = System.getProperty("test.src");
  52 
  53         fm = new JavacFileManager(new Context(), false, null);
  54         JavaFileObject f = fm.getJavaFileObject(testSrc + File.separatorChar + self + ".java");
  55 
  56         test(fm, f,
  57              new Option[] { new Option("-d", ".")},
  58              8);
  59 
  60         test(fm, f,
  61              new Option[] { new XOption("-XprintRounds"),
  62                             new Option("-processorpath", "."),
  63                             new Option("-processor", self) },
  64              13);
  65     }
  66 
  67     static void test(JavacFileManager fm, JavaFileObject f, Option[] opts, int expect) throws Throwable {
  68         PrintWriter out = new PrintWriter(System.err, true);
  69 
  70         JavacTool tool = JavacTool.create();
  71         List<String> flags = new ArrayList<String>();
  72         flags.addAll(Arrays.asList(
  73                 "--add-exports", "jdk.compiler/com.sun.tools.javac.api=ALL-UNNAMED",
  74                 "--add-exports", "jdk.compiler/com.sun.tools.javac.file=ALL-UNNAMED",
  75                 "--add-exports", "jdk.compiler/com.sun.tools.javac.util=ALL-UNNAMED"));
  76         for (Option opt: opts) {
  77             flags.add(opt.name);
  78             for (Object arg : opt.args)
  79                 flags.add(arg.toString());
  80         }
  81 
  82         JavacTaskImpl task = (JavacTaskImpl) tool.getTask(out,
  83                                                           fm,
  84                                                           null,
  85                                                           flags,
  86                                                           null,
  87                                                           Arrays.asList(f));
  88         MyTaskListener tl = new MyTaskListener();
  89         task.setTaskListener(tl);
  90         task.call();
  91         if (tl.started != expect)
  92             throw new AssertionError("Unexpected number of TaskListener events; "
  93                                      + "expected " + expect + ", found " + tl.started);
  94     }
  95 
  96     public boolean process(Set<? extends TypeElement> tes, RoundEnvironment renv) {
  97         return true;
  98     }
  99 
 100     static class MyTaskListener implements TaskListener {
 101         public void started(TaskEvent e) {
 102             System.err.println("Started: " + e);
 103             started++;
 104         }
 105         public void finished(TaskEvent e) {
 106         }
 107 
 108         int started = 0;
 109     }
 110 
 111     static class Option {
 112         Option(String name, String... args) {
 113             this.name = name;
 114             this.args = args;
 115         }
 116         public final String name;
 117         public final String[] args;
 118     }
 119 
 120     static class XOption extends Option {
 121         XOption(String name, String... args) {
 122             super(name, args);
 123         }
 124     }
 125 }