test/tools/javac/processing/environment/round/TestContext.java

Wed, 27 Apr 2016 01:34:52 +0800

author
aoqi
date
Wed, 27 Apr 2016 01:34:52 +0800
changeset 0
959103a6100f
child 2525
2eb010b6cb22
permissions
-rw-r--r--

Initial load
http://hg.openjdk.java.net/jdk8u/jdk8u/langtools/
changeset: 2573:53ca196be1ae
tag: jdk8u25-b17

aoqi@0 1 /*
aoqi@0 2 * Copyright (c) 2010, 2011, Oracle and/or its affiliates. All rights reserved.
aoqi@0 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
aoqi@0 4 *
aoqi@0 5 * This code is free software; you can redistribute it and/or modify it
aoqi@0 6 * under the terms of the GNU General Public License version 2 only, as
aoqi@0 7 * published by the Free Software Foundation.
aoqi@0 8 *
aoqi@0 9 * This code is distributed in the hope that it will be useful, but WITHOUT
aoqi@0 10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
aoqi@0 11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
aoqi@0 12 * version 2 for more details (a copy is included in the LICENSE file that
aoqi@0 13 * accompanied this code).
aoqi@0 14 *
aoqi@0 15 * You should have received a copy of the GNU General Public License version
aoqi@0 16 * 2 along with this work; if not, write to the Free Software Foundation,
aoqi@0 17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
aoqi@0 18 *
aoqi@0 19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
aoqi@0 20 * or visit www.oracle.com if you need additional information or have any
aoqi@0 21 * questions.
aoqi@0 22 */
aoqi@0 23
aoqi@0 24 /*
aoqi@0 25 * @test
aoqi@0 26 * @bug 6988836
aoqi@0 27 * @summary A new JavacElements is created for each round of annotation processing
aoqi@0 28 * @library /tools/javac/lib
aoqi@0 29 * @build JavacTestingAbstractProcessor TestContext
aoqi@0 30 * @compile/process -processor TestContext -XprintRounds TestContext
aoqi@0 31 */
aoqi@0 32
aoqi@0 33 import java.io.*;
aoqi@0 34 import java.util.*;
aoqi@0 35 import javax.annotation.processing.*;
aoqi@0 36 import javax.lang.model.element.*;
aoqi@0 37 import static javax.tools.Diagnostic.Kind.*;
aoqi@0 38
aoqi@0 39 import com.sun.source.util.Trees;
aoqi@0 40 import com.sun.tools.javac.api.JavacTrees;
aoqi@0 41 import com.sun.tools.javac.model.JavacElements;
aoqi@0 42 import com.sun.tools.javac.model.JavacTypes;
aoqi@0 43 import com.sun.tools.javac.processing.JavacProcessingEnvironment;
aoqi@0 44 import com.sun.tools.javac.util.Context;
aoqi@0 45
aoqi@0 46 public class TestContext extends JavacTestingAbstractProcessor {
aoqi@0 47
aoqi@0 48 Trees treeUtils;
aoqi@0 49 int round = 0;
aoqi@0 50
aoqi@0 51 @Override
aoqi@0 52 public void init(ProcessingEnvironment pEnv) {
aoqi@0 53 super.init(pEnv);
aoqi@0 54 treeUtils = Trees.instance(processingEnv);
aoqi@0 55 }
aoqi@0 56
aoqi@0 57 @Override
aoqi@0 58 public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
aoqi@0 59 round++;
aoqi@0 60
aoqi@0 61 JavacProcessingEnvironment jpe = (JavacProcessingEnvironment) processingEnv;
aoqi@0 62 Context c = jpe.getContext();
aoqi@0 63 check(c.get(JavacElements.class), eltUtils);
aoqi@0 64 check(c.get(JavacTypes.class), typeUtils);
aoqi@0 65 check(c.get(JavacTrees.class), treeUtils);
aoqi@0 66
aoqi@0 67 final int MAXROUNDS = 3;
aoqi@0 68 if (round < MAXROUNDS)
aoqi@0 69 generateSource("Gen" + round);
aoqi@0 70
aoqi@0 71 return true;
aoqi@0 72 }
aoqi@0 73
aoqi@0 74 <T> void check(T actual, T expected) {
aoqi@0 75 // messager.printMessage(NOTE, "expect: " + expected);
aoqi@0 76 // messager.printMessage(NOTE, "actual: " + actual);
aoqi@0 77
aoqi@0 78 if (actual != expected) {
aoqi@0 79 messager.printMessage(ERROR,
aoqi@0 80 "round " + round + " unexpected value for " + expected.getClass().getName() + ": " + actual);
aoqi@0 81 }
aoqi@0 82 }
aoqi@0 83
aoqi@0 84 void generateSource(String name) {
aoqi@0 85 String text = "class " + name + " { }\n";
aoqi@0 86
aoqi@0 87 try (Writer out = filer.createSourceFile(name).openWriter()) {
aoqi@0 88 out.write(text);
aoqi@0 89 } catch (IOException e) {
aoqi@0 90 throw new Error(e);
aoqi@0 91 }
aoqi@0 92 }
aoqi@0 93
aoqi@0 94 }
aoqi@0 95

mercurial