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

changeset 609
13354e1abba7
child 697
28b021bb889f
equal deleted inserted replaced
608:472e74211e11 609:13354e1abba7
1 /*
2 * Copyright (c) 2010, 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 6911256 6964740
27 * @summary Test that the resource variable kind is appropriately set
28 * @author Joseph D. Darcy
29 * @build TestResourceVariable
30 * @compile/fail -processor TestResourceVariable -proc:only TestResourceVariable.java
31 */
32
33 // Bug should be filed for this misbehavior
34
35 import java.io.*;
36 import javax.annotation.processing.*;
37 import javax.lang.model.*;
38 import javax.lang.model.element.*;
39 import javax.lang.model.type.*;
40 import javax.lang.model.util.*;
41 import java.util.*;
42 import com.sun.source.tree.*;
43 import com.sun.source.util.*;
44 import static javax.tools.Diagnostic.Kind.*;
45
46 /**
47 * Using the tree API, retrieve element representations of the
48 * resource of an ARM block and verify their kind tags are set
49 * appropriately.
50 */
51 @SupportedAnnotationTypes("*")
52 public class TestResourceVariable extends AbstractProcessor implements AutoCloseable {
53 int resourceVariableCount = 0;
54
55 public boolean process(Set<? extends TypeElement> annotations,
56 RoundEnvironment roundEnv) {
57 if (!roundEnv.processingOver()) {
58 Trees trees = Trees.instance(processingEnv);
59
60 for(Element rootElement : roundEnv.getRootElements()) {
61 TreePath treePath = trees.getPath(rootElement);
62
63 (new ResourceVariableScanner(trees)).
64 scan(trees.getTree(rootElement),
65 treePath.getCompilationUnit());
66 }
67 if (resourceVariableCount != 3)
68 throw new RuntimeException("Bad resource variable count " +
69 resourceVariableCount);
70 }
71 return true;
72 }
73
74 @Override
75 public void close() {}
76
77 private void test1() {
78 try(TestResourceVariable trv = this) {}
79 }
80
81 private void test2() {
82 try(TestResourceVariable trv1 = this; TestResourceVariable trv2 = trv1) {}
83 }
84
85 class ResourceVariableScanner extends TreeScanner<Void, CompilationUnitTree> {
86 private Trees trees;
87
88 public ResourceVariableScanner(Trees trees) {
89 super();
90 this.trees = trees;
91 }
92 @Override
93 public Void visitVariable(VariableTree node, CompilationUnitTree cu) {
94 Element element = trees.getElement(trees.getPath(cu, node));
95 if (element == null) {
96 System.out.println("Null variable element: " + node);
97 } else {
98 System.out.println("Name: " + element.getSimpleName() +
99 "\tKind: " + element.getKind());
100 }
101 if (element != null &&
102 element.getKind() == ElementKind.RESOURCE_VARIABLE) {
103 resourceVariableCount++;
104 }
105 return super.visitVariable(node, cu);
106 }
107 }
108
109 @Override
110 public SourceVersion getSupportedSourceVersion() {
111 return SourceVersion.latest();
112 }
113 }

mercurial