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

Fri, 21 Dec 2012 08:45:43 -0800

author
darcy
date
Fri, 21 Dec 2012 08:45:43 -0800
changeset 1466
b52a38d4536c
parent 1054
111bbf1ad913
child 2525
2eb010b6cb22
permissions
-rw-r--r--

8005282: Use @library tag with non-relative path for javac tests
Reviewed-by: jjg

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

mercurial