test/tools/javac/processing/model/element/TestResourceVariable.java

Thu, 31 Aug 2017 15:17:03 +0800

author
aoqi
date
Thu, 31 Aug 2017 15:17:03 +0800
changeset 2525
2eb010b6cb22
parent 1466
b52a38d4536c
parent 0
959103a6100f
permissions
-rw-r--r--

merge

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 6911256 6964740 6967842 6961571 7025809
aoqi@0 27 * @summary Test that the resource variable kind is appropriately set
aoqi@0 28 * @author Joseph D. Darcy
aoqi@0 29 * @library /tools/javac/lib
aoqi@0 30 * @build JavacTestingAbstractProcessor TestResourceVariable
aoqi@0 31 * @compile -processor TestResourceVariable -proc:only TestResourceVariable.java
aoqi@0 32 */
aoqi@0 33
aoqi@0 34 import java.io.*;
aoqi@0 35 import javax.annotation.processing.*;
aoqi@0 36 import javax.lang.model.*;
aoqi@0 37 import javax.lang.model.element.*;
aoqi@0 38 import javax.lang.model.type.*;
aoqi@0 39 import javax.lang.model.util.*;
aoqi@0 40 import java.util.*;
aoqi@0 41 import com.sun.source.tree.*;
aoqi@0 42 import com.sun.source.util.*;
aoqi@0 43 import static javax.tools.Diagnostic.Kind.*;
aoqi@0 44
aoqi@0 45 /**
aoqi@0 46 * Using the tree API, retrieve element representations of the
aoqi@0 47 * resource of a try-with-resources statement and verify their kind
aoqi@0 48 * tags are set appropriately.
aoqi@0 49 */
aoqi@0 50 public class TestResourceVariable extends JavacTestingAbstractProcessor implements AutoCloseable {
aoqi@0 51 int resourceVariableCount = 0;
aoqi@0 52
aoqi@0 53 public boolean process(Set<? extends TypeElement> annotations,
aoqi@0 54 RoundEnvironment roundEnv) {
aoqi@0 55 if (!roundEnv.processingOver()) {
aoqi@0 56 Trees trees = Trees.instance(processingEnv);
aoqi@0 57
aoqi@0 58 for(Element rootElement : roundEnv.getRootElements()) {
aoqi@0 59 TreePath treePath = trees.getPath(rootElement);
aoqi@0 60
aoqi@0 61 (new ResourceVariableScanner(trees)).
aoqi@0 62 scan(trees.getTree(rootElement),
aoqi@0 63 treePath.getCompilationUnit());
aoqi@0 64 }
aoqi@0 65 if (resourceVariableCount != 3)
aoqi@0 66 throw new RuntimeException("Bad resource variable count " +
aoqi@0 67 resourceVariableCount);
aoqi@0 68 }
aoqi@0 69 return true;
aoqi@0 70 }
aoqi@0 71
aoqi@0 72 @Override
aoqi@0 73 public void close() {}
aoqi@0 74
aoqi@0 75 private void test1() {
aoqi@0 76 try(TestResourceVariable trv = this) {}
aoqi@0 77 }
aoqi@0 78
aoqi@0 79 private void test2() {
aoqi@0 80 try(TestResourceVariable trv1 = this; TestResourceVariable trv2 = trv1) {}
aoqi@0 81 }
aoqi@0 82
aoqi@0 83 /**
aoqi@0 84 * Verify that a resource variable modeled as an element behaves
aoqi@0 85 * as expected under 6 and latest specific visitors.
aoqi@0 86 */
aoqi@0 87 private static void testResourceVariable(Element element) {
aoqi@0 88 ElementVisitor visitor6 = new ElementKindVisitor6<Void, Void>() {};
aoqi@0 89
aoqi@0 90 try {
aoqi@0 91 visitor6.visit(element);
aoqi@0 92 throw new RuntimeException("Expected UnknownElementException not thrown.");
aoqi@0 93 } catch (UnknownElementException uee) {
aoqi@0 94 ; // Expected.
aoqi@0 95 }
aoqi@0 96
aoqi@0 97 ElementKindVisitor visitorLatest =
aoqi@0 98 new ElementKindVisitor<Object, Void>() {
aoqi@0 99 @Override
aoqi@0 100 public Object visitVariableAsResourceVariable(VariableElement e,
aoqi@0 101 Void p) {
aoqi@0 102 return e; // a non-null value
aoqi@0 103 }
aoqi@0 104 };
aoqi@0 105
aoqi@0 106 if (visitorLatest.visit(element) == null) {
aoqi@0 107 throw new RuntimeException("Null result of resource variable visitation.");
aoqi@0 108 }
aoqi@0 109 }
aoqi@0 110
aoqi@0 111 class ResourceVariableScanner extends TreeScanner<Void, CompilationUnitTree> {
aoqi@0 112 private Trees trees;
aoqi@0 113
aoqi@0 114 public ResourceVariableScanner(Trees trees) {
aoqi@0 115 super();
aoqi@0 116 this.trees = trees;
aoqi@0 117 }
aoqi@0 118 @Override
aoqi@0 119 public Void visitVariable(VariableTree node, CompilationUnitTree cu) {
aoqi@0 120 Element element = trees.getElement(trees.getPath(cu, node));
aoqi@0 121
aoqi@0 122 System.out.println("Name: " + element.getSimpleName() +
aoqi@0 123 "\tKind: " + element.getKind());
aoqi@0 124 if (element.getKind() == ElementKind.RESOURCE_VARIABLE) {
aoqi@0 125 testResourceVariable(element);
aoqi@0 126 resourceVariableCount++;
aoqi@0 127 }
aoqi@0 128 return super.visitVariable(node, cu);
aoqi@0 129 }
aoqi@0 130 }
aoqi@0 131 }

mercurial