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;
main(String... args)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 
test(JavacFileManager fm, JavaFileObject f, Option[] opts, int expect)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 
process(Set<? extends TypeElement> tes, RoundEnvironment renv)96     public boolean process(Set<? extends TypeElement> tes, RoundEnvironment renv) {
97         return true;
98     }
99 
100     static class MyTaskListener implements TaskListener {
started(TaskEvent e)101         public void started(TaskEvent e) {
102             System.err.println("Started: " + e);
103             started++;
104         }
finished(TaskEvent e)105         public void finished(TaskEvent e) {
106         }
107 
108         int started = 0;
109     }
110 
111     static class Option {
Option(String name, String... args)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 {
XOption(String name, String... args)121         XOption(String name, String... args) {
122             super(name, args);
123         }
124     }
125 }
126