jdk

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
}
126