test/tools/javac/util/T6597678.java

changeset 0
959103a6100f
child 2525
2eb010b6cb22
equal deleted inserted replaced
-1:000000000000 0:959103a6100f
1 /*
2 * Copyright (c) 2011, 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 6597678 6449184
27 * @summary Ensure Messages propogated between rounds
28 * @library /tools/javac/lib
29 * @build JavacTestingAbstractProcessor T6597678
30 * @run main T6597678
31 */
32
33 import java.io.*;
34 import java.util.*;
35 import javax.annotation.processing.RoundEnvironment;
36 import javax.annotation.processing.SupportedOptions;
37 import javax.lang.model.element.TypeElement;
38 import javax.tools.Diagnostic;
39
40
41 import com.sun.tools.javac.processing.JavacProcessingEnvironment;
42 import com.sun.tools.javac.util.Context;
43 import com.sun.tools.javac.util.JavacMessages;
44 import com.sun.tools.javac.util.Log;
45
46 @SupportedOptions("WriterString")
47 public class T6597678 extends JavacTestingAbstractProcessor {
48 public static void main(String... args) throws Exception {
49 new T6597678().run();
50 }
51
52 void run() throws Exception {
53 String myName = T6597678.class.getSimpleName();
54 File testSrc = new File(System.getProperty("test.src"));
55 File file = new File(testSrc, myName + ".java");
56
57 StringWriter sw = new StringWriter();
58 PrintWriter pw = new PrintWriter(sw);
59
60 compile(sw, pw,
61 "-proc:only",
62 "-processor", myName,
63 "-AWriterString=" + pw.toString(),
64 file.getPath());
65 }
66
67 void compile(StringWriter sw, PrintWriter pw, String... args) throws Exception {
68 int rc = com.sun.tools.javac.Main.compile(args, pw);
69 pw.close();
70 String out = sw.toString();
71 if (!out.isEmpty())
72 System.err.println(out);
73 if (rc != 0)
74 throw new Exception("compilation failed unexpectedly: rc=" + rc);
75 }
76
77 //---------------
78
79 @Override
80 public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
81 Context context = ((JavacProcessingEnvironment) processingEnv).getContext();
82 Log log = Log.instance(context);
83 PrintWriter noteOut = log.getWriter(Log.WriterKind.NOTICE);
84 PrintWriter warnOut = log.getWriter(Log.WriterKind.WARNING);
85 PrintWriter errOut = log.getWriter(Log.WriterKind.ERROR);
86 Locale locale = context.get(Locale.class);
87 JavacMessages messages = context.get(JavacMessages.messagesKey);
88
89 round++;
90 if (round == 1) {
91 initialLocale = locale;
92 initialMessages = messages;
93 initialNoteWriter = noteOut;
94 initialWarnWriter = warnOut;
95 initialErrWriter = errOut;
96
97 String writerStringOpt = options.get("WriterString").intern();
98 checkEqual("noteWriterString", noteOut.toString().intern(), writerStringOpt);
99 checkEqual("warnWriterString", warnOut.toString().intern(), writerStringOpt);
100 checkEqual("errWriterString", errOut.toString().intern(), writerStringOpt);
101 } else {
102 checkEqual("locale", locale, initialLocale);
103 checkEqual("messages", messages, initialMessages);
104 checkEqual("noteWriter", noteOut, initialNoteWriter);
105 checkEqual("warnWriter", warnOut, initialWarnWriter);
106 checkEqual("errWriter", errOut, initialErrWriter);
107 }
108
109 return true;
110 }
111
112 <T> void checkEqual(String label, T actual, T expected) {
113 if (actual != expected)
114 messager.printMessage(Diagnostic.Kind.ERROR,
115 "Unexpected value for " + label
116 + "; expected: " + expected
117 + "; found: " + actual);
118 }
119
120 int round = 0;
121 Locale initialLocale;
122 JavacMessages initialMessages;
123 PrintWriter initialNoteWriter;
124 PrintWriter initialWarnWriter;
125 PrintWriter initialErrWriter;
126 }

mercurial