1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/test/tools/javac/processing/environment/round/TestContext.java Tue Oct 05 11:34:43 2010 -0700
1.3 @@ -0,0 +1,96 @@
1.4 +/*
1.5 + * Copyright (c) 2010, Oracle and/or its affiliates. All rights reserved.
1.6 + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
1.7 + *
1.8 + * This code is free software; you can redistribute it and/or modify it
1.9 + * under the terms of the GNU General Public License version 2 only, as
1.10 + * published by the Free Software Foundation.
1.11 + *
1.12 + * This code is distributed in the hope that it will be useful, but WITHOUT
1.13 + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
1.14 + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
1.15 + * version 2 for more details (a copy is included in the LICENSE file that
1.16 + * accompanied this code).
1.17 + *
1.18 + * You should have received a copy of the GNU General Public License version
1.19 + * 2 along with this work; if not, write to the Free Software Foundation,
1.20 + * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
1.21 + *
1.22 + * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
1.23 + * or visit www.oracle.com if you need additional information or have any
1.24 + * questions.
1.25 + */
1.26 +
1.27 +/*
1.28 + * @test
1.29 + * @bug 6988836
1.30 + * @summary A new JavacElements is created for each round of annotation processing
1.31 + * @library ../../../lib
1.32 + * @build JavacTestingAbstractProcessor TestContext
1.33 + * @compile/process -processor TestContext -XprintRounds TestContext
1.34 + */
1.35 +
1.36 +import java.io.*;
1.37 +import java.util.*;
1.38 +import javax.annotation.processing.*;
1.39 +import javax.lang.model.element.*;
1.40 +import javax.tools.*;
1.41 +import static javax.tools.Diagnostic.Kind.*;
1.42 +
1.43 +import com.sun.source.util.Trees;
1.44 +import com.sun.tools.javac.api.JavacTrees;
1.45 +import com.sun.tools.javac.model.JavacElements;
1.46 +import com.sun.tools.javac.model.JavacTypes;
1.47 +import com.sun.tools.javac.processing.JavacProcessingEnvironment;
1.48 +import com.sun.tools.javac.util.Context;
1.49 +
1.50 +public class TestContext extends JavacTestingAbstractProcessor {
1.51 +
1.52 + Trees treeUtils;
1.53 + int round = 0;
1.54 +
1.55 + @Override
1.56 + public void init(ProcessingEnvironment pEnv) {
1.57 + super.init(pEnv);
1.58 + treeUtils = Trees.instance(processingEnv);
1.59 + }
1.60 +
1.61 + @Override
1.62 + public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
1.63 + round++;
1.64 +
1.65 + JavacProcessingEnvironment jpe = (JavacProcessingEnvironment) processingEnv;
1.66 + Context c = jpe.getContext();
1.67 + check(c.get(JavacElements.class), eltUtils);
1.68 + check(c.get(JavacTypes.class), typeUtils);
1.69 + check(c.get(JavacTrees.class), treeUtils);
1.70 +
1.71 + final int MAXROUNDS = 3;
1.72 + if (round < MAXROUNDS)
1.73 + generateSource("Gen" + round);
1.74 +
1.75 + return true;
1.76 + }
1.77 +
1.78 + <T> void check(T actual, T expected) {
1.79 +// messager.printMessage(NOTE, "expect: " + expected);
1.80 +// messager.printMessage(NOTE, "actual: " + actual);
1.81 +
1.82 + if (actual != expected) {
1.83 + messager.printMessage(ERROR,
1.84 + "round " + round + " unexpected value for " + expected.getClass().getName() + ": " + actual);
1.85 + }
1.86 + }
1.87 +
1.88 + void generateSource(String name) {
1.89 + String text = "class " + name + " { }\n";
1.90 +
1.91 + try (Writer out = filer.createSourceFile(name).openWriter()) {
1.92 + out.write(text);
1.93 + } catch (IOException e) {
1.94 + throw new Error(e);
1.95 + }
1.96 + }
1.97 +
1.98 +}
1.99 +
