1 /*
   2  * Copyright (c) 2013, 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 8012929
  27  * @summary Trees.getElement should work not only for declaration trees, but also for use-trees
  28  * @build TestGetElementReference
  29  * @run main TestGetElementReference
  30  */
  31 
  32 import com.sun.source.tree.CompilationUnitTree;
  33 import com.sun.source.tree.Tree;
  34 import com.sun.source.util.*;
  35 import java.io.File;
  36 import java.io.IOException;
  37 import java.net.URI;
  38 import java.util.Arrays;
  39 import java.util.regex.Matcher;
  40 import java.util.regex.Pattern;
  41 import javax.lang.model.element.Element;
  42 import javax.tools.Diagnostic;
  43 import javax.tools.DiagnosticCollector;
  44 import javax.tools.JavaFileObject;
  45 import javax.tools.SimpleJavaFileObject;
  46 import javax.tools.StandardJavaFileManager;
  47 import javax.tools.ToolProvider;
  48 
  49 public class TestGetElementReference {
  50 
  51     public static void main(String... args) throws IOException {
  52         File source = new File(System.getProperty("test.src", "."), "TestGetElementReferenceData.java").getAbsoluteFile();
  53         StandardJavaFileManager fm = ToolProvider.getSystemJavaCompiler().getStandardFileManager(null, null, null);
  54         DiagnosticCollector<JavaFileObject> diagnostics = new DiagnosticCollector<>();
  55         JavacTask ct = (JavacTask) ToolProvider.getSystemJavaCompiler().getTask(null, null, diagnostics, Arrays.asList("-Xjcov", "-source", "1.8"), null, fm.getJavaFileObjects(source));
  56         Trees trees = Trees.instance(ct);
  57         CompilationUnitTree cut = ct.parse().iterator().next();
  58 
  59         ct.analyze();
  60 
  61         for (Diagnostic<? extends JavaFileObject> d : diagnostics.getDiagnostics()) {
  62             if (d.getKind() == Diagnostic.Kind.ERROR) {
  63                 throw new IllegalStateException("Should have been attributed without errors: " + diagnostics.getDiagnostics());
  64             }
  65         }
  66 
  67         Pattern p = Pattern.compile("/\\*getElement:(.*?)\\*/");
  68         Matcher m = p.matcher(cut.getSourceFile().getCharContent(false));
  69 
  70         while (m.find()) {
  71             TreePath tp = pathFor(trees, cut, m.start() - 1);
  72             Element found = trees.getElement(tp);
  73             String expected = m.group(1);
  74             String actual = found != null ? found.getKind() + ":" + symbolToString(found) : "<null>";
  75 
  76             if (!expected.equals(actual)) {
  77                 throw new IllegalStateException("expected=" + expected + "; actual=" + actual);
  78             }
  79         }
  80     }
  81 
  82     private static TreePath pathFor(final Trees trees, final CompilationUnitTree cut, final int pos) {
  83         final TreePath[] result = new TreePath[1];
  84 
  85         new TreePathScanner<Void, Void>() {
  86             @Override public Void scan(Tree node, Void p) {
  87                 if (   node != null
  88                     && trees.getSourcePositions().getStartPosition(cut, node) <= pos
  89                     && pos <= trees.getSourcePositions().getEndPosition(cut, node)) {
  90                     result[0] = new TreePath(getCurrentPath(), node);
  91                     return super.scan(node, p);
  92                 }
  93                 return null;
  94             }
  95         }.scan(cut, null);
  96 
  97         return result[0];
  98     }
  99 
 100     private static String symbolToString(Element el) {
 101         switch (el.getKind()) {
 102             case METHOD: return symbolToString(el.getEnclosingElement()) + "." + el.toString();
 103             case CONSTRUCTOR: return symbolToString(el.getEnclosingElement().getEnclosingElement()) + "." + el.toString();
 104             default:
 105                 return el.toString();
 106         }
 107     }
 108 
 109     static class TestFileObject extends SimpleJavaFileObject {
 110         private final String text;
 111         public TestFileObject(String text) {
 112             super(URI.create("myfo:/Test.java"), JavaFileObject.Kind.SOURCE);
 113             this.text = text;
 114         }
 115         @Override public CharSequence getCharContent(boolean ignoreEncodingErrors) {
 116             return text;
 117         }
 118     }
 119 
 120 }