T6358024.java revision 3573:c4a18ee691c4 
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
33import java.io.*;
34import java.util.*;
35import java.util.List;
36import javax.annotation.processing.*;
37import javax.lang.model.element.*;
38import javax.tools.*;
39import com.sun.source.util.*;
40import com.sun.tools.javac.api.*;
41import com.sun.tools.javac.file.JavacFileManager;
42import com.sun.tools.javac.util.*;
43
44
45@SupportedAnnotationTypes("*")
46public 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}
126