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

Fri, 16 Jul 2010 19:35:24 -0700

author
darcy
date
Fri, 16 Jul 2010 19:35:24 -0700
changeset 609
13354e1abba7
child 697
28b021bb889f
permissions
-rw-r--r--

6911256: Project Coin: Support Automatic Resource Management (ARM) blocks in the compiler
6964740: Project Coin: More tests for ARM compiler changes
6965277: Project Coin: Correctness issues in ARM implementation
6967065: add -Xlint warning category for Automatic Resource Management (ARM)
Reviewed-by: jjb, darcy, mcimadamore, jjg, briangoetz
Contributed-by: tball@google.com

     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  */
    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  */
    33 // Bug should be filed for this misbehavior
    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.*;
    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;
    55     public boolean process(Set<? extends TypeElement> annotations,
    56                           RoundEnvironment roundEnv) {
    57        if (!roundEnv.processingOver()) {
    58            Trees trees = Trees.instance(processingEnv);
    60            for(Element rootElement : roundEnv.getRootElements()) {
    61                TreePath treePath = trees.getPath(rootElement);
    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     }
    74     @Override
    75     public void close() {}
    77     private void test1() {
    78         try(TestResourceVariable trv = this) {}
    79     }
    81     private void test2() {
    82         try(TestResourceVariable trv1 = this; TestResourceVariable trv2 = trv1) {}
    83     }
    85     class ResourceVariableScanner extends TreeScanner<Void, CompilationUnitTree> {
    86        private Trees trees;
    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    }
   109    @Override
   110    public SourceVersion getSupportedSourceVersion() {
   111        return SourceVersion.latest();
   112    }
   113 }

mercurial